File size: 2,708 Bytes
2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 2171485 de01232 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
from flask import Flask, render_template, request, jsonify, session
from google import genai
import os
from datetime import datetime
import uuid
app = Flask(__name__)
app.secret_key = os.urandom(24) # Requis pour utiliser les sessions Flask
# Configuration de l'API Gemini (idéalement à déplacer dans un fichier de configuration)
# os.environ["GOOGLE_API_KEY"] = "VOTRE_GEMINI_API_KEY"
client = genai.Client(api_key=os.environ.get("GOOGLE_API_KEY", "GEMINI_API_KEY"))
@app.route('/')
def index():
# Créer un ID de session unique s'il n'existe pas déjà
if 'session_id' not in session:
session['session_id'] = str(uuid.uuid4())
return render_template('index.html')
@app.route('/chat', methods=['POST'])
def chat():
data = request.json
user_message = data.get('message', '')
# Récupérer ou créer une session de chat pour cet utilisateur
session_id = session.get('session_id')
chat_session = get_or_create_chat_session(session_id)
try:
# Envoyer le message à Gemini et obtenir la réponse
response = chat_session.send_message(user_message)
# Formater l'historique pour l'interface
history = []
for msg in chat_session.get_history():
history.append({
'role': msg.role,
'content': msg.parts[0].text,
'timestamp': datetime.now().strftime("%H:%M")
})
return jsonify({
'response': response.text,
'history': history
})
except Exception as e:
return jsonify({'error': str(e)}), 500
def get_or_create_chat_session(session_id):
"""
Récupère une session de chat existante ou en crée une nouvelle.
Dans une application réelle, vous pourriez stocker cela dans une base de données.
"""
# Utilisation d'un dictionnaire global pour stocker les sessions (pour cet exemple)
# Dans une application réelle, utilisez Redis ou une base de données
if not hasattr(app, 'chat_sessions'):
app.chat_sessions = {}
if session_id not in app.chat_sessions:
app.chat_sessions[session_id] = client.chats.create(model="gemini-2.0-flash")
return app.chat_sessions[session_id]
@app.route('/reset', methods=['POST'])
def reset_chat():
"""Réinitialise la session de chat actuelle"""
session_id = session.get('session_id')
if hasattr(app, 'chat_sessions') and session_id in app.chat_sessions:
# Créer une nouvelle session
app.chat_sessions[session_id] = client.chats.create(model="gemini-2.0-flash")
return jsonify({'status': 'success'})
if __name__ == '__main__':
app.run(debug=True) |