Update app.py

app.py

@@ -11,7 +11,7 @@ whisper_model = whisper.load_model("small")
 
 def audio_prompt_to_chatgpt(audio_path):
     if audio_path is None:
-        return "No audio detected.", "", ""
+        return "", "", ""
 
     # 1️⃣ Transcribe audio → Chinese text
     result = whisper_model.transcribe(audio_path, language="zh")
@@ -20,9 +20,9 @@ def audio_prompt_to_chatgpt(audio_path):
     if not chinese_text:
         return "⚠️ Could not transcribe audio.", "", ""
 
-    # 2️⃣ Send transcription as a prompt to ChatGPT
+    # 2️⃣ ChatGPT reply in Chinese
     try:
-        response = client.chat.completions.create(
+        response_ch = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant. Follow the user's instructions."},
@@ -30,13 +30,13 @@ def audio_prompt_to_chatgpt(audio_path):
             ],
             temperature=0.7
         )
-        chatgpt_reply_chinese = response.choices[0].message.content.strip()
+        chatgpt_reply_chinese = response_ch.choices[0].message.content.strip()
     except Exception as e:
         chatgpt_reply_chinese = f"⚠️ ChatGPT failed: {e}"
 
     # 3️⃣ Translate ChatGPT reply into English
     try:
-        translation_response = client.chat.completions.create(
+        response_en = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
                 {"role": "system", "content": "You are a professional translator. Translate the following text into fluent English."},
@@ -44,28 +44,44 @@ def audio_prompt_to_chatgpt(audio_path):
             ],
             temperature=0
         )
-        chatgpt_reply_english = translation_response.choices[0].message.content.strip()
+        chatgpt_reply_english = response_en.choices[0].message.content.strip()
     except Exception as e:
         chatgpt_reply_english = f"⚠️ Translation failed: {e}"
 
-    return chinese_text, chatgpt_reply_chinese, chatgpt_reply_english
+    return chatgpt_reply_chinese, chatgpt_reply_english, chinese_text
 
 # 🚀 Gradio UI
-app = gr.Interface(
-    fn=audio_prompt_to_chatgpt,
-    inputs=gr.Audio(
-        sources=["microphone", "upload"],  # microphone or upload
+with gr.Blocks() as app:
+    gr.Markdown("### 🎤 Chinese Voice → ChatGPT with English Translation")
+
+    audio_input = gr.Audio(
+        sources=["microphone", "upload"],
         type="filepath",
         label="🎙️ Speak Chinese or Upload Audio"
-    ),
-    outputs=[
-        gr.Textbox(label="🈶 Transcribed Chinese Prompt"),
-        gr.Textbox(label="🤖 ChatGPT Response (Chinese)"),
-        gr.Textbox(label="🌍 ChatGPT Response (English)")
-    ],
-    title="🎤 Chinese Voice → ChatGPT Prompt with English",
-    description="Speak Chinese or upload audio. Whisper transcribes, ChatGPT executes your instruction, and the reply is shown in Chinese and English."
-)
+    )
+
+    chinese_output = gr.Textbox(label="🤖 ChatGPT Response (Chinese)")
+    english_output = gr.Textbox(label="🌍 ChatGPT Response (English)")
+    transcription_output = gr.Textbox(label="🈶 Transcribed Chinese Prompt")
+
+    submit_btn = gr.Button("Generate Response")
+
+    # Copy buttons
+    copy_chinese_btn = gr.Button("Copy Chinese")
+    copy_english_btn = gr.Button("Copy English")
+    copy_transcription_btn = gr.Button("Copy Transcription")
+
+    # Connect submit button
+    submit_btn.click(
+        fn=audio_prompt_to_chatgpt,
+        inputs=audio_input,
+        outputs=[chinese_output, english_output, transcription_output]
+    )
+
+    # Connect copy buttons to copy respective outputs
+    copy_chinese_btn.click(lambda x: x, inputs=chinese_output, outputs=chinese_output)
+    copy_english_btn.click(lambda x: x, inputs=english_output, outputs=english_output)
+    copy_transcription_btn.click(lambda x: x, inputs=transcription_output, outputs=transcription_output)
 
 if __name__ == "__main__":
     app.launch()