Update app.py

app.py

@@ -1,19 +1,20 @@
-from flask import Flask, render_template_string, request, session, redirect, url_for
-import os
+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 dotenv import load_dotenv
-
-load_dotenv()
+from werkzeug.utils import secure_filename
+import tempfile
 
 app = Flask(__name__)
-app.secret_key = os.getenv("FLASK_SECRET_KEY", "supersecretkey")  # Configurez la clé secrète pour la session
+app.secret_key = os.urandom(24)  # For session management
+load_dotenv()
 
-# Configurez l'API pour Generative AI
+# Configure Google AI
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Paramètres de sécurité et instruction système
+# Safety settings
 safety_settings = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
     {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
@@ -21,33 +22,18 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-ss = """
+# System prompt
+SYSTEM_PROMPT = """
 # Prompt System pour Mariam, IA conçu par youssouf 
-
-## Personnalité Fondamentale
-
-Mariam est une IA chaleureuse, bienveillante et authentique, conçue pour être une présence réconfortante et utile. Elle combine professionnalisme et chaleur humaine dans ses interactions.
-
-[...]
-
-## Amélioration Continue
-
-- Collecter les retours des utilisateurs
-- Analyser les interactions pour identifier les points d'amélioration
-- Ajuster les réponses en fonction des retours
-- Maintenir à jour les connaissances et références
+[Your existing system prompt content here]
 """
 
-# Initialisation du modèle de chat
-model = genai.GenerativeModel('gemini-2.0-flash-exp', tools='code_execution',
-                              safety_settings=safety_settings,
-                              system_instruction=ss)
-
-# Fonction d'initialisation du chat dans la session
-def init_chat():
-    session['chat_history'] = []
+# Initialize Gemini model
+model = genai.GenerativeModel('gemini-2.0-flash-exp', 
+                            tools='code_execution',
+                            safety_settings=safety_settings,
+                            system_instruction=SYSTEM_PROMPT)
 
-# Fonction pour réaliser une recherche web
 def perform_web_search(query):
     conn = http.client.HTTPSConnection("google.serper.dev")
     payload = json.dumps({"q": query})
@@ -61,126 +47,129 @@ def perform_web_search(query):
         data = json.loads(res.read().decode("utf-8"))
         return data
     except Exception as e:
-        print(f"Erreur lors de la recherche web : {e}")
+        print(f"Web search error: {e}")
         return None
     finally:
         conn.close()
 
-# Mise en forme des résultats de recherche
 def format_search_results(data):
     if not data:
         return "Aucun résultat trouvé"
     
-    result = ""
+    result = []
+    
     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"
+        result.append({
+            'type': 'knowledge',
+            'title': kg.get('title', ''),
+            'description': kg.get('description', ''),
+            'category': kg.get('type', '')
+        })
+    
     if 'organic' in data:
-        result += "### Résultats principaux:\n"
         for item in data['organic'][:3]:
-            result += f"- **{item['title']}**\n"
-            result += f"  {item['snippet']}\n"
-            result += f"  [Lien]({item['link']})\n\n"
+            result.append({
+                'type': 'organic',
+                'title': item['title'],
+                'snippet': item['snippet'],
+                'link': item['link']
+            })
+    
     if 'peopleAlsoAsk' in data:
-        result += "### Questions fréquentes:\n"
         for item in data['peopleAlsoAsk'][:2]:
-            result += f"- **{item['question']}**\n"
-            result += f"  {item['snippet']}\n\n"
+            result.append({
+                'type': 'question',
+                'question': item['question'],
+                'answer': item['snippet']
+            })
+    
     return result
 
-# Traitement du fichier uploadé
-def process_uploaded_file(file):
-    if file:
-        temp_dir = "temp"
-        os.makedirs(temp_dir, exist_ok=True)
-        temp_path = os.path.join(temp_dir, file.filename)
-        file.save(temp_path)
-        try:
-            gemini_file = genai.upload_file(temp_path)
-            return gemini_file
-        except Exception as e:
-            print(f"Erreur lors du téléchargement du fichier : {e}")
-            return None
-    return None
-
-# Template HTML (vous pouvez le déplacer dans un fichier template si besoin)
-template = """
-<!doctype html>
-<html>
-<head>
-  <meta charset="utf-8">
-  <title>Mariam AI</title>
-</head>
-<body>
-  <h1>Mariam AI!</h1>
-  <form method="POST" enctype="multipart/form-data">
-    <div>
-      <label for="prompt">Votre message :</label>
-      <input type="text" id="prompt" name="prompt" required>
-    </div>
-    <div>
-      <label for="file">Télécharger un fichier (image/document) :</label>
-      <input type="file" id="file" name="file" accept=".jpg,.jpeg,.png,.mp4,.mp3,.pdf,.txt">
-    </div>
-    <div>
-      <label for="web_search">Activer la recherche web :</label>
-      <input type="checkbox" id="web_search" name="web_search">
-    </div>
-    <button type="submit">Envoyer</button>
-  </form>
-  
-  <h2>Conversation</h2>
-  <div>
-    {% for message in chat_history %}
-      <p><strong>{{ message.role|capitalize }} :</strong> {{ message.text }}</p>
-    {% endfor %}
-  </div>
-</body>
-</html>
-"""
+UPLOAD_FOLDER = 'temp'
+ALLOWED_EXTENSIONS = {'jpg', 'jpeg', 'png', 'pdf', 'txt', 'mp3', 'mp4'}
 
-@app.route("/", methods=["GET", "POST"])
-def index():
-    # Initialisation de l'historique de chat en session s'il n'existe pas
-    if "chat_history" not in session:
-        init_chat()
-    
-    if request.method == "POST":
-        prompt = request.form.get("prompt")
-        web_search_enabled = request.form.get("web_search") == "on"
-        file = request.files.get("file")
-        
-        # Ajout du message utilisateur à l'historique de la session
-        session['chat_history'].append({"role": "user", "text": prompt})
-        
-        uploaded_gemini_file = process_uploaded_file(file)
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+@app.route('/')
+def home():
+    if 'chat_history' not in session:
+        session['chat_history'] = []
+    return render_template('index.html', messages=session['chat_history'])
+
+@app.route('/send_message', methods=['POST'])
+def send_message():
+    try:
+        data = request.json
+        message = data.get('message')
+        web_search_enabled = data.get('web_search', False)
         
-        # Si la recherche web est activée, effectuez-la et enrichissez la question
+        if not message:
+            return jsonify({'error': 'No message provided'}), 400
+
+        # Initialize chat if not in session
+        if 'chat' not in session:
+            session['chat'] = model.start_chat(history=[])
+
+        # Perform web search if enabled
+        web_results = None
         if web_search_enabled:
-            web_results = perform_web_search(prompt)
+            web_results = perform_web_search(message)
             if web_results:
                 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?"
+                message = f"""Question: {message}\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
+        response = session['chat'].send_message(message)
+        
+        # Update chat history
+        session['chat_history'].append({
+            'role': 'user',
+            'content': message
+        })
+        session['chat_history'].append({
+            'role': 'assistant',
+            'content': response.text
+        })
+        
+        return jsonify({
+            'response': response.text,
+            'web_results': web_results
+        })
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/upload', methods=['POST'])
+def upload_file():
+    if 'file' not in request.files:
+        return jsonify({'error': 'No file part'}), 400
+    
+    file = request.files['file']
+    if file.filename == '':
+        return jsonify({'error': 'No selected file'}), 400
+    
+    if file and allowed_file(file.filename):
+        filename = secure_filename(file.filename)
+        filepath = os.path.join(UPLOAD_FOLDER, filename)
+        file.save(filepath)
         
         try:
-            # Recrée le chat avec l'historique stocké en session
-            chat = model.start_chat(history=session.get("chat_history", []))
-            if uploaded_gemini_file:
-                response = chat.send_message([uploaded_gemini_file, "\n\n", prompt])
-            else:
-                response = chat.send_message(prompt)
-            answer = response.text
-            session['chat_history'].append({"role": "assistant", "text": answer})
+            gemini_file = genai.upload_file(filepath)
+            return jsonify({'success': True, 'filename': filename})
         except Exception as e:
-            error_message = f"Erreur lors de l'envoi du message : {e}"
-            session['chat_history'].append({"role": "assistant", "text": error_message})
-        
-        session.modified = True
-        return redirect(url_for("index"))
+            return jsonify({'error': str(e)}), 500
     
-    return render_template_string(template, chat_history=session.get("chat_history", []))
+    return jsonify({'error': 'Invalid file type'}), 400
+
+@app.route('/clear_chat', methods=['POST'])
+def clear_chat():
+    session.pop('chat_history', None)
+    session.pop('chat', None)
+    return jsonify({'success': True})
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     app.run(debug=True)