Update app.py

app.py

@@ -1,42 +1,17 @@
 import gradio as gr
-from model import load_stable_diffusion_model, load_gpt_model, generate_image_from_prompt, generate_prompt
-from utils import load_whisper_model, transcribe_audio
+from whisper_model import transcribe_audio
+from model import generate_prompt, generate_image_from_prompt
 
-# 加载模型
-stable_diff_pipe = load_stable_diffusion_model()
-gpt_model = load_gpt_model()
-whisper_model = load_whisper_model()
-
-# 生成图像和提示
-def generate_response(description):
+def gradio_interface_with_audio(audio_file):
+    """ Gradio 界面函数，支持语音输入 """
+    description = transcribe_audio(audio_file)
     # 生成提示
-    prompt = generate_prompt(gpt_model, description)
+    prompt = generate_prompt(description)
     # 生成图像
-    image = generate_image_from_prompt(stable_diff_pipe, prompt)
+    image = generate_image_from_prompt(prompt)
     return prompt, image
 
-# 语音输入和图像生成
-def handle_audio(audio_file):
-    text = transcribe_audio(whisper_model, audio_file)
-    return generate_response(text)
-
-# Gradio界面设置
-iface = gr.Interface(
-    fn=generate_response, 
-    inputs="text", 
-    outputs=["text", "image"], 
-    title="图像生成器",
-    description="输入简短的描述，生成详细提示并生成图像。"
-)
-
-# 语音输入功能
-iface_with_audio = gr.Interface(
-    fn=handle_audio, 
-    inputs=gr.Audio(source="microphone", type="filepath"), 
-    outputs=["text", "image"],
-    title="语音图像生成器",
-    description="通过语音输入来生成图像。"
-)
-
-# 运行界面
+# 创建 Gradio 界面
+iface = gr.Interface(fn=gradio_interface_with_audio, inputs="audio", outputs=["text", "image"])
 iface.launch()
+