Update app.py

app.py

@@ -27,9 +27,10 @@ class BasicAgent:
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
         })
         self.cache = {}
+        
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = agent.answer_question({question})
+        fixed_answer = self.agent.generate({question})
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
@@ -240,25 +241,6 @@ class BasicAgent:
         
         return min(0.99, max(0.1, confidence))
 
-# Example usage
-if __name__ == "__main__":
-    agent = BasicAgent()
-    
-    questions = [
-        "How many studio albums did Taylor Swift release between 2010 and 2015?",
-        "When was the first iPhone released?",
-        "What is the capital of Canada?",
-        "List the planets in our solar system"
-    ]
-    
-    for question in questions:
-        result = agent.answer_question(question)
-        print(f"\nAnswer: {result['answer']}")
-        #print(f"Source: {result['source']}")
-        #print(f"Confidence: {result['confidence']}")
-        #print("="*50)
-
-
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """