Update app.py

app.py

@@ -1,12 +1,13 @@
-from flask import Flask, request, render_template, jsonify
-import PIL.Image
-import google.generativeai as genai
+from flask import Flask, request, jsonify, send_file
+from flask_cors import CORS
+from PIL import Image, ImageDraw
+import io
+import json
 import os
-from tempfile import NamedTemporaryFile
-import tempfile
-app = Flask(__name__)
+import uuid
+import google.generativeai as genai
+
 
-# Configuration de Gemini
 generation_config = {
   "temperature": 1,
   "max_output_tokens": 8192,
@@ -33,128 +34,101 @@ safety_settings = [
 
 GOOGLE_API_KEY = os.environ.get("TOKEN")
 
-genai.configure(api_key=GOOGLE_API_KEY)
-
-@app.route('/', methods=['GET'])
-def svt():
-    """Renders the SVT page."""
-    return render_template("svt.html")
-
-    
-methodologie_svt = {
-    
-    "Restitution organisée des connaissances": """
-**Restitution organisée des connaissances (ROC)**
-
-**Objectif:** Exposer, dans un texte structuré, scientifiquement et grammaticalement correct, illustré si nécessaire, des connaissances sur un point du programme.
-
-**Structure de la ROC:**
-
-*   **Introduction:**
-    *   Contexte: Synthèse des savoirs (prérequis) nécessaires pour aborder le thème.
-    *   Problème: Reformulation de la consigne sous forme interrogative, découlant logiquement du contexte.
-    *   Plan: Annonce des parties du développement.
-*   **Développement:**
-    *   Au moins deux paragraphes séparés par une ligne, débutant par un titre souligné.
-    *   Titres: Reprise des parties annoncées dans le plan.
-    *   Contenu: Solution du problème, articulation logique des paragraphes.
-    *   Schéma: Si la consigne l'exige.
-*   **Conclusion:**
-    *   Réponse logique au problème posé dans l'introduction.
-    *   Intégration des aspects développés.
-    *   Correspondance à la thématique de l'exercice.
-
-**Conseils:**
-
-*   L'exercice ne comporte pas de documents.
-*   Le sujet comporte un thème, un contexte et une consigne.
-*   L'exercice est pondéré sur 7 à 8 points.
-""",
-    "Exploitation du document": """
-
-    
-**Exploitation de documents (ED)**
-
-**Objectif:** Trouver le lien entre les informations présentées par un des documents et les connaissances d'un segment de connaissances (partie du programme) en vue de la résolution d'un problème scientifique.
-
-**Structure de l'ED:**
-
-*   **Introduction** (peut être écrite au brouillon) : Problème (reformulation de la consigne)
-*   **Pour chaque document :**
-    *   **Présentation (CP1):** Type de document + objet d'étude (cf. titre).
-    *   **Analyse (CP2):** Description du fait expérimental (comparaison de courbes, résultats d'expérience) et/ou présentation du fait d'observation (changement de coloration, % de phénotypes).
-    *   **Information saisie (CP3):** Conclusion partielle, fait à interpréter.
-    *   **Mise en relation (CP4):** Interprétation de l'information saisie en utilisant les connaissances acquises, signification permettant la résolution du problème.
-*   **Synthèse des mises en relation (CP5)** : Lien pertinent et cohérent entre toutes les significations (mise en relation) des informations utiles pour résoudre le problème (répondre à la consigne). Cette partie est séparée du reste par deux lignes.
-
-**Conseils:**
-
-*   L'exercice comporte un thème, un contexte, une consigne, un ou deux documents, et une pondération (7-8 points).
-*   Le contexte établit le lien entre le thème et les documents.
-*   La consigne guide l'élève dans les différentes tâches.
-*   Les documents doivent comporter un titre et une source, être pertinents, lisibles, et suivre l'ordre chronologique de la résolution.
-*   Ne pas paraphraser, copier ou faire une description intégrale dans l'analyse (CP2).
-*   La tâche 4 (CP4) est spécifique à cet exercice.
-*   Mentionner le document traité (ex: Document 1).
-
-
-réponse attendu uniquement en langue française.
-""",
-    "Synthèse": """
-**Élaboration d'une synthèse (ES)**
-
-**Objectif:** Dégager des informations pertinentes d'un ensemble de documents en vue de résoudre un problème scientifique. La résolution du problème ne fait pas appel directement aux connaissances du cours.
-
-**Structure de l'ES:**
-
-*   **Introduction** (peut être écrite au brouillon) : Problème (reformulation de la consigne)
-*   **Pour chaque document :**
-    *   **Présentation (CP1):** Type de document + objet d'étude (cf. titre).
-    *   **Analyse (CP2):** Description du fait expérimental (comparaison de courbes, résultats d'expérience) et/ou présentation du fait d'observation (changement de coloration, % de phénotypes).
-    *   **Conclusion partielle (CP3):** Synthèse de l'analyse, élément de réponse au problème.
-*   **Conclusion générale (CP4):** Récapitulation des conclusions partielles, réponse à la consigne.
-
-**Conseils:**
-
-*   L'exercice comporte un thème, un contexte, une consigne, deux ou trois documents, et une pondération (5 points).
-*   Le contexte établit le lien entre le thème et les documents.
-*   La consigne guide l'élève dans les différentes tâches.
-*   Les documents doivent comporter un titre et une source, être pertinents, lisibles, et suivre l'ordre chronologique de la résolution.
-*   Ne pas paraphraser, copier ou faire une description intégrale dans l'analyse (CP2).
-*   La tâche 3 (CP3) est spécifique à cet exercice.
-*   Mentionner le document traité (ex: Document 1).
-*   Toutes les informations nécessaires sont dans les documents.
-"""
-}
 
+# Configuration de l'API Gemini
+genai.configure(api_key=GOOGLE_API_KEY)  # Remplacez par votre clé API
 
-@app.route('/svt_submit', methods=['POST'])
-def svt_submit():
-    """Handles the submission of SVT exercises."""
-    option = request.form.get('option')
-    images = request.files.getlist('images')
+app = Flask(__name__)
+CORS(app)
+
+# Prompt pour la détection d'objets
+DETECTION_PROMPT = "Detect items, with no more than 20 items. Output a json list where each entry contains the 2D bounding box in \"box_2d\" and a text label in \"label\"."
+
+# Prompt pour la description d'image satellite militaire
+DESCRIPTION_PROMPT = """
+Décrivez en détail cette image satellite militaire. Soyez précis et exhaustif dans votre analyse. 
+Identifiez les éléments clés tels que :
+- **Infrastructures** : Bâtiments, routes, ponts, aéroports, ports, etc.
+- **Véhicules** : Chars, avions, navires, véhicules de transport de troupes, etc.
+- **Unités militaires** : Formations de troupes, positions d'artillerie, camps, etc.
+- **Défenses** : Bunkers, tranchées, barbelés, etc.
+- **Éléments géographiques** : Relief, végétation, cours d'eau, etc.
+- **Activités** : Mouvements de troupes, entraînements, constructions, etc.
+- **Anomalies** : Tout ce qui semble inhabituel ou suspect.
+
+Fournissez une évaluation globale de la situation et des implications stratégiques possibles.
+"""
 
-    content = [f"J'aimerais que tu traites entièrement cet exercice en respectant scrupuleusement la méthodologie d'SVT suivante :\n\n{methodologie_svt[option]}\n\nLe type d'exercice selon la méthodologie est : {option}. Voici les images de l'exercice:"]
-    temp_files = []
+# Dossier pour enregistrer temporairement les images
+UPLOAD_FOLDER = 'uploads'
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
+@app.route('/analyze', methods=['POST'])
+def analyze_image():
     try:
-        for image in images:
-            if image:
-                with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(image.filename)[1]) as temp_file:
-                    image.save(temp_file.name)
-                    temp_files.append(temp_file.name)
-                    content.append(PIL.Image.open(temp_file.name))
-
-        model = genai.GenerativeModel(model_name="models/gemini-2.0-flash-exp", safety_settings=safety_settings)
-        response = model.generate_content(content, request_options={"timeout": 600})
+        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:
+            # Générer un nom de fichier unique pour éviter les conflits
+            unique_filename = str(uuid.uuid4()) + os.path.splitext(file.filename)[1]
+            filename = os.path.join(app.config['UPLOAD_FOLDER'], unique_filename)
+            file.save(filename)
+
+            # 1. Détection d'objets avec Gemini
+            model = genai.GenerativeModel("gemini-1.5-flash-exp",safety_settings=safety_settings,generation_config=generation_config)
+            image_part = {
+              "mime_type": "image/jpeg",  # Assurez-vous que cela corresponde au type de votre image
+              "data": open(filename, "rb").read() # Le fichier est lu ici
+            }
+            response = model.generate_content([DETECTION_PROMPT, image_part])
 
-        return jsonify({"response": response.text})
+            try:
+                detection_results = json.loads(response.text)
+            except json.JSONDecodeError:
+                print(f"Erreur de décodage JSON : {response.text}")
+                return jsonify({'error': 'Erreur lors de la détection des objets (JSON invalide)'}), 500
+
+            # 2. Dessiner les boîtes englobantes
+            image = Image.open(filename)
+            draw = ImageDraw.Draw(image)
+
+            for item in detection_results:
+                box = item['box_2d']
+                label = item['label']
+                draw.rectangle(box, outline=(255, 0, 0), width=2)
+                text_position = (box[0], box[1] - 10)
+                draw.text(text_position, label, fill=(255, 255, 255))
+
+            # Enregistrer l'image avec les boîtes
+            output_filename = os.path.join(app.config['UPLOAD_FOLDER'], 'output_' + unique_filename)
+            image.save(output_filename)
+
+            # 3. Générer la description
+            
+            response = model.generate_content([DESCRIPTION_PROMPT, image_part])
+            description = response.text
+
+            # 4. Renvoyer les résultats
+            return jsonify({
+                'image_path': '/uploads/' + 'output_' + unique_filename, # Chemin relatif vers l'image
+                'description': description,
+                'detected_objects': detection_results
+            })
 
     except Exception as e:
-        return jsonify({"error": str(e)}), 500
-    finally:
-        for temp_file in temp_files:
-            try:
-                os.unlink(temp_file)
-            except Exception as e:
-                print(f"Error deleting temporary file {temp_file}: {e}")
+        print(f"Une erreur s'est produite : {e}")
+        return jsonify({'error': f'Erreur lors du traitement de l\'image : {e}'}), 500
+
+# Servir les fichiers statiques depuis le dossier 'uploads'
+@app.route('/uploads/<filename>')
+def uploaded_file(filename):
+    return send_file(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+
+if __name__ == '__main__':
+    app.run(debug=True)