Update app.py

app.py

@@ -353,10 +353,6 @@ def summarize_web_results(query: str, search_results: List[Dict[str, str]], conv
     except Exception as e:
         return f"An error occurred during summarization: {str(e)}"
 
-
-import os
-import google.generativeai as genai
-
 def get_response_from_gemini(query, context, file_type, num_calls=1, temperature=0.2):
     # Configure the Gemini API
     genai.configure(api_key=os.environ["GEMINI_API_KEY"])
@@ -388,16 +384,21 @@ def get_response_from_gemini(query, context, file_type, num_calls=1, temperature
 
     full_prompt = f"{system_instruction}\n\nContext:\n{context}\n\nUser query: {query}"
 
+    full_response = ""
     for _ in range(num_calls):
         try:
             # Generate content with streaming enabled
             response = model.generate_content(full_prompt, stream=True)
             for chunk in response:
                 if chunk.text:
-                    yield chunk.text
+                    full_response += chunk.text
+                    yield full_response  # Yield the accumulated response so far
         except Exception as e:
             yield f"An error occurred with the Gemini model: {str(e)}. Please try again."
 
+    if not full_response:
+        yield "No response generated from the Gemini model."
+
 def get_response_from_excel(query, model, context, num_calls=3, temperature=0.2):
     logging.info(f"Getting response from Excel using model: {model}")