Update app.py

app.py

@@ -4,6 +4,8 @@ import os
 from dotenv import load_dotenv
 import http.client
 import json
+import asyncio
+from typing import AsyncGenerator
 
 load_dotenv()
 
@@ -17,7 +19,8 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-model = genai.GenerativeModel('gemini-2.0-flash-exp', tools='code_execution',
+model = genai.GenerativeModel('gemini-2.0-flash-exp', 
+                            tools='code_execution',
                             safety_settings=safety_settings,
                             system_instruction="Tu es un assistant intelligent. ton but est d'assister au mieux que tu peux. tu as été créé par Aenir et tu t'appelles Mariam")
 
@@ -45,25 +48,22 @@ def format_search_results(data):
     
     result = ""
     
-    # Knowledge Graph
     if 'knowledgeGraph' in data:
         kg = data['knowledgeGraph']
         result += f"### {kg.get('title', '')}\n"
         result += f"*{kg.get('type', '')}*\n\n"
         result += f"{kg.get('description', '')}\n\n"
     
-    # Organic Results
     if 'organic' in data:
         result += "### Résultats principaux:\n"
-        for item in data['organic'][:3]:  # Limit to top 3 results
+        for item in data['organic'][:3]:
             result += f"- **{item['title']}**\n"
             result += f"  {item['snippet']}\n"
             result += f"  [Lien]({item['link']})\n\n"
     
-    # People Also Ask
     if 'peopleAlsoAsk' in data:
         result += "### Questions fréquentes:\n"
-        for item in data['peopleAlsoAsk'][:2]:  # Limit to top 2 questions
+        for item in data['peopleAlsoAsk'][:2]:
             result += f"- **{item['question']}**\n"
             result += f"  {item['snippet']}\n\n"
     
@@ -75,7 +75,7 @@ def role_to_streamlit(role):
     else:
         return role
 
-# Add chat and settings to session state
+# Initialize session state
 if "chat" not in st.session_state:
     st.session_state.chat = model.start_chat(history=[])
 if "web_search" not in st.session_state:
@@ -97,7 +97,6 @@ for message in st.session_state.chat.history:
     with st.chat_message(role_to_streamlit(message.role)):
         st.markdown(message.parts[0].text)
 
-# Function to handle file upload with Gemini
 def process_uploaded_file(file):
     if file is not None:
         with open(os.path.join("temp", file.name), "wb") as f:
@@ -109,6 +108,20 @@ def process_uploaded_file(file):
             st.error(f"Erreur lors du téléchargement du fichier : {e}")
             return None
 
+async def stream_response(prompt: str, uploaded_gemini_file=None) -> AsyncGenerator[str, None]:
+    try:
+        if uploaded_gemini_file:
+            response = await st.session_state.chat.send_message_async([uploaded_gemini_file, "\n\n", prompt], stream=True)
+        else:
+            response = await st.session_state.chat.send_message_async(prompt, stream=True)
+        
+        async for chunk in response:
+            if chunk.text:
+                yield chunk.text
+    except Exception as e:
+        st.error(f"Erreur lors du streaming : {e}")
+        yield "Désolé, une erreur s'est produite lors de la génération de la réponse."
+
 # Chat input and processing
 if prompt := st.chat_input("Hey?"):
     uploaded_gemini_file = None
@@ -117,8 +130,7 @@ if prompt := st.chat_input("Hey?"):
     
     # Display user message
     st.chat_message("user").markdown(prompt)
-    print(prompt)
-    print("------------")
+    
     try:
         # Perform web search if enabled
         web_results = None
@@ -129,16 +141,18 @@ if prompt := st.chat_input("Hey?"):
                     formatted_results = format_search_results(web_results)
                     prompt = f"""Question: {prompt}\n\nRésultats de recherche web:\n{formatted_results}\n\nPourrais-tu analyser ces informations et me donner une réponse complète?"""
         
-        # Send message to Gemini
-        if uploaded_gemini_file:
-            response = st.session_state.chat.send_message([uploaded_gemini_file, "\n\n", prompt])
-        else:
-            response = st.session_state.chat.send_message(prompt)
-
-        print(response.text)
-        # Display assistant response
+        # Create a placeholder for the streaming response
         with st.chat_message("assistant"):
-            st.markdown(response.text)
+            response_placeholder = st.empty()
+            full_response = ""
+            
+            # Stream the response
+            for response_chunk in asyncio.run(stream_response(prompt, uploaded_gemini_file)):
+                full_response += response_chunk
+                response_placeholder.markdown(full_response + "▌")
+            
+            # Update the placeholder with the complete response
+            response_placeholder.markdown(full_response)
     
     except Exception as e:
         st.error(f"Erreur lors de l'envoi du message : {e}")