Update app.py

app.py

@@ -45,25 +45,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"
     
@@ -129,16 +126,28 @@ 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
+        # Add user message to chat history first
+        response_text = ""
+        
+        # Display assistant message with streaming
         with st.chat_message("assistant"):
-            st.markdown(response.text)
+            message_placeholder = st.empty()
+            
+            # Send message to chat with streaming
+            if uploaded_gemini_file:
+                response = st.session_state.chat.send_message([uploaded_gemini_file, "\n\n", prompt], stream=True)
+            else:
+                response = st.session_state.chat.send_message(prompt, stream=True)
+            
+            # Process the streaming response
+            for chunk in response:
+                if chunk.text:
+                    response_text += chunk.text
+                    # Update the message placeholder with the accumulated response
+                    message_placeholder.markdown(response_text + "▌")
+            
+            # Remove the cursor and display final response
+            message_placeholder.markdown(response_text)
     
     except Exception as e:
         st.error(f"Erreur lors de l'envoi du message : {e}")