Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import pandas as pd
 from langchain_core.messages import HumanMessage
-from agent import build_graph  # Now this will work
+from agent import build_graph
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -18,7 +18,7 @@ class BasicAgent:
         # Check environment variables
         print("=== ENVIRONMENT VARIABLES ===")
         for key in sorted(os.environ.keys()):
-            if any(term in key.upper() for term in ['OPENAI', 'API_KEY', 'TOKEN']):
+            if any(term in key.upper() for term in ['OPENAI', 'API_KEY', 'TOKEN', 'TAVILY']):
                 value = os.environ[key]
                 print(f"{key}: {value[:10] if value else 'None'}...")
         
@@ -46,26 +46,12 @@ class BasicAgent:
             print(f"Invoking graph with messages: {len(messages)}")
             
             result = self.graph.invoke({"messages": messages})
-            
-            # Debug: Print result structure
-            print(f"Result type: {type(result)}")
-            if isinstance(result, dict):
-                print(f"Result keys: {list(result.keys())}")
-                if 'messages' in result:
-                    print(f"Number of messages: {len(result['messages'])}")
+            print(f"Graph result keys: {result.keys() if isinstance(result, dict) else 'Not a dict'}")
             
             if 'messages' in result and result['messages']:
-                # Find the final answer in the messages
-                for msg in reversed(result['messages']):
-                    if msg.type == "ai" and not hasattr(msg, 'tool_calls'):
-                        # Extract just the final answer if it's marked
-                        if "Final Answer:" in msg.content:
-                            return msg.content.split("Final Answer:")[-1].strip()
-                        return msg.content
-                        
-                # If no AI message found, return the last message content
-                last_msg = result['messages'][-1]
-                return last_msg.content
+                answer = result['messages'][-1].content
+                print(f"Answer (first 100 chars): {answer[:100]}...")
+                return answer
             else:
                 print("No messages in result")
                 return "I apologize, but I couldn't generate a response."
@@ -178,8 +164,7 @@ def test_agent():
         test_questions = [
             "What is 2 + 3?",
             "What is 10 * 5?",
-            "What is the current population of Canada? Respond with just the number.",
-            "Convert 'July 20, 1969' to YYYY-MM-DD format."
+            "Search for information about Python programming"
         ]
         
         for q in test_questions:
@@ -198,12 +183,13 @@ with gr.Blocks() as demo:
     ## Setup Instructions:
     1. Get an OpenAI API key from https://platform.openai.com/api-keys
     2. Add it as `OPENAI_API_KEY` in your Hugging Face Space secrets
-    3. Log in with the button below
-    4. Click **Run Evaluation & Submit All Answers**
+    3. (Optional) Add `TAVILY_API_KEY` for web search functionality
+    4. Log in with the button below
+    5. Click **Run Evaluation & Submit All Answers**
     
     ## Current Configuration:
-    - Model: GPT-4-turbo
-    - Tools: Web search, Image description, Math calculator, Date/currency conversion
+    - Model: GPT-3.5-turbo (change to GPT-4 in agent.py if you have access)
+    - Tools: Math operations, Wikipedia search, Arxiv search, Web search (if Tavily configured)
     """)
     
     with gr.Row():
@@ -228,4 +214,4 @@ if __name__ == "__main__":
     else:
         print("⚠️  OpenAI API Key not found - please add OPENAI_API_KEY to your Space secrets")
     
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)