Update app.py

app.py

@@ -294,17 +294,18 @@ def process_parallel(tasks):
         results = pool.map(lambda task: task(), tasks)
     return results
 
-def generate_response_from_prompt(prompt, generator):
+def generate_response_from_prompt(prompt, model_name="google/flan-t5-xl"):
+    generator = pipeline('text-generation', model=model_name, tokenizer=model_name)
     responses = generate_random_response([prompt], generator)
     return responses[0]
 
-def generate_image_from_prompt(prompt, image_type):
+def generate_image_from_prompt(prompt, image_type, model_name):
     if image_type == "diffusers":
-        image = generate_diffusers_image_from_redis("diffusers_model_name", prompt)
+        image = generate_diffusers_image_from_redis(model_name, prompt)
     elif image_type == "stable-diffusion":
-        image = generate_stable_diffusion_image_from_redis("stable_diffusion_model_name", prompt)
+        image = generate_stable_diffusion_image_from_redis(model_name, prompt)
     elif image_type == "img2img":
-        image = generate_img2img_from_redis("img2img_model_name", "init_image.png", prompt)
+        image = generate_img2img_from_redis(model_name, "init_image.png", prompt)
     return image
 
 def gradio_app():
@@ -314,36 +315,36 @@ def gradio_app():
             prompt_text = gr.Textbox(label="Texto de Entrada")
             text_output = gr.Textbox(label="Respuesta")
             text_button = gr.Button("Generar Texto")
-            text_button.click(generate_response_from_prompt, inputs=prompt_text, outputs=text_output)
+        text_button.click(generate_response_from_prompt, inputs=prompt_text, outputs=text_output)
 
         gr.Markdown("## Generación de Imágenes con Diffusers, Stable Diffusion e Img2Img")
         with gr.Row():
             prompt_image = gr.Textbox(label="Prompt de Imagen")
             image_type = gr.Dropdown(["diffusers", "stable-diffusion", "img2img"], label="Tipo de Imagen")
-            image_output = gr.Image(type="pil", label="Imagen Generada")
+            image_output = gr.Image(label="Imagen Generada")
             image_button = gr.Button("Generar Imagen")
-            image_button.click(generate_image_from_prompt, inputs=[prompt_image, image_type], outputs=image_output)
+        image_button.click(generate_image_from_prompt, inputs=[prompt_image, image_type], outputs=image_output)
 
         gr.Markdown("## Generación de Video")
         with gr.Row():
             prompt_video = gr.Textbox(label="Prompt de Video")
-            video_output = gr.Video(type="file", label="Video Generado")
+            video_output = gr.Video(label="Video Generado")
             video_button = gr.Button("Generar Video")
-            video_button.click(generate_video_from_redis, inputs=prompt_video, outputs=video_output)
+        video_button.click(generate_video_from_redis, inputs=prompt_video, outputs=video_output)
 
         gr.Markdown("## Generación de Audio con MusicGen")
         with gr.Row():
             text_prompts_audio = gr.Textbox(label="Prompts de Audio")
-            audio_output = gr.Audio(type="file", label="Audio Generado")
+            audio_output = gr.Audio(label="Audio Generado")
             audio_button = gr.Button("Generar Audio")
-            audio_button.click(generate_musicgen_audio_from_redis, inputs=text_prompts_audio, outputs=audio_output)
+        audio_button.click(generate_musicgen_audio_from_redis, inputs=text_prompts_audio, outputs=audio_output)
 
         gr.Markdown("## Transcripción de Audio con Whisper")
         with gr.Row():
-            audio_file = gr.Audio(type="file", label="Archivo de Audio")
+            audio_file = gr.Audio(type="filepath", label="Archivo de Audio")
             transcription_output = gr.Textbox(label="Transcripción")
             audio_button = gr.Button("Transcribir Audio")
-            audio_button.click(transcribe_audio_from_redis, inputs=audio_file, outputs=transcription_output)
+        audio_button.click(transcribe_audio_from_redis, inputs=audio_file, outputs=transcription_output)
 
         gr.Markdown("## Traducción de Texto")
         with gr.Row():
@@ -352,16 +353,16 @@ def gradio_app():
             src_lang_input = gr.Textbox(label="Idioma de Origen", value="en")
             tgt_lang_input = gr.Textbox(label="Idioma de Destino", value="es")
             translate_button = gr.Button("Traducir Texto")
-            translate_button.click(translate_text_from_redis, inputs=[text_input, src_lang_input, tgt_lang_input], outputs=translation_output)
+        translate_button.click(translate_text_from_redis, inputs=[text_input, src_lang_input, tgt_lang_input], outputs=translation_output)
 
         gr.Markdown("## Resumen de Texto")
         with gr.Row():
             text_to_summarize = gr.Textbox(label="Texto para Resumir")
             summary_output = gr.Textbox(label="Resumen")
             summarize_button = gr.Button("Generar Resumen")
-            summarize_button.click(summarize_text_from_redis, inputs=text_to_summarize, outputs=summary_output)
+        summarize_button.click(summarize_text_from_redis, inputs=text_to_summarize, outputs=summary_output)
 
     app.launch()
 
 if __name__ == "__main__":
-    gradio_app()
+    gradio_app()