Update app.py

app.py

@@ -121,14 +121,31 @@ class SimpleAgent:
             raise ValueError("Agent not initialized. Call setup() first.")
         
         try:
-            # For now, we'll use a simple approach with the inference client
-            # In a real implementation, you'd want to use proper agent tools
+            # Check if the question mentions an attached file
+            has_attachment = "Attached file available at:" in question
             
             # Create a prompt that encourages the model to give a direct answer
-            prompt = f"""Question: {question}
+            if has_attachment:
+                prompt = f"""You are an AI assistant that can analyze questions and attached files. 
+
+Question: {question}
+Question Type: {question_type}
+
+For questions with attached files, you should:
+1. First acknowledge the attached file
+2. If needed, mention that you would analyze the file content
+3. Provide the best answer you can based on the question
+
+Please provide a direct, factual answer. End your response with "Final Answer: [your answer]"
+
+Response:"""
+            else:
+                prompt = f"""You are an AI assistant that answers questions directly and accurately.
+
+Question: {question}
 Question Type: {question_type}
 
-Please provide a direct answer to this question. End your response with "Final Answer: [your answer]"
+Please provide a direct, factual answer to this question. Think step by step if needed, but focus on giving a clear, concise answer. End your response with "Final Answer: [your answer]"
 
 Response:"""
             
@@ -292,6 +309,7 @@ def run_and_submit_all(request: gr.Request):
             question_text = item.get("question", "")
             question_type = item.get("question_type", "")
             question_preview = item.get("question_preview", question_text[:50] + "...")
+            file_name = item.get("file_name")  # Check for attached files
             tools_code = item.get("tools_code")  # This might be used in advanced implementations
 
             if not task_id:
@@ -311,8 +329,11 @@ def run_and_submit_all(request: gr.Request):
             status = "ERROR"
 
             try:
+                # Prepare question with file attachment if present
+                question_with_attachment = compute_question_with_attachment(question_text, task_id, file_name, api_url)
+                
                 # Call the agent
-                submitted_answer = agent(question_text, question_type, tools_code)
+                submitted_answer = agent(question_with_attachment, question_type, tools_code)
                 processing_time = time.time() - start_time
                 status = "OK" if submitted_answer != "ERROR" else "ERROR"
                 
@@ -330,9 +351,13 @@ def run_and_submit_all(request: gr.Request):
                     "Question Preview": question_preview[:80] + "..." if len(question_preview) > 80 else question_preview, 
                     "Submitted Answer": submitted_answer[:100] + "..." if len(submitted_answer) > 100 else submitted_answer, 
                     "Processing Time (s)": f"{processing_time:.2f}",
-                    "Status": status
+                    "Status": status,
+                    "Has File": "Yes" if file_name else "No"
                 })
 
+                # Add rate limiting delay to prevent API errors
+                time.sleep(2)  # Start with a smaller delay, increase if needed
+
             except Exception as e:
                 processing_time = time.time() - start_time
                 logging.exception(f"Error running agent on task {task_id}: {e}")
@@ -350,13 +375,15 @@ def run_and_submit_all(request: gr.Request):
                     "Question Preview": question_preview[:80] + "..." if len(question_preview) > 80 else question_preview, 
                     "Submitted Answer": submitted_answer[:100] + "..." if len(submitted_answer) > 100 else submitted_answer, 
                     "Processing Time (s)": "ERROR",
-                    "Status": status
+                    "Status": status,
+                    "Has File": "Yes" if file_name else "No"
                 })
 
             # Yield progressive updates to Gradio UI with better formatting
             progress_text = (
                 f"🔄 Processing Question {i}/{len(questions_data)}\n"
                 f"📝 Question Type: {question_type}\n"
+                f"📎 Has Attachment: {'Yes' if file_name else 'No'}\n"
                 f"❓ Question Preview: {question_preview[:100]}{'...' if len(question_preview) > 100 else ''}\n"
                 f"💭 Agent Answer: {submitted_answer[:150]}{'...' if len(submitted_answer) > 150 else ''}\n"
                 f"⏱️ Processing Time: {processing_time:.2f}s\n"
@@ -434,6 +461,18 @@ def run_and_submit_all(request: gr.Request):
             agent.cleanup()
 
 
+def compute_question_with_attachment(question: str, task_id: str, file_name: str, api_url: str) -> str:
+    """
+    Adds file attachment information to the question if a file is present.
+    This allows the agent to know about and access attached files.
+    """
+    if file_name:
+        file_url = f"{api_url}/files/{task_id}"
+        return f"{question}\n\nAttached file available at: {file_url}"
+    else:
+        return question
+
+
 # --- Gradio Interface ---
 with gr.Blocks() as demo:
     gr.Markdown("# Hugging Face Agent Certification Runner")