Update app.py

app.py

@@ -44,7 +44,7 @@ if GOOGLE_API_KEY:
     except Exception as e:
         logger.critical(f"Erreur critique lors de l'initialisation du client Gemini: {e}", exc_info=True)
 else:
-    logger.critical("GEMINI_API_KEY non trouvé dans les variables d'environnement. Le service ne fonctionnera pas.")
+    logger.critical("GOOGLE_API_KEY non trouvé dans les variables d'environnement. Le service ne fonctionnera pas.")
 
 task_results = {}
 
@@ -97,6 +97,52 @@ def save_user_image(image_data, filename, task_id):
         logger.error(f"Erreur lors de la sauvegarde de l'image {filename}: {e}")
         return None
 
+def upload_file_to_genai_api(file_data, filename, mime_type):
+    """Upload un fichier vers l'API Files de Google GenAI."""
+    try:
+        # Déterminer l'extension appropriée
+        if mime_type.startswith('image/'):
+            if mime_type == 'image/jpeg':
+                suffix = '.jpg'
+            elif mime_type == 'image/png':
+                suffix = '.png'
+            elif mime_type == 'image/gif':
+                suffix = '.gif'
+            elif mime_type == 'image/webp':
+                suffix = '.webp'
+            else:
+                suffix = '.jpg'  # Par défaut
+        elif mime_type == 'application/pdf':
+            suffix = '.pdf'
+        else:
+            suffix = '.tmp'
+        
+        # Créer un fichier temporaire
+        with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as temp_file:
+            temp_file.write(file_data)
+            temp_file_path = temp_file.name
+        
+        logger.info(f"Upload du fichier '{filename}' ({len(file_data)} bytes) vers l'API Files de Google GenAI...")
+        
+        # Upload vers l'API Files
+        file_ref = client.files.upload(file=temp_file_path)
+        
+        # Nettoyer le fichier temporaire
+        os.unlink(temp_file_path)
+        
+        logger.info(f"Fichier '{filename}' uploadé avec succès. Référence: {file_ref.name}")
+        return file_ref
+        
+    except Exception as e:
+        logger.error(f"Erreur lors de l'upload du fichier '{filename}' vers l'API Files: {e}")
+        # Nettoyer le fichier temporaire en cas d'erreur
+        try:
+            if 'temp_file_path' in locals():
+                os.unlink(temp_file_path)
+        except:
+            pass
+        return None
+
 def get_all_tasks_info():
     """Récupère toutes les informations des tâches pour le centre de gestion."""
     tasks_info = []
@@ -251,33 +297,31 @@ def process_files_background(task_id, files_data, resolution_style):
 
         contents = []
         logger.info(f"[Task {task_id}] Préparation des fichiers pour l'API Gemini.")
+        
+        # Traiter tous les fichiers en utilisant l'API Files
         for file_info in files_data:
+            logger.info(f"[Task {task_id}] Traitement du fichier '{file_info['filename']}' ({file_info['type']}).")
+            
+            # Sauvegarder l'image utilisateur si c'est une image
             if file_info['type'].startswith('image/'):
-                logger.info(f"[Task {task_id}] Traitement de l'image '{file_info['filename']}'.")
-                
-                # Sauvegarder l'image utilisateur
                 saved_filename = save_user_image(file_info['data'], file_info['filename'], task_id)
-                
-                img = Image.open(io.BytesIO(file_info['data']))
-                buffered = io.BytesIO()
-                img.save(buffered, format="PNG")
-                img_base64_str = base64.b64encode(buffered.getvalue()).decode()
-                contents.append({'inline_data': {'mime_type': 'image/png', 'data': img_base64_str}})
             
-            elif file_info['type'] == 'application/pdf':
-                logger.info(f"[Task {task_id}] Upload du PDF '{file_info['filename']}' vers Google GenAI File API.")
-                with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as temp_pdf:
-                    temp_pdf.write(file_info['data'])
-                    temp_pdf_path = temp_pdf.name
-                
-                file_ref = client.files.upload(file=temp_pdf_path)
+            # Upload vers l'API Files de Google GenAI
+            file_ref = upload_file_to_genai_api(
+                file_info['data'], 
+                file_info['filename'], 
+                file_info['type']
+            )
+            
+            if file_ref:
                 uploaded_file_refs.append(file_ref)
                 contents.append(file_ref)
-                os.unlink(temp_pdf_path)
-                logger.info(f"[Task {task_id}] PDF '{file_info['filename']}' uploadé avec succès. Référence: {file_ref.name}")
+                logger.info(f"[Task {task_id}] Fichier '{file_info['filename']}' uploadé avec succès. Référence: {file_ref.name}")
+            else:
+                logger.warning(f"[Task {task_id}] Échec de l'upload du fichier '{file_info['filename']}'. Fichier ignoré.")
 
         if not contents:
-            raise ValueError("Aucun contenu valide (image ou PDF) n'a été traité.")
+            raise ValueError("Aucun fichier n'a pu être uploadé vers l'API Files de Google GenAI.")
         
         prompt_to_use = get_prompt_for_style(resolution_style)
         if not prompt_to_use:
@@ -285,7 +329,8 @@ def process_files_background(task_id, files_data, resolution_style):
         contents.append(prompt_to_use)
 
         task_results[task_id]['status'] = 'generating_latex'
-        logger.info(f"[Task {task_id}] Envoi de la requête à l'API Gemini (modèle gemini-2.5-pro).")
+        logger.info(f"[Task {task_id}] Envoi de la requête à l'API Gemini (modèle gemini-2.5-flash) avec {len(uploaded_file_refs)} fichier(s).")
+        
         gemini_response = client.models.generate_content(
             model="gemini-2.5-flash", 
             contents=contents,
@@ -325,14 +370,15 @@ def process_files_background(task_id, files_data, resolution_style):
         task_results[task_id]['error'] = str(e)
         task_results[task_id]['response'] = f"Une erreur est survenue: {str(e)}"
     finally:
+        # Nettoyer tous les fichiers uploadés vers l'API Files
         if uploaded_file_refs:
-            logger.info(f"[Task {task_id}] Nettoyage des {len(uploaded_file_refs)} fichiers temporaires de l'API Gemini.")
+            logger.info(f"[Task {task_id}] Nettoyage des {len(uploaded_file_refs)} fichiers temporaires de l'API Files.")
             for file_ref in uploaded_file_refs:
                 try:
                     client.files.delete(file_ref)
-                    logger.info(f"[Task {task_id}] Fichier temporaire Gemini '{file_ref.name}' supprimé.")
+                    logger.info(f"[Task {task_id}] Fichier temporaire '{file_ref.name}' supprimé de l'API Files.")
                 except Exception as del_e:
-                    logger.warning(f"[Task {task_id}] Échec de la suppression du fichier temporaire Gemini '{file_ref.name}': {del_e}")
+                    logger.warning(f"[Task {task_id}] Échec de la suppression du fichier temporaire '{file_ref.name}': {del_e}")
 
 # --- Routes Flask (API Endpoints) ---