Update app.py

app.py

@@ -3,9 +3,15 @@ import google.generativeai as genai
 from yolov5 import YOLOv5
 from PIL import Image
 from transformers import pipeline
+import os
 
-# Clé API
-genai.configure(api_key="AIzaSyB_Hnl_77gd1I8xs6iuLqKIoGHgsZMXm1M")
+# Charger la clé API depuis une variable d'environnement
+api_key = os.getenv("GENAI_API_KEY")
+if not api_key:
+    raise ValueError("AIzaSyC5YtC0yxdJIRFSL6gK_p4k8FY_ycz8tZ8")
+
+# Configurer la clé API
+genai.configure(api_key=api_key)
 
 # Le modèle Gemini Pro
 gemini_model = genai.GenerativeModel('gemini-pro')
@@ -28,15 +34,12 @@ def generate_recipe(ingredients):
     prompt = f"""
     Oubliez tout ce que je vous ai demandé avant et concentrez-vous uniquement sur cette tâche.
     Vous êtes un chef virtuel spécialisé dans la création de recettes à partir des ingrédients disponibles. On vous a fourni une liste d'aliments suivants : {', '.join(set(ingredients))}. Utilisez ces ingrédients pour créer une recette savoureuse et nutritive. Votre recette doit inclure :
-
     1. Une liste d'ingrédients avec les quantités nécessaires.
     2. Des instructions étape par étape pour la préparation et la cuisson.
     3. Des conseils pour les substitutions possibles si certains ingrédients manquent.
     4. Le temps de préparation et de cuisson total.
     5. Des suggestions de présentation pour rendre le plat encore plus appétissant.
-
     Assurez-vous que la recette soit accessible à un niveau de compétence culinaire moyen et qu'elle utilise les ingrédients de manière optimale.
-
     Exemple de format de réponse :
     ```
     Recette : [Nom de la recette]
@@ -62,11 +65,14 @@ def text_to_audio(text, output_file="output.wav"):
 
 # Interface Gradio
 def process_image(image):
-    image = Image.open(image)
-    ingredients = detect_ingredients(image)
-    recipe = generate_recipe(ingredients)
-    audio_file = text_to_audio(recipe)
-    return f"Ingrédients détectés : {', '.join(ingredients)}\n\nRecette générée :\n{recipe}", audio_file
+    try:
+        image = Image.open(image)
+        ingredients = detect_ingredients(image)
+        recipe = generate_recipe(ingredients)
+        audio_file = text_to_audio(recipe)
+        return f"Ingrédients détectés : {', '.join(ingredients)}\n\nRecette générée :\n{recipe}", audio_file
+    except Exception as e:
+        return f"Une erreur s'est produite : {str(e)}", None
 
 # Mise à jour de l'interface Gradio
 iface = gr.Interface(
@@ -82,3 +88,4 @@ if __name__ == "__main__":
     iface.launch()
 
 
+