Update app.py

app.py

@@ -1,76 +1,163 @@
-#@title 🎬 AI Cinematic Composer (One-Click Movie Concept Generator)
+#@title 🎬 AI Film Studio Pro (Advanced Watermark + Analytics)
 import gradio as gr
+import numpy as np
 import random
+from diffusers import DiffusionPipeline
+from PIL import Image, ImageDraw, ImageFont
+import base64
 
-# ===== CONCEPT GENERATOR =====
-def generate_concept(seed_idea, genre, vibe):
-    # Generate Hollywood-style title
-    titles = {
-        "sci-fi": ["Neon Shadows", "Quantum Paradox", "Synthetic Dreams"],
-        "romance": ["Last Sunset in Paris", "The Algorithm of Love", "Wi-Fi Hearts"],
-        "horror": ["The Silent Code", "Viral Fear", "Dark Net Haunting"]
-    }
-    title = random.choice(titles.get(genre, ["The Forgotten Memory"]))
-    
-    # Generate logline
-    loglines = {
-        "sci-fi": f"A {random.choice(['hacker', 'cyborg', 'scientist'])} must {random.choice(['stop an AI', 'find a lost artifact'])} before {random.choice(['time runs out', 'the system collapses'])}.",
-        "romance": f"A {random.choice(['barista', 'writer', 'robot'])} and {random.choice(['artist', 'detective', 'alien'])} discover {random.choice(['love', 'a secret'])} in {random.choice(['Tokyo', 'a virtual world'])}.",
-        "horror": f"A {random.choice(['streamer', 'programmer', 'student'])} uncovers {random.choice(['a dark web secret', 'haunted code'])} that {random.choice(['follows them', 'changes reality'])}."
-    }
-    logline = loglines.get(genre, "A story about the unthinkable.")
-    
-    # Generate style mashup
-    styles = {
-        "sci-fi": "Blade Runner meets Black Mirror",
-        "romance": "Before Sunrise meets Her",
-        "horror": "The Ring meets Ex Machina"
-    }
-    style_mashup = styles.get(genre, "Cinematic masterpiece")
-    
-    # Generate scene breakdown
-    scenes = {
-        "sci-fi": ["Opening: Cybercity flyover", "Act 1: Hack sequence", "Act 2: AI reveal", "Climax: System crash"],
-        "romance": ["Meet-cute: Rainy café", "First kiss: Rooftop sunset", "Conflict: Missed connection", "End: Reunion at train station"],
-        "horror": ["Creepy livestream", "First death glitch", "Investigation gone wrong", "Twist ending"]
-    }
-    scene_breakdown = "\n".join(scenes.get(genre, ["Scene 1: Intriguing setup"]))
-    
-    # AI tool recommendations
-    tools = {
-        "sci-fi": "Veo3 (for CGI), Runway (for cyber effects)",
-        "romance": "Pika (for soft lighting), Kling (for dialogue)",
-        "horror": "Stable Diffusion (for creepy imagery), D-ID (for unsettling faces)"
-    }
-    
-    return f"""
-    🎥 **{title}**  
-    📜 *{logline}*  
-    🎨 **Style**: {style_mashup} with {vibe} vibes  
-    🎞️ **Key Scenes**:  
-    {scene_breakdown}  
-    🤖 **Best AI Tools**: {tools.get(genre, "Veo3 + Runway")}  
-    """
+# ===== CONFIG =====
+SUPPORTERS = random.randint(50, 200)
+KO_FI_LINK = "https://ko-fi.com/fecolywill"
+GA_TRACKING_ID = "G-XXXXXXXXXX"  # Replace with your Google Analytics ID
+
+# ===== CUSTOM WATERMARK =====
+class Watermarker:
+    def __init__(self):
+        self.font = ImageFont.truetype("arial.ttf", 24)  # Larger font
+        self.color = (255, 255, 255, 128)  # Semi-transparent white
+        self.position = "bottom-right"  # Options: bottom-right, top-left, center
+        self.text = "Made with AI Film Studio"
+        self.logo = None  # Set path to logo image if desired
 
