Upload 7 files

README.md

@@ -1,42 +1,24 @@
-# Songsmith AI: Music Lyric Creator
+# AI Song Lyric Generator
 
-This Hugging Face Space generates professional-quality song lyrics with proper musical structure. It creates catchy, singable lyrics with verse-chorus format and genre-specific style enhancements.
+This Hugging Face Space generates song lyrics using AI. It allows you to specify:
 
-## Features
+- Theme or topic
+- Music genre
+- Rhyme scheme
+- Maximum length
 
-- **Genre-specific enhancements**: Each genre (pop, rock, rap, folk, r&b, country, indie, dance) has custom intensifiers, metaphors, and structural patterns
-- **Professional song structure**: Proper verse-chorus format with optional bridge sections
-- **Enhanced rhyming**: Intelligent rhyme patterns that match your selected scheme
-- **Musical phrasing**: Short, catchy lines optimized for singing
-- **Emotional hooks**: Repetition and emphasis in the right places
+The app uses a lightweight model (DistilGPT-2) optimized for Hugging Face Spaces.
 
 ## How to Use
 
 1. Enter a theme or topic for your song
-2. Select a music genre (affects style, metaphors, and structure)
+2. Select a music genre (pop, rock, folk, or rap)
 3. Choose a rhyme scheme (AABB or ABAB)
-4. Adjust the maximum length as needed (longer = more sections)
+4. Adjust the maximum length as needed
 5. Click "Generate Lyrics" and enjoy your AI-created song!
 
 ## Technical Details
 
-- Built with Gradio 5.1.0
-- Uses Hugging Face Transformers with DistilGPT-2
-- Enhanced with custom lyric processing modules:
-  - IntensifierGenerator: Adds genre-specific intensifiers and metaphors
-  - RhymeEnhancer: Improves rhyming patterns
-  - SongStructureEnhancer: Creates proper verse-chorus-bridge structure
+- Built with Gradio 4.19.0
+- Uses Hugging Face Transformers
 - Optimized for free-tier Spaces
-
----
-title: Harmonyaixlyricgenerator
-emoji: 🌍
-colorFrom: red
-colorTo: pink
-sdk: gradio
-sdk_version: 5.44.1
-app_file: app.py
-pinned: false
-license: mit
-short_description: Lyric generator
----

app.py

@@ -1,7 +1,5 @@
 from transformers import pipeline
 import gradio as gr
-import random
-from lyric_enhancer import IntensifierGenerator, RhymeEnhancer, SongStructureEnhancer
 
 # Initialize models suitable for free tier
 try:
@@ -18,16 +16,11 @@ except Exception as e:
 def generate_lyrics(theme, genre, rhyme_scheme="AABB", max_length=150):
     """Generate song lyrics based on input parameters"""
     
-    # Create enhancers
-    intensifier = IntensifierGenerator(genre=genre)
-    rhymer = RhymeEnhancer()
-    structure_enhancer = SongStructureEnhancer(genre=genre)
-    
-    # Build a more musical prompt
-    prompt = f"""Write a catchy {genre} song about {theme} with {rhyme_scheme} rhyme pattern.
-Keep lines short and musical. Include emotional hooks and repetition.
+    prompt = f"""Write a short {genre} song about {theme} with {rhyme_scheme} rhyme scheme.
+
+Start with:
 
