Update app.py

app.py

@@ -243,12 +243,14 @@ model = genai.GenerativeModel(
     safety_settings=safety_settings
 )
 
-def generate_table(image):
-    """Génère le tableau d'analyse à partir de l'image"""
-    response = model.generate_content([prompt_tableau, image])
-    
+def generate_table(image, consignes=""):
+    prompt = prompt_tableau
+    if consignes:
+         prompt += "\n" + consignes
+    response = model.generate_content([prompt, image])
     return response.text
 
+
 def generate_dissertation(tableau):
     """Génère la dissertation basée sur le tableau"""
     prompt = f"""
@@ -297,27 +299,20 @@ def analyzed():
         os.unlink(temp_file.name)
 
 
-
-
 @app.route('/analyze', methods=['POST'])
 def analyze():
     if 'image' not in request.files:
         return jsonify({'error': 'No image uploaded'}), 400
-    
-    
+
+    consignes = request.form.get("consignes", "")
+
     image_file = request.files['image']
-    
-    # Sauvegarder temporairement l'image
     with tempfile.NamedTemporaryFile(delete=False) as temp_file:
         image_file.save(temp_file.name)
         image = Image.open(temp_file.name)
     
-    
     try:
-        # Première génération : le tableau
-        tableau = generate_table(image)
-        
-        # Deuxième génération : la dissertation
+        tableau = generate_table(image, consignes)
         dissertation = generate_dissertation(tableau)
         
         return jsonify({
@@ -327,5 +322,4 @@ def analyze():
     except Exception as e:
         return jsonify({'error': "Erreur."}), 500
     finally:
-        # Nettoyer le fichier temporaire
-        os.unlink(temp_file.name)
+        os.unlink(temp_file.name)