Update app.py

app.py

@@ -7,11 +7,12 @@ import base64
 from werkzeug.utils import secure_filename
 import mimetypes
 from dotenv import load_dotenv
-
+from datetime import datetime
 
 app = Flask(__name__)
 app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB max file size
 load_dotenv()
+
 # Configuration du client Gemini
 API_KEY = os.getenv("GOOGLE_API_KEY")
 SYSTEM_INSTRUCTION = "Tu es un assistant intelligent et amical nommé Mariam. Tu assistes les utilisateurs au mieux de tes capacités. Tu as été créé par Aenir."
@@ -34,13 +35,75 @@ DEFAULT_TOOLS = [
     types.Tool(url_context=types.UrlContext())
 ]
 
-# Stockage des conversations (en production, utilisez une base de données)
+# Stockage des conversations avec métadonnées (en production, utilisez une base de données)
 conversations = {}
+conversation_metadata = {}
+
+def add_message_to_history(conversation_id, role, content, has_file=False):
+    """Ajoute un message à l'historique de la conversation"""
+    if conversation_id not in conversation_metadata:
+        conversation_metadata[conversation_id] = {
+            'id': conversation_id,
+            'created_at': datetime.now().isoformat(),
+            'last_activity': datetime.now().isoformat(),
+            'messages': [],
+            'status': 'active'
+        }
+    
+    conversation_metadata[conversation_id]['messages'].append({
+        'role': role,
+        'content': content,
+        'timestamp': datetime.now().isoformat(),
+        'hasFile': has_file
+    })
+    conversation_metadata[conversation_id]['last_activity'] = datetime.now().isoformat()
 
 @app.route('/')
 def index():
     return render_template('index.html')
 
+@app.route('/admin')
+def admin():
+    """Page d'administration"""
+    return render_template('admin.html')
+
+@app.route('/admin/conversations')
+def get_conversations():
+    """API pour récupérer les conversations pour l'admin"""
+    try:
+        # Calculer les statistiques
+        total_conversations = len(conversation_metadata)
+        total_messages = sum(len(conv['messages']) for conv in conversation_metadata.values())
+        active_conversations = sum(1 for conv in conversation_metadata.values() if conv.get('status') == 'active')
+        conversations_with_files = sum(1 for conv in conversation_metadata.values() 
+                                     if any(msg.get('hasFile') for msg in conv['messages']))
+        
+        # Préparer les données des conversations
+        conversations_data = []
+        for conv_id, conv_data in conversation_metadata.items():
+            conversations_data.append({
+                'id': conv_id,
+                'createdAt': conv_data.get('created_at'),
+                'lastActivity': conv_data.get('last_activity'),
+                'status': conv_data.get('status', 'active'),
+                'messages': conv_data.get('messages', [])
+            })
+        
+        # Trier par dernière activité (plus récent en premier)
+        conversations_data.sort(key=lambda x: x.get('lastActivity', ''), reverse=True)
+        
+        return jsonify({
+            'conversations': conversations_data,
+            'stats': {
+                'total': total_conversations,
+                'totalMessages': total_messages,
+                'active': active_conversations,
+                'withFiles': conversations_with_files
+            }
+        })
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
 @app.route('/chat', methods=['POST'])
 def chat():
     try:
@@ -49,10 +112,13 @@ def chat():
         thinking_enabled = data.get('thinking_enabled', True)
         conversation_id = data.get('conversation_id', 'default')
         
+        # Ajouter le message de l'utilisateur à l'historique
+        add_message_to_history(conversation_id, 'user', message)
+        
         # Configuration du thinking
         config_dict = DEFAULT_CONFIG.copy()
         config_dict["system_instruction"] = SYSTEM_INSTRUCTION
