Update app.py

app.py

@@ -1,13 +1,15 @@
+import numpy as np
 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')
 
@@ -15,6 +17,21 @@ 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):
@@ -50,33 +67,83 @@ def generate_recipe(ingredients):
     response = gemini_model.generate_content(prompt)
     return response.text
 
-    response = gemini_model.generatecontent(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)
 
 # 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
-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.",
-)
-
-# Pour lancer l'application
-if __name__ == "__main__":
-    iface.launch()
-
-
-
-
-
-
+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)
+
+block.queue()
+block.launch()