Update app.py

app.py

@@ -1,50 +1,44 @@
 from flask import Flask, request, jsonify, render_template, send_file
 from flask_cors import CORS
-from PIL import Image
+from PIL import Image, ImageDraw
 import io
 import json
 import os
 import uuid
 import google.generativeai as genai
-import numpy as np
 
+# Configuration de l'API Gemini
 generation_config = {
-    "temperature": 1,
-    "max_output_tokens": 8192,
+  "temperature": 1,
+  "max_output_tokens": 8192,
 }
 
 safety_settings = [
-    {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "threshold": "BLOCK_NONE"
-    },
-    {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "threshold": "BLOCK_NONE"
-    },
-    {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "threshold": "BLOCK_NONE"
-    },
-    {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "threshold": "BLOCK_NONE"
-    },
+  {
+    "category": "HARM_CATEGORY_HARASSMENT",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_HATE_SPEECH",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+    "threshold": "BLOCK_NONE"
+  },
 ]
 
-GOOGLE_API_KEY = os.environ.get("TOKEN")
+GOOGLE_API_KEY = os.environ.get("TOKEN") # Assurez vous que la variable d'environnement TOKEN est bien définie
 
-# Configuration de l'API Gemini
 genai.configure(api_key=GOOGLE_API_KEY)
 
 app = Flask(__name__)
 CORS(app)
 
-@app.route('/', methods=['GET'])
-def svt():
-    """Renders the SVT page."""
-    return render_template("svt.html")
-
 # 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\"."
 
@@ -68,7 +62,10 @@ UPLOAD_FOLDER = 'uploads'
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
-
+@app.route('/', methods=['GET'])
+def svt():
+    """Renders the SVT page."""
+    return render_template("templates_svt (3).html")
 
 @app.route('/analyze', methods=['POST'])
 def analyze_image():
@@ -89,8 +86,8 @@ def analyze_image():
             # 1. Détection d'objets avec Gemini
             model = genai.GenerativeModel("gemini-2.0-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
+              "mime_type": "image/jpeg",
+              "data": open(filename, "rb").read()
             }
             response = model.generate_content([DETECTION_PROMPT, image_part])
 
@@ -162,6 +159,7 @@ def analyze_image():
         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))