Update app.py

app.py

@@ -5,7 +5,11 @@ import os
 from PIL import Image
 import tempfile
 import json
-import logging # Optional: for better logging
+import logging
+import sqlite3
+from datetime import datetime
+import base64
+import io
 
 app = Flask(__name__)
 
@@ -39,6 +43,76 @@ SAFETY_SETTINGS = [
     )
 ]
 
+# --- Database Setup ---
+def init_db():
+    """Initialize the SQLite database"""
+    conn = sqlite3.connect('analysis_data.db')
+    cursor = conn.cursor()
+    
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS analyses (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
+            image_name TEXT,
+            image_data TEXT,
+            consignes TEXT,
+            use_deepthink BOOLEAN,
+            model_used TEXT,
+            tableau_output TEXT,
+            dissertation_output TEXT,
+            user_ip TEXT,
+            processing_time REAL
+        )
+    ''')
+    
+    conn.commit()
+    conn.close()
+
+def save_analysis_data(image_name, image_data, consignes, use_deepthink, model_used, tableau_output, dissertation_output, user_ip, processing_time):
+    """Save analysis data to database"""
+    conn = sqlite3.connect('analysis_data.db')
+    cursor = conn.cursor()
+    
+    cursor.execute('''
+        INSERT INTO analyses 
+        (image_name, image_data, consignes, use_deepthink, model_used, tableau_output, dissertation_output, user_ip, processing_time)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+    ''', (image_name, image_data, consignes, use_deepthink, model_used, tableau_output, dissertation_output, user_ip, processing_time))
+    
+    conn.commit()
+    conn.close()
+
+def get_all_analyses():
+    """Get all analyses from database"""
+    conn = sqlite3.connect('analysis_data.db')
+    cursor = conn.cursor()
+    
+    cursor.execute('''
+        SELECT id, timestamp, image_name, consignes, use_deepthink, model_used, 
+               tableau_output, dissertation_output, user_ip, processing_time
+        FROM analyses 
+        ORDER BY timestamp DESC
+    ''')
+    
+    analyses = cursor.fetchall()
+    conn.close()
+    
+    return analyses
+
+def get_analysis_image(analysis_id):
+    """Get image data for a specific analysis"""
+    conn = sqlite3.connect('analysis_data.db')
+    cursor = conn.cursor()
+    
+    cursor.execute('SELECT image_data FROM analyses WHERE id = ?', (analysis_id,))
+    result = cursor.fetchone()
+    conn.close()
+    
+    return result[0] if result else None
+
+# Initialize database on startup
+init_db()
+
 # --- Prompts (Keep them as they are) ---
 prompt_tableau = """
 Traite selon cette méthodologie :
@@ -120,20 +194,20 @@ Voici un exercice a trous présentant la rédaction.. référence textuelle= rep
 
 EXERCICE À TROUS
 
-Le thème de ... (thème du texte) a souvent fait l'objet de nombreuses préoccupations dans le monde littéraire (mais pas que). C’est dans ce cadre que s'inscrit l'extrait ... (titre du texte) qui fait l'objet de notre étude, de l'écrivain... (nom de l'auteur), tiré de son œuvre ... (préciser le genre littéraire de l'œuvre) ... (maison d'édition), en ... (date de publication) à la (aux) page (s) …. Dans ce texte (type du texte) à ton ... (tonalité du texte/facultative), structure, nous verrons en premier lieu,...(axe 1) et en second lieu, ...(axe 2).
+Le thème de ... (thème du texte) a souvent fait l'objet de nombreuses préoccupations dans le monde littéraire (mais pas que). C'est dans ce cadre que s'inscrit l'extrait ... (titre du texte) qui fait l'objet de notre étude, de l'écrivain... (nom de l'auteur), tiré de son œuvre ... (préciser le genre littéraire de l'œuvre) ... (maison d'édition), en ... (date de publication) à la (aux) page (s) …. Dans ce texte (type du texte) à ton ... (tonalité du texte/facultative), structure, nous verrons en premier lieu,...(axe 1) et en second lieu, ...(axe 2).
 
-Dans son extrait (poème), l’auteur met en relief ... (axe 1) à travers ... (sous-axe 1) et ... (sous-axe 2).
+Dans son extrait (poème), l'auteur met en relief ... (axe 1) à travers ... (sous-axe 1) et ... (sous-axe 2).
 
-S'agissant de ... (sous-axe 1), l’écrivain utilise ... (outil d'analyse 1 + référence textuelle) pour montrer ... (interprétation). Aussi (de plus), par l'usage de... (outil d'analyse 2 + référence textuelle), l'écrivain ... (interprétation). Mieux encore, ... (outil d'analyse 3 + référence textuelle) nous donne également la possibilité d’appréhender ... (sous-axe 2).
+S'agissant de ... (sous-axe 1), l'écrivain utilise ... (outil d'analyse 1 + référence textuelle) pour montrer ... (interprétation). Aussi (de plus), par l'usage de... (outil d'analyse 2 + référence textuelle), l'écrivain ... (interprétation). Mieux encore, ... (outil d'analyse 3 + référence textuelle) nous donne également la possibilité d'appréhender ... (sous-axe 2).
 De plus, l'homme de lettres emploie ... (outil d'analyse 1 + référence textuelle) pour ... (interprétation). Il se sert aussi de ...(outil d'analyse 2 + référence textuelle) afin de ... (interprétation). Pour continuer sa (description, représentation), le ...(nationalité de l'auteur) se manque pas de faire recours à ... (outil d'analyse 3 référence textuelle) Ici, il s'agit pour l'auteur autour de ... (sous-axe 1) et ... (sous-axe 2).
 
 Après avoir démontré ... (axe 1), voyons à présent ... (axe 2).
 
-En second lieu, le poète (l’écrivain ou l’homme de lettres) met en exergue ... (axe 2) en s’appuyant d’une part, sur... (sous-axe 1) et d’autre part, sur... (sous-axe 2). En ce qui concerne ... (sous-axe 1), l'homme de lettres met d'abord en évidence l'aspect (le caractère) ... (interprétation) comme en témoigne l'emploi (l'usage de) ... (outil d’analyse 1 + référence textuelle). Ensuite, ... (outil d'analyse 2 + référence textuelle) dévoile que... (interprétation) Enfin, ... (outil d'analyse 3 + référence textuelle) suggère que… (interprétation) . ... (axe 2) se révèle grâce à ….
-En parlant de ... (sous-axe 2), l'auteur met l'accent en premier sur… (interprétation), comme nous pouvons le voir avec la récurrence de (du/des) ... (rappel du sous-axe 2), le poète (l'auteur) souligne ... (interprétation) toujours dans le même sens de ... (rappel du sous-axe 2) . Il use de ... (outil d’analyse 2 + référence textuelle). Dès lors, on peut déduire que ...(interprétation) utilise ... (outil d’analyse 3 + référence textuelle).
+En second lieu, le poète (l'écrivain ou l'homme de lettres) met en exergue ... (axe 2) en s'appuyant d'une part, sur... (sous-axe 1) et d'autre part, sur... (sous-axe 2). En ce qui concerne ... (sous-axe 1), l'homme de lettres met d'abord en évidence l'aspect (le caractère) ... (interprétation) comme en témoigne l'emploi (l'usage de) ... (outil d'analyse 1 + référence textuelle). Ensuite, ... (outil d'analyse 2 + référence textuelle) dévoile que... (interprétation) Enfin, ... (outil d'analyse 3 + référence textuelle) suggère que… (interprétation) . ... (axe 2) se révèle grâce à ….
+En parlant de ... (sous-axe 2), l'auteur met l'accent en premier sur… (interprétation), comme nous pouvons le voir avec la récurrence de (du/des) ... (rappel du sous-axe 2), le poète (l'auteur) souligne ... (interprétation) toujours dans le même sens de ... (rappel du sous-axe 2) . Il use de ... (outil d'analyse 2 + référence textuelle). Dès lors, on peut déduire que ...(interprétation) utilise ... (outil d'analyse 3 + référence textuelle).
 Ainsi, ... (axe 2) est lié (e) à ... (sous-axe 1) et à ... (sous-axe 2).
 
-Somme toute, ... (titre du texte) organise son sens autour de … (axe 1) et de ... (axe 2). De ces deux centres d’intérêt découlent respectivement, d’une part, … (sous-axe 1 de l'axe 1) et ... (sous-axe 2 de l'axe 1) et, d’autre part, … (sous-axe 1 de l'axe 2) et … (sous-axe 2 de l'axe 2). À travers ce texte, ...(nom de l'auteur) nous ... (opinion personnelle). Une telle optique est perceptible dans la logique de... (nom de l'auteur nous permettant de faire un rapprochement thématique), dans son œuvre ...(titre de l’œuvre), dans lequel il aborde… (bref résumé de l'œuvre en question qui peut être facultatif).
+Somme toute, ... (titre du texte) organise son sens autour de … (axe 1) et de ... (axe 2). De ces deux centres d'intérêt découlent respectivement, d'une part, … (sous-axe 1 de l'axe 1) et ... (sous-axe 2 de l'axe 1) et, d'autre part, … (sous-axe 1 de l'axe 2) et … (sous-axe 2 de l'axe 2). À travers ce texte, ...(nom de l'auteur) nous ... (opinion personnelle). Une telle optique est perceptible dans la logique de... (nom de l'auteur nous permettant de faire un rapprochement thématique), dans son œuvre ...(titre de l'œuvre), dans lequel il aborde… (bref résumé de l'œuvre en question qui peut être facultatif).
 """
 # --- End Prompts ---
 
@@ -205,10 +279,49 @@ def index():
     # Assuming your main HTML file is index.html
     return render_template('index.html')
 
-# Removed /free route as it wasn't mentioned in the final HTML
-# @app.route('/free')
-# def free():
-#     return render_template('free.html')
+@app.route('/gestion')
+def gestion():
+    """Page de gestion pour afficher toutes les analyses"""
+    return render_template('gestion.html')
+
+@app.route('/api/analyses')
+def get_analyses():
+    """API endpoint to get all analyses data"""
+    try:
+        analyses = get_all_analyses()
+        analyses_data = []
+        
+        for analysis in analyses:
+            analyses_data.append({
+                'id': analysis[0],
+                'timestamp': analysis[1],
+                'image_name': analysis[2],
+                'consignes': analysis[3],
+                'use_deepthink': analysis[4],
+                'model_used': analysis[5],
+                'tableau_output': analysis[6],
+                'dissertation_output': analysis[7],
+                'user_ip': analysis[8],
+                'processing_time': analysis[9]
+            })
+        
+        return jsonify(analyses_data)
+    except Exception as e:
+        logging.error(f"Error fetching analyses: {e}")
+        return jsonify({'error': 'Erreur lors de la récupération des données'}), 500
+
+@app.route('/api/analysis/<int:analysis_id>/image')
+def get_analysis_image_endpoint(analysis_id):
+    """API endpoint to get image for a specific analysis"""
+    try:
+        image_data = get_analysis_image(analysis_id)
+        if image_data:
+            return Response(base64.b64decode(image_data), mimetype='image/jpeg')
+        else:
+            return jsonify({'error': 'Image non trouvée'}), 404
+    except Exception as e:
+        logging.error(f"Error fetching image: {e}")
+        return jsonify({'error': 'Erreur lors de la récupération de l\'image'}), 500
 
 @app.route('/analyze', methods=['POST'])
 def analyze():
@@ -229,6 +342,12 @@ def analyze():
         model_id = MODEL_ID_STANDARD
         logging.info("Standard analysis requested, using standard model.")
 
+    # Get user IP
+    user_ip = request.remote_addr
+    
+    # Track processing time
+    start_time = datetime.now()
+
     # Use a temporary file to handle the image upload
     temp_file = None # Initialize to None
     try:
@@ -242,6 +361,12 @@ def analyze():
                 image.verify() # Verify image header
                 # Re-open after verify
                 image = Image.open(temp_file.name)
+                
+                # Convert image to base64 for storage
+                img_buffer = io.BytesIO()
+                image.save(img_buffer, format='JPEG')
+                image_base64 = base64.b64encode(img_buffer.getvalue()).decode('utf-8')
+                
             except (IOError, SyntaxError) as e:
                 logging.error(f"Invalid image file uploaded: {e}")
                 return jsonify({'error': f'Invalid or corrupted image file: {e}'}), 400
@@ -254,6 +379,8 @@ def analyze():
             def generate():
                 temp_file_path = temp_file.name # Store path for finally block
                 full_tableau_content = "" # Accumulate full table content for dissertation
+                full_dissertation_content = "" # Accumulate full dissertation content
+                
                 try:
                     logging.info("Starting table generation stream...")
                     # Phase 1: Génération du tableau, passing the selected model_id
@@ -286,6 +413,7 @@ def analyze():
                                      yield chunk_json # Forward the error
                                      # Decide if you want to return here or let it finish
                                 elif chunk_data.get("type") == "dissertation":
+                                     full_dissertation_content += chunk_data.get("chunk", "")
                                      yield chunk_json # Stream chunk to client
                             except json.JSONDecodeError:
                                 logging.error(f"Received invalid JSON from dissertation stream: {chunk_json}")
@@ -297,6 +425,27 @@ def analyze():
 
                     logging.info("Dissertation generation stream finished.")
 
+                    # Save analysis data to database
+                    try:
+                        end_time = datetime.now()
+                        processing_time = (end_time - start_time).total_seconds()
+                        
+                        save_analysis_data(
+                            image_name=image_file.filename,
+                            image_data=image_base64,
+                            consignes=consignes,
+                            use_deepthink=use_deepthink,
+                            model_used=model_id,
+                            tableau_output=full_tableau_content,
+                            dissertation_output=full_dissertation_content,
+                            user_ip=user_ip,
+                            processing_time=processing_time
+                        )
+                        logging.info(f"Analysis data saved successfully for image: {image_file.filename}")
+                    except Exception as save_error:
+                        logging.error(f"Error saving analysis data: {save_error}")
+                        # Don't yield error to client as analysis was successful
+
                 except Exception as e:
                     logging.error(f"Error during streaming generation: {e}", exc_info=True)
                     # Yield a JSON error message for the client