Update app.py

app.py

@@ -258,41 +258,41 @@ class BasicAgent:
     
     def _formulate_direct_answer(self, relevant_info, question):
   
-    if not self.model:
-        return f"Based on available information: {relevant_info}"
-        
-    if self.model.startswith('gemini'):
-        try:
-          
-            if not hasattr(self, 'gemini_model') or self.gemini_model is None:
-                self._init_gemini_model()
-            
-        
-            prompt = f"""
-            Question: {question}
-
-            Relevant information: {relevant_info}
-
-            Instructions:
-            1. Provide a concise answer based only on the given information
-            2. If the information doesn't contain the answer, say so honestly
-            3. Use only facts from the provided information
-            4. Format your response as a direct answer to the user
-            """
+        if not self.model:
+            return f"Based on available information: {relevant_info}"
             
-            response = self.gemini_model.generate_content(prompt)
-            if response and hasattr(response, 'text'):
-                return response.text
-            else:
-                print("Gemini response was empty or invalid")  
-                return f"Based on the information: {relevant_info[:200]}..."
+        if self.model.startswith('gemini'):
+            try:
+              
+                if not hasattr(self, 'gemini_model') or self.gemini_model is None:
+                    self._init_gemini_model()
                 
-        except Exception as e:
-            print(f"Error using Gemini model: {e}") 
-            traceback.print_exc()  
-            return f"Based on the search: {relevant_info[:200]}..."
             
-    return f"Based on the information: {relevant_info[:200]}..." 
+                prompt = f"""
+                Question: {question}
+    
+                Relevant information: {relevant_info}
+    
+                Instructions:
+                1. Provide a concise answer based only on the given information
+                2. If the information doesn't contain the answer, say so honestly
+                3. Use only facts from the provided information
+                4. Format your response as a direct answer to the user
+                """
+                
+                response = self.gemini_model.generate_content(prompt)
+                if response and hasattr(response, 'text'):
+                    return response.text
+                else:
+                    print("Gemini response was empty or invalid")  
+                    return f"Based on the information: {relevant_info[:200]}..."
+                    
+            except Exception as e:
+                print(f"Error using Gemini model: {e}") 
+                traceback.print_exc()  
+                return f"Based on the search: {relevant_info[:200]}..."
+                
+        return f"Based on the information: {relevant_info[:200]}..."