-[Verse 1]
+[Verse]
 """
     
     try:
@@ -43,14 +36,14 @@ Keep lines short and musical. Include emotional hooks and repetition.
         lyrics = result[0]['generated_text']
         
         # Check if we got repetitive content
-        if lyrics.count("[Verse]") > 3 or lyrics.count("Write a") > 1:
+        if lyrics.count("[Verse]") > 3 or lyrics.count("Start with:") > 1:
             raise ValueError("Repetitive output detected")
             
     except Exception as e:
         print(f"Error with primary generation: {e}")
         
         # Try with a simpler prompt
-        simple_prompt = f"{genre} song lyrics about {theme}:\n\n[Verse 1]\n"
+        simple_prompt = f"A {genre} song about {theme}:\n\n[Verse]\n"
         
         try:
             # Try with different parameters
@@ -67,18 +60,18 @@ Keep lines short and musical. Include emotional hooks and repetition.
             return f"Sorry, I couldn't generate lyrics at this time. Please try again with a different theme or genre."
     
     # Better handling of output formatting
-    # Find the first occurrence of [Verse] or [Verse 1] and keep everything after it
-    verse_index = max(lyrics.find("[Verse]"), lyrics.find("[Verse 1]"))
+    # Find the first occurrence of [Verse] and keep everything after it
+    verse_index = lyrics.find("[Verse]")
     
     if verse_index != -1:
         # Keep everything from [Verse] onwards
         formatted_lyrics = lyrics[verse_index:].strip()
     else:
-        # If no verse markers found, try to find the actual content
+        # If [Verse] not found, try to find the actual content
         lines = lyrics.split("\n")
         # Skip empty lines and prompt-related lines
         content_lines = []
-        skip_keywords = ["Write", "catchy", "rhyme pattern", "Keep lines", "Include"]
+        skip_keywords = ["Write a", "Start with:", "rhyme scheme", "using the"]
         
         for line in lines:
             # Skip empty lines and lines containing prompt keywords
@@ -99,43 +92,14 @@ Keep lines short and musical. Include emotional hooks and repetition.
     # 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."
-    
-    # Apply our enhancers to make the lyrics more musical
-    
-    # 1. First, enhance the structure
-    enhanced_lyrics = structure_enhancer.enhance_structure(formatted_lyrics, add_bridge=(max_length > 150))
-    
-    # 2. Apply line-by-line enhancements
-    enhanced_lines = []
-    current_section = None
-    section_lines = []
-    
-    for line in enhanced_lyrics.split("\n"):
-        # Track sections for rhyming purposes
-        if line.startswith("["):
-            # Process previous section
-            if section_lines and current_section:
-                # Apply rhyme enhancement to the section
-                rhymed_lines = rhymer.enhance_section(section_lines, genre)
-                enhanced_lines.extend(rhymed_lines)
-                section_lines = []
-            
-            # Start new section
-            enhanced_lines.append(line)
-            current_section = line
-            continue
         
-        # Apply intensifiers to individual lines
-        intensified_line = intensifier.generate(line)
-        section_lines.append(intensified_line)
-    
-    # Process the last section
-    if section_lines and current_section:
-        rhymed_lines = rhymer.enhance_section(section_lines, genre)
-        enhanced_lines.extend(rhymed_lines)
-    
-    # Final enhanced lyrics
-    formatted_lyrics = "\n".join(enhanced_lines)
+    # 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
 
@@ -169,21 +133,17 @@ css = """
 
 # Example themes for the interface
 examples = [
-    ["heartbreak and moving on", "pop", "AABB", 150],
-    ["dancing all night", "dance", "ABAB", 150],
-    ["fighting for your dreams", "rock", "AABB", 180],
-    ["summer romance", "folk", "ABAB", 150],
-    ["city lights and late nights", "rap", "ABAB", 200],
-    ["lost love and redemption", "country", "AABB", 180],
-    ["finding yourself", "indie", "ABAB", 170],
-    ["midnight feelings", "r&b", "AABB", 160],
+    ["lost in the forest", "folk", "AABB", 150],
+    ["falling in love", "pop", "ABAB", 150],
+    ["overcoming challenges", "rock", "AABB", 180],
+    ["city life", "rap", "ABAB", 200],
 ]
 
 # Create Gradio interface with a custom theme
 with gr.Blocks(css=css, theme="soft") as demo:
     with gr.Column(elem_classes="container"):
-        gr.Markdown("# 🎵 Songsmith AI: Music Lyric Creator", elem_classes="title")
-        gr.Markdown("Create catchy, singable song lyrics with verse-chorus structure and genre-specific style", elem_classes="subtitle")
+        gr.Markdown("# 🎵 AI Song Lyric Generator", elem_classes="title")
+        gr.Markdown("Create unique song lyrics with AI assistance", elem_classes="subtitle")
         
         with gr.Row():
             with gr.Column():
@@ -193,10 +153,10 @@ with gr.Blocks(css=css, theme="soft") as demo:
                     info="What should your song be about?"
                 )
                 genre_input = gr.Dropdown(
-                    ["pop", "rock", "folk", "rap", "r&b", "country", "indie", "dance"], 
+                    ["pop", "rock", "folk", "rap"], 
                     label="Music Genre",
                     value="pop",
-                    info="Select the musical style for genre-specific enhancements"
+                    info="Select the musical style"
                 )
                 rhyme_input = gr.Dropdown(
                     ["AABB", "ABAB"], 
@@ -230,21 +190,14 @@ with gr.Blocks(css=css, theme="soft") as demo:
             """
             ### How to use
             1. Enter a theme or topic for your song
-            2. Select a music genre (affects style, metaphors, and structure)
+            2. Select a music genre
             3. Choose a rhyme scheme
-            4. Adjust the length as needed (longer = more sections)
+            4. Adjust the length as needed
             5. Click "Generate Lyrics" and enjoy your AI-created song!
             
-            ### Features
-            - Genre-specific intensifiers and metaphors
-            - Enhanced rhyming patterns
-            - Proper song structure with verse-chorus format
-            - Bridge sections in longer songs
-            - Catchy, singable lines optimized for each genre
-            
             ---
             
-            *Note: This app uses a lightweight AI model (DistilGPT-2) enhanced with specialized lyric processing.*
+            *Note: This app uses a lightweight AI model (DistilGPT-2) optimized for Hugging Face Spaces.*
             """,
             elem_classes="footer"
         )

gitignore

@@ -0,0 +1,36 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual Environment
+venv/
+ENV/
+.env
+
+# IDE files
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# OS specific
+.DS_Store
+Thumbs.db

package-lock.json

@@ -0,0 +1,6 @@
+{
+  "name": "harmonyaixlyricgenerator",
+  "lockfileVersion": 3,
+  "requires": true,
+  "packages": {}
+}

requirements.txt

@@ -1,4 +1,4 @@
 transformers>=4.30.0
-gradio==5.1.0
+gradio>=4.19.0
 torch>=2.0.0
 accelerate>=0.20.0