Update app.py

app.py

@@ -11,14 +11,14 @@ whisper_model = whisper.load_model("small")
 
 def audio_prompt_to_chatgpt(audio_path):
     if audio_path is None:
-        return "No audio detected.", ""
+        return "No audio detected.", "", ""
 
     # 1️⃣ Transcribe audio → Chinese text
     result = whisper_model.transcribe(audio_path, language="zh")
     chinese_text = result.get("text", "").strip()
 
     if not chinese_text:
-        return "⚠️ Could not transcribe audio.", ""
+        return "⚠️ Could not transcribe audio.", "", ""
 
     # 2️⃣ Send transcription as a prompt to ChatGPT
     try:
@@ -30,11 +30,25 @@ def audio_prompt_to_chatgpt(audio_path):
             ],
             temperature=0.7
         )
-        chatgpt_reply = response.choices[0].message.content.strip()
+        chatgpt_reply_chinese = response.choices[0].message.content.strip()
     except Exception as e:
-        chatgpt_reply = f"⚠️ ChatGPT failed: {e}"
+        chatgpt_reply_chinese = f"⚠️ ChatGPT failed: {e}"
 
-    return chinese_text, chatgpt_reply
+    # 3️⃣ Translate ChatGPT reply into English
+    try:
+        translation_response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "You are a professional translator. Translate the following text into fluent English."},
+                {"role": "user", "content": chatgpt_reply_chinese}
+            ],
+            temperature=0
+        )
+        chatgpt_reply_english = translation_response.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
 
 # 🚀 Gradio UI
 app = gr.Interface(
@@ -46,10 +60,11 @@ app = gr.Interface(
     ),
     outputs=[
         gr.Textbox(label="🈶 Transcribed Chinese Prompt"),
-        gr.Textbox(label="🤖 ChatGPT Response")
+        gr.Textbox(label="🤖 ChatGPT Response (Chinese)"),
+        gr.Textbox(label="🌍 ChatGPT Response (English)")
     ],
-    title="🎤 Chinese Voice → ChatGPT Prompt",
-    description="Speak Chinese or upload audio. Whisper transcribes, then ChatGPT executes your instructions."
+    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."
 )
 
 if __name__ == "__main__":