.

app.py

@@ -13,7 +13,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"
-LYRICS_API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
+LYRICS_API_URL = "https://api-inference.huggingface.co/models/gpt2-xl"
 headers = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"}
 
 def format_error(message):
@@ -30,27 +30,15 @@ def create_lyrics_prompt(classification_results):
     # Get additional musical elements
     additional_elements = [r['label'] for r in classification_results[1:3]]
     
-    # Create a detailed creative prompt
-    prompt = f"""Write creative and original song lyrics that capture the following musical elements:
+    # Create a more focused prompt for GPT2-XL
+    prompt = f"""Write song lyrics in the style of {genre}.
+Theme: A {genre} song with elements of {' and '.join(additional_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]
-"""
+[Verse 1]"""
     return prompt
 
 def generate_lyrics_with_retry(prompt, max_retries=5, initial_wait=2):
-    """Generate lyrics using GPT-J-6B with retry logic"""
+    """Generate lyrics using GPT2-XL with retry logic"""
     wait_time = initial_wait
     
     for attempt in range(max_retries):
@@ -61,9 +49,9 @@ def generate_lyrics_with_retry(prompt, max_retries=5, initial_wait=2):
                 json={
                     "inputs": prompt,
                     "parameters": {
-                        "max_new_tokens": 250,
-                        "temperature": 0.8,
-                        "top_p": 0.92,
+                        "max_new_tokens": 200,
+                        "temperature": 0.9,
+                        "top_p": 0.95,
                         "do_sample": True,
                         "return_full_text": False,
                         "stop": ["[End]", "\n\n\n"]
@@ -81,10 +69,19 @@ def generate_lyrics_with_retry(prompt, max_retries=5, initial_wait=2):
                     # Clean up and format the generated text
                     lines = generated_text.split('\n')
                     cleaned_lines = []
+                    current_section = "[Verse 1]"
+                    
                     for line in lines:
                         line = line.strip()
                         if line and not line.startswith('###') and not line.startswith('```'):
+                            if line.lower().startswith('[verse') or line.lower().startswith('[chorus'):
+                                current_section = line
                             cleaned_lines.append(line)
+                            
+                            # Add chorus after first verse if not present
+                            if len(cleaned_lines) == 4 and current_section == "[Verse 1]":
+                                cleaned_lines.append("\n[Chorus]")
+                    
                     return "\n".join(cleaned_lines)
                 return "Error: No text generated"
             elif response.status_code == 503: