Update app.py

app.py

@@ -1,14 +1,16 @@
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, request, jsonify, session
 import google.generativeai as genai
 import os
 from dotenv import load_dotenv
 import http.client
 import json
 from werkzeug.utils import secure_filename
+import markdown2
 
 app = Flask(__name__)
 app.config['UPLOAD_FOLDER'] = 'temp'
 app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB max file size
+app.secret_key = 'your-secret-key-here'  # Change this to a secure secret key
 
 load_dotenv()
 
@@ -22,10 +24,11 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-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")
+def get_chat_model():
+    return 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")
 
 def perform_web_search(query):
     conn = http.client.HTTPSConnection("google.serper.dev")
@@ -67,7 +70,9 @@ def format_search_results(data):
 
 @app.route('/')
 def home():
-    return render_template('index.html')
+    if 'chat_history' not in session:
+        session['chat_history'] = []
+    return render_template('index.html', chat_history=session['chat_history'])
 
 @app.route('/chat', methods=['POST'])
 def chat():
@@ -75,6 +80,9 @@ def chat():
     prompt = data.get('message')
     web_search_enabled = data.get('web_search', False)
     
+    if 'chat' not in session:
+        session['chat'] = get_chat_model().start_chat(history=[])
+    
     try:
         web_results = None
         if web_search_enabled:
@@ -83,8 +91,31 @@ def chat():
                 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?"""
         
-        response = model.generate_content(prompt)
-        return jsonify({"response": response.text})
+        chat = session['chat']
+        response = chat.send_message(prompt)
+        
+        # Convert Markdown to HTML for the response
+        response_html = markdown2.markdown(response.text, extras=["fenced-code-blocks", "tables"])
+        
+        # Update chat history
+        if 'chat_history' not in session:
+            session['chat_history'] = []
+        
+        session['chat_history'].append({
+            'role': 'user',
+            'content': prompt
+        })
+        session['chat_history'].append({
+            'role': 'assistant',
+            'content': response.text,
+            'content_html': response_html
+        })
+        session.modified = True
+        
+        return jsonify({
+            "response": response.text,
+            "response_html": response_html
+        })
     
     except Exception as e:
         return jsonify({"error": str(e)}), 500
@@ -109,6 +140,11 @@ def upload_file():
         except Exception as e:
             return jsonify({"error": str(e)}), 500
 
+@app.route('/clear', methods=['POST'])
+def clear_history():
+    session.clear()
+    return jsonify({"success": True})
+
 if __name__ == '__main__':
     os.makedirs("temp", exist_ok=True)
     app.run(debug=True)