Update app.py

app.py

@@ -9,7 +9,6 @@ import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 from smolagents.models import OpenAIServerModel
 
-# System prompt for GAIA evaluation
 SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
 Report your thoughts, and finish your answer with the following template:
 FINAL ANSWER: [YOUR FINAL ANSWER].
@@ -18,10 +17,9 @@ of numbers and/or strings. If you are asked for a number, don't use comma to wri
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-def extract_final_answer(response: str) -> str:
-    """Extract the FINAL ANSWER: portion from the model's response."""
-    match = re.search(r"FINAL ANSWER:\s*(.*)", response, re.IGNORECASE)
-    return match.group(1).strip() if match else response.strip()
+def extract_final_answer(response_text: str) -> str:
+    match = re.search(r"FINAL ANSWER:\s*(.*)", response_text, re.IGNORECASE)
+    return match.group(1).strip() if match else response_text.strip()
 
 class MyAgent:
     def __init__(self):
@@ -37,11 +35,9 @@ class MyAgent:
             {"role": "user", "content": question}
         ]
         try:
-            response = self.model(messages)       
-        # If response is a dict, get the text content inside
-            if isinstance(response, dict):        
-            # Adjust depending on exact structure, commonly:
-            # {'choices': [{'message': {'content': 'the text'}}], ...}
+            response = self.model(messages)
+            # Extract the actual text from response
+            if isinstance(response, dict):
                 choices = response.get('choices')
                 if choices and len(choices) > 0:
                     text = choices[0].get('message', {}).get('content', '')
@@ -50,14 +46,13 @@ class MyAgent:
             elif isinstance(response, str):
                 text = response
             else:
-                text = str(response)  # fallback
-        
-        return extract_final_answer(text)
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return f"AGENT ERROR: {e}"
+                text = str(response)
 
+            return extract_final_answer(text)
+        except Exception as e:
+            import traceback
+            traceback.print_exc()
+            return f"AGENT ERROR: {e}"
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
@@ -179,3 +174,4 @@ if __name__ == "__main__":
     demo.launch(debug=True, share=False)
 
 
+