fixing agent

app.py

@@ -35,8 +35,11 @@ class BasicAgent:
             raise e
     
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        print(f"🤖 Agent received question: {question[:100]}...")
+        
         try:
+            print("🔄 Making API call to HF Inference...")
+            
             # Create a more detailed prompt for better responses
             system_prompt = "You are a helpful AI assistant. Provide clear, accurate, and concise answers to questions."
             
@@ -56,14 +59,33 @@ class BasicAgent:
                 temperature=0.7,
             )
             
-            answer = completion.choices[0].message.content
-            print(f"Agent returning answer: {answer[:100]}...")
+            print("✅ API call successful!")
+            answer = completion.choices[0].message.content.strip()
+            print(f"📝 Agent returning answer: {answer[:200]}...")
+            
+            # Ensure we're not returning empty or None answers
+            if not answer or answer.lower().strip() == "":
+                return "I apologize, but I couldn't generate a proper response to your question."
+            
             return answer
             
         except Exception as e:
-            error_msg = f"Error processing question: {str(e)}"
-            print(f"❌ {error_msg}")
-            return error_msg
+            error_msg = f"❌ HF API Error: {str(e)}"
+            print(error_msg)
+            print(f"📋 Full error details: {repr(e)}")
+            
+            # Return a more informative error message
+            return f"Sorry, I encountered an error while processing your question: {str(e)}"
+    
+    def test_connection(self):
+        """Test if the HF API connection is working"""
+        try:
+            test_response = self("What is 2+2?")
+            print(f"🧪 Test response: {test_response}")
+            return True, test_response
+        except Exception as e:
+            print(f"🚫 Test failed: {e}")
+            return False, str(e)
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
@@ -94,10 +116,21 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent
     try:
+        print("🚀 Initializing BasicAgent...")
         agent = BasicAgent()
+        
+        # Test the agent before proceeding
+        print("🧪 Testing agent connection...")
+        test_success, test_result = agent.test_connection()
+        if not test_success:
+            return f"❌ Agent test failed: {test_result}\nPlease check your HF_TOKEN in environment variables.", None
+        
+        print(f"✅ Agent test successful: {test_result[:100]}...")
+        
     except Exception as e:
-        print(f"Error instantiating agent: {e}")
-        return f"Error initializing agent: {e}", None
+        error_msg = f"❌ Error initializing agent: {e}"
+        print(error_msg)
+        return error_msg, None
 
     # 2. Fetch Questions
     print(f"Fetching questions from: {questions_url}")