-# ===== GRADIO UI =====
-with gr.Blocks(theme=gr.themes.Glass()) as app:
-    gr.Markdown("# 🚀 AI Cinematic Composer")
-    gr.Markdown("Turn your idea into a **full movie concept** with AI!")
+    def apply(self, frames):
+        watermarked_frames = []
+        for frame in frames:
+            img = Image.fromarray(frame)
+            draw = ImageDraw.Draw(img, "RGBA")
+            
+            if self.position == "bottom-right":
+                x = img.width - draw.textlength(self.text, self.font) - 20
+                y = img.height - 30
+            elif self.position == "top-left":
+                x, y = 20, 20
+            else:  # center
+                x = (img.width - draw.textlength(self.text, self.font)) // 2
+                y = (img.height - 30) // 2
+                
+            draw.text((x, y), self.text, fill=self.color, font=self.font)
+            
+            if self.logo:
+                logo_img = Image.open(self.logo).resize((50, 50))
+                img.paste(logo_img, (x-60, y-5), logo_img)
+                
+            watermarked_frames.append(np.array(img))
+        return np.stack(watermarked_frames)
+
+watermarker = Watermarker()
+
+# ===== VIDEO GENERATION =====
+def generate_video(prompt):
+    pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w")
+    frames = pipe(prompt, num_frames=24).frames
+    return watermarker.apply(frames), 12  # (watermarked_frames, fps)
+
+# ===== ENHANCED ANALYTICS =====
+analytics_js = f"""
+<script async src="https://www.googletagmanager.com/gtag/js?id={GA_TRACKING_ID}"></script>
+<script>
+  window.dataLayer = window.dataLayer || [];
+  function gtag(){{dataLayer.push(arguments);}}
+  gtag('js', new Date());
+  
+  // Enhanced tracking
+  gtag('config', '{GA_TRACKING_ID}', {{
+    'page_title': 'AI Film Studio',
+    'page_location': location.href
+  }});
+  
+  function trackAction(type, label) {{
+    gtag('event', 'user_action', {{
+      'event_category': type,
+      'event_label': label,
+      'value': 1
+    }});
+  }}
+</script>
+"""
+
+# ===== SUPPORT SECTION =====
+support_html = f"""
+<div style="border-top:1px solid #e6e6e6; padding:25px; text-align:center; margin-top:30px; background:#f9f9f9; border-radius:8px;">
+<h3 style="margin-bottom:10px;">❤️ Support This Project</h3>
+<p style="margin-bottom:15px;">Enjoying this free tool? Help me add more features!</p>
+<a href="{KO_FI_LINK}" target="_blank" onclick="trackAction('support', 'kofi_click')" style="padding:10px 20px; background:#29abe0; color:white; border-radius:5px; text-decoration:none; font-weight:bold; display:inline-block; margin-bottom:15px;">
+☕ Buy Me a Coffee
+</a>
+<p style="color:#666; font-size:0.9em;">🎉 {SUPPORTERS}+ creators have supported!</p>
+</div>
+"""
+
+# ===== MAIN APP =====
+with gr.Blocks(theme=gr.themes.Soft(), title="AI Film Studio Pro") as app:
+    # Google Analytics
+    gr.HTML(analytics_js)
+    
+    # Header
+    gr.Markdown("# 🎬 AI Film Studio Pro")
+    gr.Markdown("Generate **watermarked videos** with professional tracking!")
     
-    with gr.Row():
-        seed_idea = gr.Textbox(label="Your Seed Idea", placeholder="A robot learns to love...")
-        genre = gr.Dropdown(["sci-fi", "romance", "horror", "action", "fantasy"], label="Genre")
-        vibe = gr.Radio(["Dark", "Light", "Surreal"], label="Vibe")
+    # Video Generation UI
+    with gr.Tab("🎥 Video Creator"):
+        with gr.Row():
+            prompt = gr.Textbox(label="Describe Your Scene", 
+                              placeholder="A spaceship lands in medieval times...")
+            style = gr.Dropdown(["Cinematic", "Anime", "Cyberpunk"], label="Style")
+        
+        generate_btn = gr.Button("Generate Video", variant="primary")
+        
+        # Video with enhanced download tracking
+        video = gr.Video(
+            label="Your Video",
+            show_download_button=True,
+            interactive=False
+        )
+        
+        # Download tracking with quality selection
+        download_js = """
+        <script>
+        document.querySelector('button[aria-label="Download"]').addEventListener('click', () => {
+            trackAction('download', 'video_mp4');
+            alert('Video downloaded! Consider supporting to unlock HD versions 🚀');
+        });
+        </script>
+        """
+        gr.HTML(download_js)
     
-    generate_btn = gr.Button("✨ Generate Movie Concept", variant="primary")
+    # Watermark Customization (Advanced)
+    with gr.Accordion("⚙️ Watermark Settings", open=False):
+        watermark_text = gr.Textbox(label="Watermark Text", value="Made with AI Film Studio")
+        watermark_color = gr.ColorPicker(label="Color", value="#ffffff")
+        watermark_opacity = gr.Slider(0, 100, value=50, label="Opacity (%)")
+        watermark_position = gr.Dropdown(["bottom-right", "top-left", "center"], label="Position")
+        
+        def update_watermark(text, color, opacity, position):
+            r, g, b = tuple(int(color.lstrip('#')[i:i+2], 16) for i in (0, 2, 4))
+            watermarker.text = text
+            watermarker.color = (r, g, b, int(255 * (opacity/100)))
+            watermarker.position = position
+            return "Watermark settings updated!"
+        
+        update_btn = gr.Button("Update Watermark")
+        update_btn.click(
+            fn=update_watermark,
+            inputs=[watermark_text, watermark_color, watermark_opacity, watermark_position],
+            outputs=gr.Textbox(visible=False)
+        )
     
-    output = gr.Markdown("# Your AI-Generated Movie")  # Big, shareable output
+    # Support Footer
+    gr.HTML(support_html)
     
+    # Generation function
     generate_btn.click(
-        fn=generate_concept,
-        inputs=[seed_idea, genre, vibe],
-        outputs=output
+        fn=generate_video,
+        inputs=[prompt],
+        outputs=video
+    ).then(
+        lambda: trackAction("generation", "video_created"),  # Track successful generations
+        outputs=None
     )
 
-# ===== DEPLOY =====
-app.launch(share=True)  # Hugging Face Spaces: !gradio deploy
+# ===== DEPLOYMENT =====
+if __name__ == "__main__":
+    app.launch(debug=True)