Update app.py

app.py

@@ -618,7 +618,7 @@ def send_document_to_telegram(content_or_path, filename="reponse.txt", caption="
         return False
 
 # --- Background Image Processing ---
-def process_image_background(task_id, image_data):
+def process_image_background(task_id, image_data, resolution_style='colorful'):
     """Traite l'image, génère LaTeX, convertit en PDF (si possible), et envoie via Telegram."""
     pdf_file_to_clean = None
     try:
@@ -629,23 +629,26 @@ def process_image_background(task_id, image_data):
 
         img = Image.open(io.BytesIO(image_data))
         buffered = io.BytesIO()
-        img.save(buffered, format="PNG") # Gemini prefers PNG or JPEG
+        img.save(buffered, format="PNG")
         img_base64_str = base64.b64encode(buffered.getvalue()).decode()
         
         full_latex_response = ""
         
         try:
             task_results[task_id]['status'] = 'generating_latex'
-            print(f"Task {task_id}: Génération LaTeX par Gemini...")
+            print(f"Task {task_id}: Génération LaTeX par Gemini (style: {resolution_style})...")
+            
+            # Choisir le prompt selon le style
+            if resolution_style == 'light':
+                prompt_to_use = ppmqth_light  # Vous devrez créer cette variable
+            else:  # colorful par défaut
+                prompt_to_use = ppmqth
             
-            # User's original model: "gemini-2.5-pro-exp-03-25"
-            # Using "gemini-1.5-pro-latest" as a robust alternative. User can change if needed.
-            # The user's original Gemini call structure:
             gemini_response = client.models.generate_content(
                 model="gemini-2.5-flash-preview-04-17", 
                 contents=[
                     {'inline_data': {'mime_type': 'image/png', 'data': img_base64_str}},
-                    ppmqth 
+                    prompt_to_use 
                 ],)
                 
 
@@ -740,7 +743,7 @@ def index():
 def free():
     return render_template('free.html')
     
-    
+
 @app.route('/solve', methods=['POST'])
 def solve():
     try:
@@ -751,33 +754,41 @@ def solve():
         if image_file.filename == '':
             return jsonify({'error': 'Aucun fichier sélectionné'}), 400
 
+        # Récupérer le style de résolution
+        resolution_style = request.form.get('style', 'colorful')  # 'colorful' par défaut
+        print(f"Style de résolution sélectionné: {resolution_style}")
+
         image_data = image_file.read()
         
-        # Envoyer l'image à Telegram (confirmation d'upload)
-        send_to_telegram(image_data, f"Nouvelle image pour résolution (Tâche à venir)")
+        # Envoyer l'image à Telegram avec info du style
+        send_to_telegram(image_data, f"Nouvelle image pour résolution (Style: {resolution_style})")
         
         task_id = str(uuid.uuid4())
         task_results[task_id] = {
             'status': 'pending',
             'response': '',
             'error': None,
-            'time_started': time.time()
+            'time_started': time.time(),
+            'style': resolution_style  # Stocker le style pour référence
         }
         
         threading.Thread(
             target=process_image_background,
-            args=(task_id, image_data)
+            args=(task_id, image_data, resolution_style)  # Passer le style
         ).start()
         
         return jsonify({
             'task_id': task_id,
-            'status': 'pending'
+            'status': 'pending',
+            'style': resolution_style
         })
 
     except Exception as e:
         print(f"Exception lors de la création de la tâche: {e}")
         return jsonify({'error': f'Une erreur serveur est survenue: {str(e)}'}), 500
 
+
+
 @app.route('/task/<task_id>', methods=['GET'])
 def get_task_status(task_id):
     if task_id not in task_results: