Update app.py

app.py

@@ -37,9 +37,48 @@ class BasicAgent:
         print(f"Agent returning answer: {final_answer[:50]}...")
         return final_answer
 
-    def process_question (self, question:str) -> str:
-        search_results = cached_search(question) if self.tools and search_tool in self.tools else "No search results available."
-        return f"Based on my analysis of '{question}', I found: {search_results[:500]}..."
+    def process_question(self, question:str) -> str:
+        try:
+            search_results = cached_search(question) if self.tools and search_tool in self.tools else "No search results available."
+        
+            # Extract key information and provide direct answer
+            relevant_info = self._extract_key_info(search_results, question)
+            direct_answer = self._formulate_direct_answer(relevant_info, question)
+        
+            return direct_answer
+        except Exception as e:
+            if "too many requests" in str(e).lower():
+                time.sleep(2)
+                try:
+                    search_results = cached_search(question)
+                    relevant_info = self._extract_key_info(search_results, question)
+                    return self._formulate_direct_answer(relevant_info, question)
+                except:
+                    return self._get_fallback_answer(question)
+            return self._get_fallback_answer(question)
+    
+    def _extract_key_info(self, search_results, question):
+        # Split results into sentences and find most relevant
+        sentences = search_results.split('. ')
+        if len(sentences) <= 3:
+            return search_results[:250]  # If few sentences, return first portion
+        
+        # Try to find sentence with keywords from question
+        keywords = [w for w in question.lower().split() if len(w) > 3]
+        for sentence in sentences:
+            sentence_lower = sentence.lower()
+            if any(keyword in sentence_lower for keyword in keywords):
+                return sentence
+        
+        # Fallback to first few sentences
+        return '. '.join(sentences[:2])
+
+    def _formulate_direct_answer(self, relevant_info, question):
+        # Format as direct, concise answer
+        return relevant_info
+    
+    def _get_fallback_answer(self, question):
+        return f"Based on the information available, I cannot provide a specific answer to your question about {question.split()[0:3]}..."
         
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):