.

app.py

@@ -11,7 +11,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 # API URLs and headers
 AUDIO_API_URL = "https://api-inference.huggingface.co/models/MIT/ast-finetuned-audioset-10-10-0.4593"
-JANUS_API_URL = "https://api-inference.huggingface.co/models/deepseek-ai/Janus-1.3B/text-generation"
+LYRICS_API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
 headers = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"}
 
 def format_error(message):
@@ -24,37 +24,47 @@ def create_lyrics_prompt(classification_results):
     top_result = classification_results[0]
     genre = top_result['label']
     confidence = float(top_result['score'].strip('%')) / 100
-
-    # Create a detailed prompt
-    prompt = f"""Write song lyrics in the style of {genre} music. The song should capture the essence of this genre.
-    Additional musical elements detected: {', '.join(r['label'] for r in classification_results[1:3])}
     
-    Please write creative and original lyrics that:
-    1. Match the {genre} style
-    2. Have a clear structure (verse, chorus)
-    3. Reflect the mood and themes common in this genre
+    # Get additional musical elements
+    additional_elements = [r['label'] for r in classification_results[1:3]]
     
-    Generate the lyrics:
-    """
+    # Create a detailed creative prompt
+    prompt = f"""Write creative and original song lyrics that capture the following musical elements:
+
+Primary Style: {genre} ({confidence*100:.1f}% confidence)
+Secondary Elements: {', '.join(additional_elements)}
+
+Requirements:
+1. Create lyrics that strongly reflect the {genre} style
+2. Incorporate elements of {' and '.join(additional_elements)}
+3. Include both verses and a chorus
+4. Match the mood and atmosphere typical of this genre
+5. Use appropriate musical terminology and style
+
+Lyrics:
+
+[Verse 1]
+"""
     return prompt
 
 def generate_lyrics_with_retry(prompt, max_retries=5, initial_wait=2):
-    """Generate lyrics using the Janus model with retry logic"""
+    """Generate lyrics using GPT-J-6B with retry logic"""
     wait_time = initial_wait
     
     for attempt in range(max_retries):
         try:
             response = requests.post(
-                JANUS_API_URL,
+                LYRICS_API_URL,
                 headers=headers,
                 json={
                     "inputs": prompt,
                     "parameters": {
-                        "max_new_tokens": 200,
-                        "temperature": 0.7,
-                        "top_p": 0.9,
+                        "max_new_tokens": 250,
+                        "temperature": 0.8,
+                        "top_p": 0.92,
                         "do_sample": True,
-                        "return_full_text": False
+                        "return_full_text": False,
+                        "stop": ["[End]", "\n\n\n"]
                     }
                 }
             )
@@ -65,8 +75,16 @@ def generate_lyrics_with_retry(prompt, max_retries=5, initial_wait=2):
             if response.status_code == 200:
                 result = response.json()
                 if isinstance(result, list) and len(result) > 0:
-                    return result[0].get("generated_text", "Error: No text generated")
-                return "Error: Unexpected response format"
+                    generated_text = result[0].get("generated_text", "")
+                    # Clean up and format the generated text
+                    lines = generated_text.split('\n')
+                    cleaned_lines = []
+                    for line in lines:
+                        line = line.strip()
+                        if line and not line.startswith('###') and not line.startswith('```'):
+                            cleaned_lines.append(line)
+                    return "\n".join(cleaned_lines)
+                return "Error: No text generated"
             elif response.status_code == 503:
                 print(f"Model loading, attempt {attempt + 1}/{max_retries}. Waiting {wait_time} seconds...")
                 time.sleep(wait_time)