Update app.py

app.py

@@ -16,12 +16,13 @@ except Exception as e:
 def generate_lyrics(theme, genre, rhyme_scheme="AABB", max_length=150):
     """Generate song lyrics based on input parameters"""
     
-    prompt = f"""Write a short {genre} song about {theme} with {rhyme_scheme} rhyme scheme.
+    # A more structured prompt to guide the model towards a lyrical format
+    prompt = f"""Task: Write lyrics for a {genre} song.
+Theme: {theme}
+Rhyme: {rhyme_scheme}
+Style: Poetic, emotional, with verses and a chorus.
 
-Start with:
-
-[Verse]
-"""
+[Verse 1]"""
     
     try:
         # Increase temperature for more creativity and reduce repetition
@@ -43,7 +44,7 @@ Start with:
         print(f"Error with primary generation: {e}")
         
         # Try with a simpler prompt
-        simple_prompt = f"A {genre} song about {theme}:\n\n[Verse]\n"
+        simple_prompt = f"A {genre} song about {theme}:\n\n[Verse 1]\n"
         
         try:
             # Try with different parameters
@@ -71,7 +72,7 @@ Start with:
         lines = lyrics.split("\n")
         # Skip empty lines and prompt-related lines
         content_lines = []
-        skip_keywords = ["Write a", "Start with:", "rhyme scheme", "using the"]
+        skip_keywords = ["Write a", "Start with:", "rhyme scheme", "using the", "Task:", "Theme:", "Rhyme:", "Style:", "Lyrics:"]
         
         for line in lines:
             # Skip empty lines and lines containing prompt keywords
@@ -92,14 +93,6 @@ Start with:
     # If we have very few unique lines compared to total lines, it's repetitive
     if len(lines) > 5 and len(unique_lines) < len(lines) / 2:
         return "The model generated repetitive content. Please try again with a different theme or genre."
-        
-    # Add [Chorus] section if it doesn't exist and the lyrics are long enough
-    if "[Chorus]" not in formatted_lyrics and len(formatted_lyrics.split()) > 20:
-        verse_lines = formatted_lyrics.split("\n")
-        # Insert a chorus after some verse lines
-        chorus_position = min(6, len(verse_lines))
-        verse_lines.insert(chorus_position, "\n[Chorus]")
-        formatted_lyrics = "\n".join(verse_lines)
     
     return formatted_lyrics