-        # <<< FIN DE LA MODIFICATION >>>
+        
         if thinking_enabled:
             config_dict["thinking_config"] = types.ThinkingConfig(
                 thinking_budget=-1,  # Dynamic thinking
@@ -66,7 +132,6 @@ def chat():
         if conversation_id not in conversations:
             conversations[conversation_id] = client.chats.create(
                 model=MODEL,
-            
                 config=generation_config
             )
         
@@ -93,6 +158,10 @@ def chat():
                                 full_response += part.text
                                 yield f"data: {json.dumps({'type': 'text', 'content': part.text})}\n\n"
                 
+                # Ajouter la réponse de l'assistant à l'historique
+                if full_response:
+                    add_message_to_history(conversation_id, 'assistant', full_response)
+                
                 # Signal de fin
                 yield f"data: {json.dumps({'type': 'end'})}\n\n"
                 
@@ -140,6 +209,12 @@ def chat_with_file():
         thinking_enabled = data.get('thinking_enabled', True)
         conversation_id = data.get('conversation_id', 'default')
         
+        # Ajouter le message de l'utilisateur à l'historique (avec indication de fichier)
+        display_message = message if message else 'Analyse ce fichier'
+        if file_data:
+            display_message += f" [Fichier: {file_data.get('filename', 'inconnu')}]"
+        add_message_to_history(conversation_id, 'user', display_message, has_file=True)
+        
         # Configuration du thinking
         config_dict = DEFAULT_CONFIG.copy()
         if thinking_enabled:
@@ -150,14 +225,12 @@ def chat_with_file():
         
         config_dict["tools"] = DEFAULT_TOOLS
         config_dict["system_instruction"] = SYSTEM_INSTRUCTION
-        # <<< FIN DE LA MODIFICATION >>>
         generation_config = types.GenerateContentConfig(**config_dict)
         
         # Gestion de la conversation
         if conversation_id not in conversations:
             conversations[conversation_id] = client.chats.create(
                 model=MODEL,
-                
                 config=generation_config
             )
         
@@ -182,14 +255,21 @@ def chat_with_file():
                     config=generation_config
                 )
                 
+                full_response = ""
+                
                 for chunk in response_stream:
                     for part in chunk.candidates[0].content.parts:
                         if part.text:
                             if part.thought and thinking_enabled:
                                 yield f"data: {json.dumps({'type': 'thought', 'content': part.text})}\n\n"
                             else:
+                                full_response += part.text
                                 yield f"data: {json.dumps({'type': 'text', 'content': part.text})}\n\n"
                 
+                # Ajouter la réponse de l'assistant à l'historique
+                if full_response:
+                    add_message_to_history(conversation_id, 'assistant', full_response)
+                
                 yield f"data: {json.dumps({'type': 'end'})}\n\n"
                 
             except Exception as e:
@@ -209,8 +289,85 @@ def reset_conversation():
         if conversation_id in conversations:
             del conversations[conversation_id]
         
+        # Marquer la conversation comme terminée dans les métadonnées
+        if conversation_id in conversation_metadata:
+            conversation_metadata[conversation_id]['status'] = 'reset'
+            conversation_metadata[conversation_id]['last_activity'] = datetime.now().isoformat()
+        
+        return jsonify({'success': True})
+        
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/admin/conversations/<conversation_id>', methods=['DELETE'])
+def delete_conversation(conversation_id):
+    """Supprimer une conversation (pour l'admin)"""
+    try:
+        if conversation_id in conversations:
+            del conversations[conversation_id]
+        
+        if conversation_id in conversation_metadata:
+            del conversation_metadata[conversation_id]
+        
         return jsonify({'success': True})
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/admin/conversations/<conversation_id>/export')
+def export_conversation(conversation_id):
+    """Exporter une conversation en JSON"""
+    try:
+        if conversation_id not in conversation_metadata:
+            return jsonify({'error': 'Conversation non trouvée'}), 404
+        
+        conversation_data = conversation_metadata[conversation_id]
         
+        return jsonify({
+            'conversation_id': conversation_id,
+            'export_date': datetime.now().isoformat(),
+            'data': conversation_data
+        })
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/admin/stats')
+def get_admin_stats():
+    """Statistiques détaillées pour l'admin"""
+    try:
+        # Statistiques générales
+        total_conversations = len(conversation_metadata)
+        total_messages = sum(len(conv['messages']) for conv in conversation_metadata.values())
+        
+        # Statistiques par statut
+        status_stats = {}
+        for conv in conversation_metadata.values():
+            status = conv.get('status', 'active')
+            status_stats[status] = status_stats.get(status, 0) + 1
+        
+        # Conversations avec fichiers
+        conversations_with_files = sum(1 for conv in conversation_metadata.values() 
+                                     if any(msg.get('hasFile') for msg in conv['messages']))
+        
+        # Activité par jour (derniers 7 jours)
+        from collections import defaultdict
+        daily_activity = defaultdict(int)
+        
+        for conv in conversation_metadata.values():
+            for message in conv['messages']:
+                if message.get('timestamp'):
+                    try:
+                        date = datetime.fromisoformat(message['timestamp']).date()
+                        daily_activity[date.isoformat()] += 1
+                    except:
+                        continue
+        
+        return jsonify({
+            'total_conversations': total_conversations,
+            'total_messages': total_messages,
+            'status_distribution': status_stats,
+            'conversations_with_files': conversations_with_files,
+            'daily_activity': dict(daily_activity)
+        })
     except Exception as e:
         return jsonify({'error': str(e)}), 500