Update app.py

app.py

@@ -45,22 +45,25 @@ 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]:
+        for item in data['organic'][:3]:  # Limit to top 3 results
             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]:
+        for item in data['peopleAlsoAsk'][:2]:  # Limit to top 2 questions
             result += f"- **{item['question']}**\n"
             result += f"  {item['snippet']}\n\n"
     
@@ -126,28 +129,16 @@ 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?"""
         
-        # Add user message to chat history first
-        response_text = ""
-        
-        # Display assistant message with streaming
+        # 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
         with st.chat_message("assistant"):
-            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)
+            st.markdown(response.text)
     
     except Exception as e:
         st.error(f"Erreur lors de l'envoi du message : {e}")