Update app.py

app.py

@@ -4,15 +4,13 @@ import os
 from dotenv import load_dotenv
 import http.client
 import json
-import secrets  # Import secrets for secure session key
-
+import secrets
 
 load_dotenv()
 
 app = Flask(__name__)
-app.secret_key = secrets.token_hex(16)  # Use secrets for a strong key
+app.secret_key = secrets.token_hex(16)  # CRUCIAL: Secure session key
 
-# Configure the API key
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 safety_settings = [
@@ -22,6 +20,8 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
+# ... (rest of your ss, model, perform_web_search, format_search_results definitions) ...
+# Use the full `ss` and `model` definitions from the previous response.
 ss = """
 # Prompt System pour Mariam, IA conçu par youssouf 
 
@@ -119,11 +119,9 @@ Mariam est une IA chaleureuse, bienveillante et authentique, conçue pour être
 
 """
 
-
-
 model = genai.GenerativeModel(
     "gemini-2.0-flash-exp",
-    tools="code_execution",  # Fix: Correct parameter name
+    tools="code_execution",
     safety_settings=safety_settings,
     system_instruction=ss,
 )
@@ -200,88 +198,81 @@ def process_uploaded_file(file):
 
 @app.route("/")
 def index():
-      # Initialize chat and web search in session if not present
     if "chat" not in session:
-        session["chat"] = []  # Store chat history directly
+        session["chat"] = []  # Store chat HISTORY as a list of dicts
         session["web_search"] = False
-        session["gemini_chat"] = model.start_chat(history=[])
-
-
     return render_template("index.html", chat=session["chat"], web_search=session["web_search"])
 
 
-
 @app.route("/send_message", methods=["POST"])
 def send_message():
     prompt = request.form.get("prompt")
-    web_search = request.form.get("web_search") == "true"  # Convert string to boolean
+    web_search = request.form.get("web_search") == "true"
     file = request.files.get("file")
 
     uploaded_gemini_file = None
     if file:
         uploaded_gemini_file = process_uploaded_file(file)
 
-    if "gemini_chat" not in session: # Initialize if it doesn't already exists.
-        session["gemini_chat"] = model.start_chat(history=[])
-        session.modified = True  #Important to save session changes!
-
     # Add user message to chat history
     session["chat"].append({"role": "user", "text": prompt})
-    session.modified = True
+    session.modified = True  # IMPORTANT: Mark session as modified
 
+    # 1. Get chat history from session
+    chat_history = session["chat"]
+
+    # 2. Start a NEW chat session with the history
+    gemini_chat = model.start_chat(history=[
+        {"role": msg["role"], "parts": [msg["text"]]} for msg in chat_history if msg["role"] != "assistant" # Recreate history format
+    ])
+    
     try:
-        # Web search
+        # Web search (same as before)
         web_results = None
         if web_search:
-            web_results = perform_web_search(prompt)
-            if web_results:
+             web_results = perform_web_search(prompt)
+             if web_results:
                 formatted_results = format_search_results(web_results)
                 prompt = (
                     f"Question: {prompt}\n\nRésultats de recherche web:\n"
                     f"{formatted_results}\n\nPourrais-tu analyser ces informations et "
                     f"me donner une réponse complète?"
                 )
-            else:
-                #Handle the "no results" case
+             else:
                 prompt = f"Question: {prompt}\n\n(Aucun résultat de recherche trouvé. Répondez en vous basant sur vos connaissances.)"
 
-        # Send message to Gemini
-        gemini_chat = session["gemini_chat"] # Get the chat object.
+        # Send message to Gemini (using the recreated chat)
         if uploaded_gemini_file:
             response = gemini_chat.send_message([uploaded_gemini_file, "\n\n", prompt])
         else:
-             response = gemini_chat.send_message(prompt)
-
-
+            response = gemini_chat.send_message(prompt)
+            
         # Add assistant response to chat history
         session["chat"].append({"role": "assistant", "text": response.text})
-        session["gemini_chat"] = gemini_chat #Reassign back!
-        session.modified = True #Important to save session changes!
-
+        session.modified = True  # IMPORTANT
 
         return jsonify({"role": "assistant", "text": response.text})
 
     except Exception as e:
         error_message = f"Error sending message: {e}"
-        print(error_message) # Log to console.
-        return jsonify({"role": "assistant", "text": error_message}), 500 #Return 500 error code
-
+        print(error_message)
+        return jsonify({"role": "assistant", "text": error_message}), 500
 
 
 @app.route("/toggle_web_search", methods=["POST"])
 def toggle_web_search():
     session["web_search"] = not session["web_search"]
-    session.modified = True  # Mark session as modified
+    session.modified = True  # IMPORTANT
     return jsonify({"web_search": session["web_search"]})
 
+
 @app.route("/clear_chat", methods=["POST"])
 def clear_chat():
-    session.pop("chat", None)
-    session.pop("gemini_chat", None)
+    session.pop("chat", None)  # Safely remove chat history
     session["web_search"] = False
-    session.modified = True
+    session.modified = True  # IMPORTANT
     return jsonify({"status": "success"})
 
 
 if __name__ == "__main__":
-    app.run(debug=True)
+    app.run(debug=True)  # Turn off debug mode