Upload app.py

app.py

@@ -70,8 +70,12 @@ wav2vec2_model.eval()
 
 label_map = {0: "Neutral", 1: "Happy", 2: "Angry", 3: "Sad", 4: "Surprise"}
 
-# ------------------ ChatGPT API Setup ---------------------
-openai.api_key = os.getenv("OPENAI_API_KEY", "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")  # Replace or set env var
+from openai import OpenAI
+import os
+
+
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY", "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"))
+
 
 def create_prompt_from_label(label):
     return f"""
@@ -84,10 +88,11 @@ Ensure your tone is friendly and supportive, and make the recommendations short,
 You can add some lovely emoji to let it become warm.
 """
 
+
 def get_recommendations(label):
     prompt = create_prompt_from_label(label)
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gpt-4",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant that provides entertainment recommendations."},
@@ -96,7 +101,7 @@ def get_recommendations(label):
             max_tokens=500,
             temperature=0.7
         )
-        return response['choices'][0]['message']['content'].strip()
+        return response.choices[0].message.content.strip()
     except Exception as e:
         return f"❌ GPT Error: {str(e)}"
 
@@ -119,7 +124,7 @@ def process_audio_and_recommend(file_path):
     pred_idx = torch.argmax(outputs, dim=1).item()
     emotion = label_map[pred_idx]
     recommendations = get_recommendations(emotion)
-    return f"🧠 Detected Emotion: {emotion}", recommendations
+    return f"{emotion}", recommendations
 
 
 # ----------------- Gradio UI ---------------------
@@ -134,16 +139,15 @@ def process_audio_and_recommend(file_path):
 # interface.launch()
 
 with gr.Blocks(theme=gr.themes.Soft()) as interface:
-    gr.Markdown("## 🎙️ 情绪检测 + 聊天机器人")
-    gr.Markdown("上传或录制一段简短的语音片段，我会识别你的情绪，并请求 GPT 做出共情的回应。")
+    gr.Markdown("##  Your Personal Emotion Assistant 😇")
+    gr.Markdown("Whisper me your thoughts, and I’ll wrap them in a better mood 🧸 ")
 
     with gr.Row():
+            audio_input = gr.Audio(label="🎙️ Audio Input", type="filepath", format="wav")
+            submit_btn = gr.Button("Submit")
         with gr.Column():
-            audio_input = gr.Audio(label="🎧 语音输入", type="filepath", format="wav")
-            submit_btn = gr.Button("🚀 提交")
-        with gr.Column():
-            output_text_1 = gr.Text(label="🧠 检测情绪")
-            output_text_2 = gr.Text(label="💬 GPT 回复")
+            output_text_1 = gr.Text(label="🪄 I feel that you are...:")
+            output_text_2 = gr.Text(label="💬 Entertainment You May Like: ")
 
     submit_btn.click(fn=process_audio_and_recommend, inputs=audio_input, outputs=[output_text_1, output_text_2])