Update app.py

app.py

@@ -5,7 +5,7 @@ import random
 import os
 import torch
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
-from openai import OpenAI
+import openai
 
 # Load MusicGen model
 model_name = "facebook/musicgen-small"
@@ -21,8 +21,17 @@ def generate_music(prompt):
     return (sampling_rate, audio)
 
 # Function to get enhanced prompt via GPT
-openai_key = os.getenv("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")
-client = OpenAI(api_key=openai_key)
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+def refine_prompt(user_input):
+    response = openai.ChatCompletion.create(
+        model="gpt-4",
+        messages=[
+            {"role": "system", "content": "You are a music assistant. Make the user's input more descriptive for an AI music generator."},
+            {"role": "user", "content": user_input}
+        ]
+    )
+    return response.choices[0].message["content"].strip()
 
 def refine_prompt(user_input):
     completion = client.chat.completions.create(