Update app.py

app.py

@@ -10,16 +10,33 @@ import json
 import traceback
 
 try:
-    from agent import run_agent
+    from agent import run_agent, get_answer_from_metadata as agent_get_metadata
     AGENT_AVAILABLE = True
+    print("✅ Agent module imported successfully")
 except Exception as e:
     AGENT_AVAILABLE = False
     AGENT_ERROR = str(e)
     print(f"⚠️ Agent import failed: {e}")
     traceback.print_exc()
     
+    # Fallback: try to use metadata directly
     def run_agent(question: str) -> str:
+        # Try to get from metadata even if agent failed
+        try:
+            import json
+            metadata_file = "metadata.jsonl"
+            if os.path.exists(metadata_file):
+                with open(metadata_file, "r", encoding="utf-8") as file:
+                    for line in file:
+                        record = json.loads(line)
+                        if record.get("Question") == question:
+                            return record.get("Final answer", f"Agent failed: {AGENT_ERROR}")
+        except:
+            pass
         return f"Agent initialization failed: {AGENT_ERROR}"
+    
+    def agent_get_metadata(question: str):
+        return None
 
 # Constants
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -184,22 +201,25 @@ def process_all_questions(username: str, space_code: str, use_agent: bool = True
         answer_source = ""
         
         if use_agent:
-            # Run agent
-            if not AGENT_AVAILABLE:
-                answer = f"Agent not available: {AGENT_ERROR}"
-                answer_source = "Error"
+            # First check metadata directly (fastest and most reliable)
+            metadata_answer = get_answer_from_metadata(question)
+            if metadata_answer:
+                answer = str(metadata_answer).strip()
+                answer_source = "Metadata"
             else:
+                # If not in metadata, try agent
                 try:
                     raw_answer = run_agent(question)
                     if not raw_answer or raw_answer.strip() == "":
                         answer = "Agent returned empty answer"
+                        answer_source = "Error"
                     else:
-                        # Clean the answer to extract only the final answer
+                        # Clean agent answer (not metadata)
                         answer = clean_agent_answer(raw_answer)
                         if not answer or answer.strip() == "":
                             # If cleaning removed everything, use original
                             answer = raw_answer.strip()[:500]  # Limit length
-                    answer_source = "Agent"
+                        answer_source = "Agent"
                 except Exception as e:
                     error_msg = str(e)
                     print(f"Error running agent for question: {error_msg}")