Update app.py

app.py

@@ -144,10 +144,10 @@ def upload_file_to_genai_api(file_data, filename, mime_type):
         return None
 
 def call_gemini_with_fallback(contents, task_id):
-    """Appelle Gemini 2.5 Pro en premier, puis 2.5 Flash en cas d'échec."""
+    """Appelle Gemini 3.5 Flash en premier, puis en fallback si échec."""
     models_to_try = [
-        {"name": "gemini-3-flash-preview", "display_name": " 2.5 Pro"},
-        {"name": "gemini-3-flash-preview", "display_name": " 2.5 Flash"}
+        {"name": "gemini-3.5-flash", "display_name": " 3.5 Flash"},
+        {"name": "gemini-3.5-flash", "display_name": " 3.5 Flash (fallback)"}
     ]
     
     last_error = None
@@ -160,12 +160,12 @@ def call_gemini_with_fallback(contents, task_id):
             logger.info(f"[Task {task_id}] Tentative {i+1}/2: Appel de {model_display}...")
             
             # Mettre à jour le statut pour indiquer quel modèle est en cours d'utilisation
-            if model_name == "gemini-3-flash-preview":
+            if i == 0:
                 task_results[task_id]['status'] = 'generating_latex_pro'
-                task_results[task_id]['current_model'] = 'Gemini 2.5 Pro'
+                task_results[task_id]['current_model'] = 'Gemini 3.5 Flash'
             else:
                 task_results[task_id]['status'] = 'generating_latex_flash'
-                task_results[task_id]['current_model'] = 'Gemini 2.5 Flash (fallback)'
+                task_results[task_id]['current_model'] = 'Gemini 3.5 Flash (fallback)'
             
             # Faire l'appel à l'API
             gemini_response = client.models.generate_content(
@@ -250,8 +250,8 @@ def get_system_stats():
     pending_tasks = sum(1 for task in task_results.values() if task['status'] not in ['completed', 'error'])
     
     # Statistiques d'utilisation des modèles
-    pro_usage = sum(1 for task in task_results.values() if task.get('used_model') == 'Gemini 2.5 Pro')
-    flash_usage = sum(1 for task in task_results.values() if task.get('used_model') == 'Gemini 2.5 Flash (fallback)')
+    pro_usage = sum(1 for task in task_results.values() if task.get('used_model') == 'Gemini 3.5 Flash')
+    flash_usage = sum(1 for task in task_results.values() if task.get('used_model') == 'Gemini 3.5 Flash (fallback)')
     
     # Compter les fichiers PDF générés
     pdf_files = 0
@@ -402,7 +402,7 @@ def process_files_background(task_id, files_data, resolution_style):
             raise ValueError(f"Le fichier de prompt pour le style '{resolution_style}' est introuvable ou vide.")
         contents.append(prompt_to_use)
 
-        # Appeler Gemini avec fallback Pro -> Flash
+        # Appeler Gemini avec fallback
         logger.info(f"[Task {task_id}] Début des appels aux modèles Gemini avec {len(uploaded_file_refs)} fichier(s).")
         full_latex_response = call_gemini_with_fallback(contents, task_id)