Update app.py

app.py

@@ -2,77 +2,264 @@ import os
 import time
 import logging
 import gradio as gr
-import modal
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-logger.info(f"Modal version: {modal.__version__}")
-logger.info("===== Environment Variables =====")
-for k in ["HF_HUB_ENABLE_HF_TRANSFER", "HF_DATASETS_TRUST_REMOTE_CODE", "MODAL_TOKEN"]:
-    logger.info(f"{k}: {os.environ.get(k, '❌ MISSING')}")
+# Check Modal availability
+MODAL_AVAILABLE = False
+generate_content_with_llm = None
 
-if os.environ.get("MODAL_TOKEN") is None:
-    raise RuntimeError("❌ MODAL_TOKEN not found in environment!")
-
-# ✅ Modal 1.x style — no modal.configure
 try:
-    generate_content_with_llm = modal.Function.from_name(
-        "mr-law-jason/main/content-creation-agent",  # App path
-        "generate_content_with_llm"  # Function name
-    )
-    logger.info("✅ Loaded remote Modal function")
-except Exception as e:
-    logger.error(f"❌ Failed to load Modal function: {e}")
-    generate_content_with_llm = None
+    import modal
+    logger.info(f"Modal version: {modal.__version__}")
+    
+    # Check for Modal token
+    if os.environ.get("MODAL_TOKEN"):
+        try:
+            generate_content_with_llm = modal.Function.from_name(
+                "mr-law-jason/main/content-creation-agent",
+                "generate_content_with_llm"
+            )
+            MODAL_AVAILABLE = True
+            logger.info("✅ Modal successfully connected")
+        except Exception as e:
+            logger.warning(f"⚠️ Modal function loading failed: {e}")
+    else:
+        logger.warning("⚠️ MODAL_TOKEN not found in environment")
+        
+except ImportError:
+    logger.warning("⚠️ Modal package not available")
+
+logger.info(f"Modal Status: {'✅ Available' if MODAL_AVAILABLE else '❌ Unavailable'}")
 
 def generate_fallback_content(lyrics, artist, title):
