Update app.py

app.py

@@ -1,15 +1,13 @@
-import numpy as np
-import gradio as gr
+ import gradio as gr
 import google.generativeai as genai
 from yolov5 import YOLOv5
 from PIL import Image
-from bark import SAMPLE_RATE, generate_audio, preload_models
-from bark.generation import SUPPORTED_LANGS
-from share_btn import community_icon_html, loading_icon_html, share_js
 
-# Clé API
+
+#Clé Api
 genai.configure(api_key="AIzaSyB_Hnl_77gd1I8xs6iuLqKIoGHgsZMXm1M")
 
+
 # Le modèle Gemini Pro
 gemini_model = genai.GenerativeModel('gemini-pro')
 
@@ -17,21 +15,6 @@ gemini_model = genai.GenerativeModel('gemini-pro')
 yolo_model_path = "yolov5s.pt"
 yolo_model = YOLOv5(yolo_model_path, device="cpu")
 
-# Précharger les modèles de bark
-DEBUG_MODE = False
-if not DEBUG_MODE:
-    _ = preload_models()
-
-# Prompts disponibles pour bark
-AVAILABLE_PROMPTS = ["Unconditional", "Announcer"]
-PROMPT_LOOKUP = {}
-for _, lang in SUPPORTED_LANGS:
-    for n in range(10):
-        label = f"Speaker {n} ({lang})"
-        AVAILABLE_PROMPTS.append(label)
-        PROMPT_LOOKUP[label] = f"{lang}_speaker_{n}"
-PROMPT_LOOKUP["Unconditional"] = None
-PROMPT_LOOKUP["Announcer"] = "announcer"
 
 # Détection d'ingrédients
 def detect_ingredients(image):
@@ -67,83 +50,27 @@ def generate_recipe(ingredients):
     response = gemini_model.generate_content(prompt)
     return response.text
 
-# Fonction de génération de TTS
-def gen_tts(text, history_prompt):
-    history_prompt = PROMPT_LOOKUP[history_prompt]
-    if DEBUG_MODE:
-        audio_arr = np.zeros(SAMPLE_RATE)
-    else:
-        audio_arr = generate_audio(text, history_prompt=history_prompt)
-    audio_arr = (audio_arr * 32767).astype(np.int16)
-    return (SAMPLE_RATE, audio_arr)
+    response = gemini_model.generatecontent(prompt)
+    return response.text
+
 
 # Interface Gradio
 def process_image(image):
-    image = Image.open(image)
-    ingredients = detect_ingredients(image)
-    recipe = generate_recipe(ingredients)
-    return f"Ingrédients détectés : {', '.join(ingredients)}\n\nRecette générée :\n{recipe}"
+        image = Image.open(image)
+        ingredients = detect_ingredients(image)
+        recipe = generate_recipe(ingredients)
+        return f"Ingrédients détectés : {', '.join(ingredients)}\n\nRecette générée :\n{recipe}"
 
-# Mise à jour de l'interface Gradio
-css = """
-        #share-btn-container {
-            display: flex;
-            padding-left: 0.5rem !important;
-            padding-right: 0.5rem !important;
-            background-color: #000000;
-            justify-content: center;
-            align-items: center;
-            border-radius: 9999px !important;
-            width: 13rem;
-            margin-top: 10px;
-            margin-left: auto;
-            flex: unset !important;
-        }
-        #share-btn {
-            all: initial;
-            color: #ffffff;
-            font-weight: 600;
-            cursor: pointer;
-            font-family: 'IBM Plex Sans', sans-serif;
-            margin-left: 0.5rem !important;
-            padding-top: 0.25rem !important;
-            padding-bottom: 0.25rem !important;
-            right:0;
-        }
-        #share-btn * {
-            all: unset !important;
-        }
-        #share-btn-container div:nth-child(-n+2){
-            width: auto !important;
-            min-height: 0px !important;
-        }
-        #share-btn-container .wrap {
-            display: none !important;
-        }
-"""
 
-with gr.Blocks(css=css) as block:
-    gr.Markdown("# Générateur de Recettes par Ingrédients")
-    gr.Markdown("Téléchargez une image d'ingrédients pour générer une recette.")
-    with gr.Row():
-        with gr.Column():
-            input_image = gr.Image(type="filepath", label="Image d'ingrédients")
-            run_button = gr.Button(text="Générer Recette", type="button")
-        with gr.Column():
-            output_text = gr.Textbox(label="Recette générée", lines=10)
-            audio_out = gr.Audio(label="Audio de la recette", type="numpy")
-            with gr.Row(visible=False) as share_row:
-                with gr.Group(elem_id="share-btn-container"):
-                    community_icon = gr.HTML(community_icon_html)
-                    loading_icon = gr.HTML(loading_icon_html)
-                    share_button = gr.Button("Share to community", elem_id="share-btn")
-                    share_button.click(None, [], [], _js=share_js)
-    inputs = [input_image]
-    outputs = [output_text, audio_out]
-    gr.Examples(examples=[], fn=process_image, inputs=inputs, outputs=outputs, cache_examples=True)
-    run_button.click(fn=lambda: gr.update(visible=False), inputs=None, outputs=share_row, queue=False).then(
-        fn=process_image, inputs=inputs, outputs=outputs, queue=True).then(
-        fn=lambda: gr.update(visible=True), inputs=None, outputs=share_row, queue=False)
+# Mise à jour de l'interface Gradio
+iface = gr.Interface(
+    fn=process_image,
+    inputs=gr.Image(type="filepath"),
+    outputs="text",
+    title="Générateur de Recettes par Ingrédients",
+    description="Téléchargez une image d'ingrédients pour générer une recette.",
+)
 
-block.queue()
-block.launch()
+# Pour lancer l'application
+if __name__ == "__main__":
+    iface.launch()