-    return {
-        'youtube': f"🎵 {title} by {artist or 'Anonymous'} 🎵\n\n{lyrics[:200]}...\n\n#music #newrelease",
-        'twitter': f"New song: {title} by {artist or 'Anonymous'} is out now! 🎵\n#newmusic #{artist.lower().replace(' ', '') if artist else ''}",
-        'instagram': f"{title} is now available! ✨\nLink in bio\n#music #instamusic",
-        'facebook': f"New music release: {title} by {artist or 'Anonymous'}!\n\n#newmusic #artist",
-        'minds': f"Independent music release: {title}\n\n#independentartist #music",
-        'gab': f"New authentic music: {title}\n\n#realmusic #artist"
+    """High-quality local content generation"""
+    
+    # Analyze lyrics for themes and mood
+    lyrics_lower = lyrics.lower()
+    
+    # Detect mood
+    mood = "uplifting"
+    if any(word in lyrics_lower for word in ['cry', 'tear', 'pain', 'hurt', 'lost', 'broken', 'sad']):
+        mood = "emotional"
+    elif any(word in lyrics_lower for word in ['energy', 'power', 'fire', 'strong', 'fight', 'drive']):
+        mood = "energetic"
+    elif any(word in lyrics_lower for word in ['love', 'happy', 'joy', 'hope', 'dream', 'light']):
+        mood = "inspiring"
+    
+    # Extract themes
+    theme_words = []
+    if any(word in lyrics_lower for word in ['love', 'heart', 'romance']):
+        theme_words.append('love')
+    if any(word in lyrics_lower for word in ['dream', 'hope', 'future']):
+        theme_words.append('dreams')
+    if any(word in lyrics_lower for word in ['night', 'dark', 'star']):
+        theme_words.append('night')
+    if any(word in lyrics_lower for word in ['life', 'living', 'journey']):
+        theme_words.append('life')
+    
+    themes_text = ', '.join(theme_words[:3]) if theme_words else 'deep emotions'
+    
+    # Detect genre for hashtags
+    genre = 'indie'
+    if any(word in lyrics_lower for word in ['rock', 'guitar', 'band', 'drums']):
+        genre = 'rock'
+    elif any(word in lyrics_lower for word in ['pop', 'radio', 'catchy', 'dance']):
+        genre = 'pop'
+    elif any(word in lyrics_lower for word in ['rap', 'flow', 'beats', 'hip']):
+        genre = 'hiphop'
+    elif any(word in lyrics_lower for word in ['electronic', 'synth', 'edm']):
+        genre = 'electronic'
+    
+    # Generate platform-specific hashtags
+    hashtags = {
+        'youtube': f"#{genre}music #newmusic #musicvideo #{mood}music #originalmusic #independentartist #songwriter #musicdiscovery #viral #subscribe #newrelease #musicproducer #instamusic #unsignedartist #emergingartist",
+        'twitter': f"#{genre} #newmusic #nowplaying #{mood} #musicvideo #viral",
+        'instagram': f"#{genre}vibes #newmusic #instamusic #{mood} #musicpost #originalmusic #artist #viral #trending #musiclover #songwriter #independentartist #newrelease #musicdiscovery #vibes",
+        'facebook': f"#{genre}music #newmusic #originalmusic #{mood} #musicvideo #viral",
+        'minds': f"#independentmusic #originalmusic #newmusic #creative #authentic",
+        'gab': f"#originalmusic #independent #authentic #realmusic #newmusic"
     }
+    
+    artist_display = artist if artist.strip() else "Anonymous Artist"
+    by_artist = f" - {artist_display}" if artist.strip() else ""
+    
+    if artist.strip():
+        # Content with artist name
+        return {
+            'youtube': f"""🎵 {title}{by_artist} 🎵
+
+Experience this latest music video! This {mood} track explores themes of {themes_text}.
+
+🎬 Watch the full music video above
+💿 Stream on all platforms  
+🔔 Subscribe for more music videos
+👍 Like if you enjoyed this track
+
+📝 LYRICS:
+{lyrics}
+
+{hashtags['youtube']}""",
+
+            'twitter': f"🎵 NEW MUSIC VIDEO: '{title}'{by_artist} is here! This {mood} track will give you all the feels 💫\n\n🎬 Watch: [Link]\n\n{hashtags['twitter']}",
+
+            'instagram': f"""🎵✨ '{title}' Official Music Video is LIVE! ✨🎵
+
+{artist_display} delivers another {mood} masterpiece that speaks to the soul 💫
+
+🎬 Full video in bio
+💭 What's your favorite lyric? Comment below!
+
+{hashtags['instagram']}""",
+
+            'facebook': f"""🎵 MUSIC VIDEO PREMIERE 🎵
+
+New music video for '{title}'{by_artist} just dropped! 
+
+This {mood} track captures the essence of {themes_text} in a beautiful way. The visuals perfectly complement the powerful lyrics.
+
+🎬 Watch the full video now!
+💬 Let us know what you think in the comments
+🔄 Share with friends who love good music
+
+{hashtags['facebook']}""",
+
+            'minds': f"""🎵 New Music Alert!
+
+{artist_display} - '{title}' Official Music Video
+
+This {mood} track showcases incredible artistry and meaningful lyrics. Independent artists like {artist_display} are creating the future of music.
+
+Support independent music and check out this latest release!
+
+{hashtags['minds']}""",
+
+            'gab': f"""New Music Video Drop!
+
+'{title}'{by_artist}
+
+Real music with real meaning. This {mood} track reminds us why authentic artistry matters in today's world.
 
-outputs = ['youtube', 'twitter', 'instagram', 'facebook', 'minds', 'gab']
+Give it a listen and support independent creators!
 
-def generate_all_content(lyrics, artist, title):
+{hashtags['gab']}"""
+        }
+    else:
+        # Anonymous content
+        return {
+            'youtube': f"""🎵 {title} 🎵
+
+Experience this captivating music video! This {mood} track explores themes of {themes_text}.
+
+🎬 Watch the full music video above
+💿 Available on all streaming platforms
+🔔 Subscribe for more amazing music videos
+👍 Like if this track moved you
+
+📝 LYRICS:
+{lyrics}
+
+{hashtags['youtube']}""",
+
+            'twitter': f"🎵 NEW MUSIC VIDEO: '{title}' is here! This {mood} track will give you all the feels 💫\n\n🎬 Watch: [Link]\n\n{hashtags['twitter']}",
+
+            'instagram': f"""🎵✨ '{title}' Official Music Video is LIVE! ✨🎵
+
+Another {mood} masterpiece that speaks to the soul 💫
+
+🎬 Full video in bio
+💭 What's your favorite lyric? Comment below!
+
+{hashtags['instagram']}""",
+
+            'facebook': f"""🎵 MUSIC VIDEO PREMIERE 🎵
+
+'{title}' official music video just dropped! 
+
+This {mood} track captures the essence of {themes_text} in a beautiful way. The visuals perfectly complement the powerful lyrics.
+
+🎬 Watch the full video now!
+💬 Let us know what you think in the comments
+🔄 Share with friends who love good music
+
+{hashtags['facebook']}""",
+
+            'minds': f"""🎵 New Music Alert!
+
+'{title}' Official Music Video
+
+This {mood} track showcases incredible artistry and meaningful lyrics. Independent music continues to push creative boundaries.
+
+Support independent music and check out this latest release!
+
+{hashtags['minds']}""",
+
+            'gab': f"""New Music Video Drop!
+
+'{title}'
+
+Real music with real meaning. This {mood} track reminds us why authentic artistry matters in today's world.
+
+Give it a listen and support independent creators!
+
+{hashtags['gab']}"""
+        }
+
+def generate_all_content(lyrics, artist, title, use_modal):
+    """Generate content with Modal toggle"""
+    if not lyrics.strip() or not title.strip():
+        error_msg = "❌ Please provide both lyrics and song title"
+        return [""] * 6 + [error_msg]
+    
     start_time = time.time()
     
     # Initial yield - empty outputs and loading status
-    empty_outputs = [""] * len(outputs)  # 6 empty strings for platforms
-    status_msg = "⏳ Generating content..."
-    yield empty_outputs + [status_msg]  # 7 total outputs
+    empty_outputs = [""] * 6
+    if use_modal and MODAL_AVAILABLE:
+        status_msg = "⏳ Generating content with Modal AI..."
+    else:
+        status_msg = "⏳ Generating content with local templates..."
+    yield empty_outputs + [status_msg]
     
     try:
-        if generate_content_with_llm:
+        if use_modal and MODAL_AVAILABLE and generate_content_with_llm:
+            # Try Modal AI generation
             result = generate_content_with_llm.remote(lyrics, artist, title)
             content = result
             status_msg = f"✅ Generated with Modal AI in {time.time() - start_time:.2f}s"
         else:
-            raise RuntimeError("Modal function not available.")
+            # Use local fallback
+            if use_modal and not MODAL_AVAILABLE:
+                status_msg = "⚠️ Modal unavailable, using local generation..."
+            content = generate_fallback_content(lyrics, artist, title)
+            if not (use_modal and not MODAL_AVAILABLE):
+                status_msg = f"✅ Generated with local templates in {time.time() - start_time:.2f}s"
+    
     except Exception as e:
         logger.warning(f"⚠️ Modal generation failed: {e}")
         content = generate_fallback_content(lyrics, artist, title)
-        status_msg = f"⚠️ Fallback used. Generated locally in {time.time() - start_time:.2f}s"
+        status_msg = f"⚠️ Modal failed, used local fallback in {time.time() - start_time:.2f}s"
     
     # Final yield - actual content for each platform + status
-    platform_outputs = [content.get(platform, f"Error generating {platform} content") for platform in outputs]
-    yield platform_outputs + [status_msg]  # 7 total outputs
+    platforms = ['youtube', 'twitter', 'instagram', 'facebook', 'minds', 'gab']
+    platform_outputs = [content.get(platform, f"Error generating {platform} content") for platform in platforms]
+    yield platform_outputs + [status_msg]
 
+# Create Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(), title="🎵 Music Content Creator") as demo:
-    gr.Markdown("""
+    gr.Markdown(f"""
     # 🎵 AI Content Generator for Musicians 🎬
     
-    Transform your song lyrics into compelling social media content using AI!
+    Transform your song lyrics into compelling social media content!
     
     **✨ Features:**
-    - 🤖 AI-powered content generation using GPT-2
-    - 📱 Platform-optimized content for YouTube, Twitter, Instagram, Facebook, Minds, and Gab
+    - 🤖 AI-powered content generation (when Modal is available)
+    - 📱 Platform-optimized content for YouTube, Twitter, Instagram, Facebook, Minds, and Gab  
     - 🎯 Smart hashtag generation based on lyrics analysis
     - 🎤 Support for both named artists and anonymous releases
+    
+    **🔧 Backend Status:** {"🟢 Modal Available" if MODAL_AVAILABLE else "🟡 Local Mode Only"}
     """)
     
     with gr.Row():
@@ -90,18 +277,29 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🎵 Music Content Creator") as de
                 info="Leave blank for anonymous releases"
             )
             title_input = gr.Textbox(
-                label="🎵 Song Title",
+                label="🎵 Song Title", 
                 placeholder="Song Title",
                 info="The title of your song"
             )
             
+            # Modal toggle with clear labeling
+            modal_toggle = gr.Radio(
+                choices=[
+                    ("🤖 AI Generation (Modal)", True),
+                    ("📝 Local Templates", False)
+                ],
+                value=MODAL_AVAILABLE,  # Default to Modal if available
+                label="Generation Method",
+                info="Choose between AI-powered generation or local templates"
+            )
+            
     run_button = gr.Button("🚀 Generate Content", variant="primary", size="lg")
     
     gr.Markdown("## 📱 Generated Social Media Content")
     
     with gr.Row():
         with gr.Column():
-            youtube_output = gr.Textbox(label="📺 YouTube Description", lines=10)
+            youtube_output = gr.Textbox(label="📺 YouTube Description", lines=12)
             twitter_output = gr.Textbox(label="🐦 Twitter/X Post", lines=4)
             instagram_output = gr.Textbox(label="📸 Instagram Caption", lines=6)
         with gr.Column():
@@ -114,7 +312,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🎵 Music Content Creator") as de
     # Connect the function with proper outputs
     run_button.click(
         fn=generate_all_content,
-        inputs=[lyrics_input, artist_input, title_input],
+        inputs=[lyrics_input, artist_input, title_input, modal_toggle],
         outputs=[youtube_output, twitter_output, instagram_output, facebook_output, minds_output, gab_output, status]
     )
     
@@ -123,18 +321,30 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🎵 Music Content Creator") as de
         examples=[
             [
                 "Here's an example with some sample lyrics\nAbout dreams and aspirations\nReaching for the stars tonight\nNothing's gonna stop this fight\nWe'll keep on climbing higher\nUntil we touch the sky",
-                "",  # Empty artist name for anonymous example
-                "Dreams Tonight"
+                "",  # Empty artist name
+                "Dreams Tonight",
+                True  # Use Modal if available
             ],
             [
                 "Love is in the air tonight\nHearts beating as one\nDancing under moonlight\nUntil the morning sun\nThis feeling never ends\nLove conquers all",
                 "Romantic Vibes",
-                "Moonlight Dance"
+                "Moonlight Dance",
+                False  # Use local generation
             ]
         ],
-        inputs=[lyrics_input, artist_input, title_input],
+        inputs=[lyrics_input, artist_input, title_input, modal_toggle],
         label="💡 Try these examples"
     )
+    
+    # Add footer with info
+    gr.Markdown("""
+    ---
+    **💡 Tips:**
+    - **AI Generation:** Uses GPT-2 on Modal for creative, context-aware content
+    - **Local Templates:** Fast, rule-based generation that works offline
+    - **Modal Credits:** Can be used for LLM inference and GPU-intensive functions ([Modal Docs](https://modal.com/docs))
+    - **Best Results:** Include rich, descriptive lyrics for better theme detection
+    """)
 
 if __name__ == "__main__":
     demo.launch()