refactor: integrate Modal API for video generation and update LoRA configurations

app.py

@@ -10,7 +10,6 @@ from google.cloud import storage
 import json
 from pathlib import Path
 import mimetypes
-from workflow_handler import WanVideoWorkflow
 from video_config import MODEL_FRAME_RATES, calculate_frames
 import asyncio
 from openai import OpenAI
@@ -21,71 +20,76 @@ from google.oauth2 import service_account
 dotenv.load_dotenv()
 
 SCRIPT_DIR = Path(__file__).parent
-CONFIG_PATH = SCRIPT_DIR / "config.json"
-WORKFLOW_PATH = SCRIPT_DIR / "wani2v.json"
+
+# Modal configuration
+MODAL_ENDPOINT = os.getenv('FAL_MODAL_ENDPOINT')
+MODAL_AUTH_TOKEN = os.getenv('MODAL_AUTH_TOKEN')
 
 loras = [
-    {
-      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/69e1adfc-b99a-4559-b745-f193a1bca0e2.gif",
-      "id": "c8972c6d-ab8a-4988-9a9d-38082264ef22",
-      "title": "Jumpscare",
+   {
+      "image": "https://huggingface.co/Remade-AI/Crash-zoom-out/resolve/main/example_videos/1.gif",
+      "id": "44c05ca1-422d-4cd4-8508-acadb6d0248c",
+      "title": "Crash Zoom Out ",
       "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
     },
-    {
-      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/2dede2c5-38e0-4acb-9e99-027e804a0455.gif",
-      "id": "d7cbf9b4-82cd-4a94-ba2f-040e809635fa",
-      "title": "Angry",
-      "example_prompt": "The video starts with a man looking at the camera with a neutral face. Then his facial expression changes to 4ngr23 angry face, and he begins to yell with clenched fists. "
+         {
+      "image": "https://huggingface.co/Remade-AI/Crash-zoom-in/resolve/main/example_videos/1.gif",
+      "id": "34a80641-4702-4c1c-91bf-c436a59c79cb",
+      "title": "Crash Zoom In ",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
     },
-    
-    {
-      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/7ba11998-2421-4ae1-8cf8-47b076387c2e.gif",
-      "id": "e17959c4-9fa5-4e5b-8f69-d1fb01bbe4fa",
-      "title": "Cartoon Jaw Drop",
-      "example_prompt": "The video shows Pluto the dog, wearing a red collar, who is smiling wide, then his mouth transforms into a dr0p_j88 comical jaw drop, extending down in a long, rectangular shape, and revealing his tongue and teeth."
+     {
+      "image": "https://huggingface.co/Remade-AI/Car-chase/resolve/main/example_videos/2.gif",
+      "id": "8b36b7fe-0a0b-4849-b0ed-d9a51ff0cc85",
+      "title": "Car Chase",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
     },
-    {
-      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/46b5c7a3-60d5-469e-9454-9d16bf20afe4.gif",
-      "id": "687255bb-959e-4422-bdbb-5aba93c7c180",
-      "title": "Kissing",
-      "example_prompt": "A man with a beard is shown smiling. A woman comes into the scene and starts passionately k144ing kissing the man."
+     {
+      "image": "https://huggingface.co/Remade-AI/Crane-down/resolve/main/example_videos/2.gif",
+      "id": "f26db0b7-1c26-4587-b2b5-1cfd0c51c5b3",
+      "title": "Crane Down ",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
     },
-    {
-      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/9f11b800-25c1-42f9-a687-97d706fef06d.gif",
-      "id": "4ac2fb4e-5ca2-4338-a59c-549167f5b6d0",
-      "title": "Laughing",
-      "example_prompt": "A [Subject] is smiling at the camera. He/she then begins l4a6ing laughing."
+     {
+      "image": "https://huggingface.co/Remade-AI/Crane_up/resolve/main/example_videos/1.gif",
+      "id": "07c5e22b-7028-437c-9479-6eb9a50cf993",
+      "title": "Crane Up ",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
     },
-    {
-      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/7cc38d79-f09e-4a5d-9e17-65867f8fd3a7.gif",
-      "id": "bcc4163d-ebda-4cdc-b153-7136cdbf563a",
-      "title": "Crying",
-      "example_prompt": "The video starts with a [Subject] with a solemn expression. Then a tear rolls down his/her cheek, as he/she is cr471ng crying."
+
+     {
+      "image": "https://huggingface.co/Remade-AI/Crane_over_the_head/resolve/main/example_videos/1.gif",
+      "id": "9393f8f4-abe6-4aa7-ba01-0b62e1507feb",
+      "title": "Crane Overhead ",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
     },
-    {
-      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/62256895-10b2-4bd1-8ffe-4ecec530c4ec.gif",
-      "id": "13093298-652c-4df8-ba28-62d9d5924754",
-      "title": "Take a selfie with your younger self",
-      "example_prompt": "The video starts with the a man with a beard smiling at the camera, then s31lf13 taking a selfie with their younger self, and the younger self appears next to him with similar facial features and eye color. The younger self wears a white t-shirt and has a cream white jacket. The younger self is smiling slightly."
+    
+     {
+      "image": "https://huggingface.co/Remade-AI/matrix-shot/resolve/main/example_videos/1.gif",
+      "id": "219ad5ad-8f23-48dc-b098-b8e6d9fbe6c0",
+      "title": "Matrix Shot ",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
+    },
+     {
+      "image": "https://huggingface.co/Remade-AI/360-Orbit/resolve/main/example_videos/1.gif",
+      "id": "aaa3e820-5d94-4612-9488-0c9a1b2f5843",
+      "title": "360 Orbit ",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
     },
-    {
-      "image": "https://huggingface.co/Remade-AI/Squish/resolve/main/example_gifs/person_squish.gif",
-      "id": "06ce6840-f976-4963-9644-b6cf7f323f90",
-      "title": "Squish",
-      "example_prompt": "In the video, a miniature rodent is presented. The rodent is held in a person's hands. The person then presses on the rodent, causing a sq41sh squish effect. The person keeps pressing down on the rodent, further showing the sq41sh squish effect.",
-    },  
-    {
-      "image": "https://huggingface.co/Remade-AI/Rotate/resolve/main/example_videos/chair-rotate.gif",
-      "id": "4ac08cfa-841e-4aa9-9022-c3fc80fb6ef4",
-      "title": "Rotate",
-      "example_prompt": "The video shows an elderly Asian man's head and shoulders with blurred background, performing a r0t4tion 360 degrees rotation.",
+         {
+      "image": "https://huggingface.co/Remade-AI/Arc_shot/resolve/main/example_videos/1.gif",
+      "id": "a5949ee3-61ea-4a18-bd4d-54c855f5401c",
+      "title": "Arc Shot ",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
+    },
+         {
+      "image": "https://huggingface.co/Remade-AI/Hero-run/resolve/main/example_videos/1.gif",
+      "id": "36b9edf7-31d7-47d3-ad3b-e166fb3a9842",
+      "title": "Hero Run ",
+      "example_prompt": "The video shows a man with a slight smile, then the j432mpscare jumpscare occurs, revealing a distorted and monstrous face with glowing red eyes, filling the frame and accompanied by a loud scream."
     },
-    {
-      "image": "https://huggingface.co/Remade-AI/Cakeify/resolve/main/example_gifs/timberland_cakeify.gif",
-      "id": "b05c1dc7-a71c-4d24-b512-4877a12dea7e",
-      "title": "Cakeify",
-      "example_prompt": "The video opens on a woman. A knife, held by a hand, is coming into frame and hovering over the woman. The knife then begins cutting into the woman to c4k3 cakeify it. As the knife slices the woman open, the inside of the woman is revealed to be cake with chocolate layers. The knife cuts through and the contents of the woman are revealed."
-    },    
+
+
 ]
 
 # Initialize Supabase client with async support
@@ -241,9 +245,21 @@ def build_lora_prompt(subject, lora_id):
             raise ValueError(f"Unknown LoRA ID: {lora_id} and no example prompt available")
 
 def poll_generation_status(generation_id):
-    """Poll generation status from database"""
+    """Poll generation status from Modal backend or database"""
     try:
-        # Query the database for the current status
+        # First try to get status from Modal backend if available
+        if MODAL_ENDPOINT:
+            try:
+                response = requests.get(
+                    f"{MODAL_ENDPOINT}/fal-effects/status?generation_id={generation_id}",
+                    headers=get_modal_auth_headers()
+                )
+        
+            except Exception as e:
+                print(f"Error polling Modal backend: {e}")
+
+         
+
         response = supabase.table('generations') \
             .select('*') \
             .eq('generation_id', generation_id) \
@@ -280,7 +296,7 @@ async def moderate_prompt(prompt: str) -> dict:
         
         # Additional checks for keywords related to minors or inappropriate content
         keywords = [
-            "child", "kid", "minor", "teen", "young", "baby", "infant", "underage",
+            "child", "kid", "minor", "teen", "baby", "infant", "underage",
             "naked", "nude", "nsfw", "porn", "xxx", "sex", "explicit",
             "inappropriate", "adult content"
         ]
@@ -345,27 +361,18 @@ async def moderate_image(image_path: str) -> dict:
             
             # Get likelihood scores
             adult_score = likelihood_values[safe_search.adult]
-            racy_score = likelihood_values[safe_search.racy]
-            violence_score = likelihood_values[safe_search.violence]
-            medical_score = likelihood_values[safe_search.medical]
+  
             
             # Determine if content is NSFW according to Vision API
             vision_reasons = []
             if adult_score >= 3:  # LIKELY or VERY_LIKELY
                 vision_reasons.append("adult content")
-            if racy_score >= 3:  # LIKELY or VERY_LIKELY
-                vision_reasons.append("suggestive content")
-            if violence_score >= 3:  # LIKELY or VERY_LIKELY
-                vision_reasons.append("violent content")
-
+   
                 
             # Print Vision API results
             print("Google Cloud Vision API Results:")
             print(f"Adult: {vision.Likelihood(safe_search.adult).name}")
-            print(f"Racy: {vision.Likelihood(safe_search.racy).name}")
-            print(f"Violence: {vision.Likelihood(safe_search.violence).name}")
-            print(f"Medical: {vision.Likelihood(safe_search.medical).name}")
-            
+  
         except Exception as vision_error:
             print(f"Error with Vision API: {vision_error}")
             vision_reasons = []  # Continue with OpenAI check if Vision API fails
@@ -485,15 +492,8 @@ Be extremely cautious - if there's any doubt, mark it as NSFW."""
             "reason": "Failed to verify content safety - please try again"
         }
 
-async def generate_video(input_image, subject, duration, selected_index, progress=gr.Progress()):
+async def generate_video(input_image, subject, selected_index, progress=gr.Progress()):
     try:
-        # Initialize workflow handler with explicit paths
-        workflow_handler = WanVideoWorkflow(
-            supabase,
-            config_path=str(CONFIG_PATH),
-            workflow_path=str(WORKFLOW_PATH)
-        )
-        
         # Check if the input is a URL (example image) or a file path (user upload)
         if input_image.startswith('http'):
             # It's already a URL, use it directly
@@ -502,12 +502,8 @@ async def generate_video(input_image, subject, duration, selected_index, progres
             # It's a file path, upload to GCS
             image_url = upload_to_gcs(input_image)
         
-        # Map duration selection to actual seconds
-        duration_mapping = {
-            "Short (3s)": 3,
-            "Long (5s)": 5
-        }
-        video_duration = duration_mapping[duration]
+        # Hardcode duration to 3 seconds
+        video_duration = 5
         
         # Get LoRA config
         lora_config = next((lora for lora in loras if lora["id"] == selected_index), None)
@@ -517,58 +513,59 @@ async def generate_video(input_image, subject, duration, selected_index, progres
         # Generate unique ID
         generation_id = str(uuid.uuid4())
 
-        # Update workflow
-        prompt = build_lora_prompt(subject, selected_index)
-        workflow_handler.update_prompt(prompt)
-        workflow_handler.update_input_image(image_url)
-        await workflow_handler.update_lora(lora_config)
-        workflow_handler.update_length(video_duration)
-        workflow_handler.update_output_name(generation_id)
-
-        # Get final workflow
-        workflow = workflow_handler.get_workflow()
-
-        # Store generation data in Supabase
-        generation_data = {
-            "generation_id": generation_id,
-            "user_id": "anonymous",
-            "status": "queued",
-            "progress": 0,
-            "worker_id": None,
-            "created_at": datetime.datetime.utcnow().isoformat(),
-            "message": {
-                "generationId": generation_id,
-                "workflow": {
-                    "prompt": workflow
-                }
-            },
-            "metadata": {
-                "prompt": {
-                    "original": subject,
-                    "enhanced": subject
-                },
-                "lora": {
-                    "id": selected_index,
-                    "strength": 1.0,
-                    "name": lora_config["title"]
-                },
-                "workflow": "img2vid",
-                "dimensions": None,
-                "input_image_url": image_url,
-                "video_length": {"duration": video_duration},
-            },
-            "error": None,
-            "output_url": None,
-            "batch_id": None,
-            "platform": "huggingface"
-        }
-        
-        # Remove await - the execute() method returns the response directly
-        response = supabase.table('generations').insert(generation_data).execute()
-        print(f"Stored generation data with ID: {generation_id}")
-        
-        # Return generation ID for tracking
-        return generation_id
+        # Build prompt for the LoRA
+        prompt = subject
+        
+        # Check if Modal endpoint is configured
+        if not MODAL_ENDPOINT:
+            raise ValueError("Modal endpoint not configured - FAL_MODAL_ENDPOINT environment variable not found")
+
+        # Calculate frames based on duration and frame rate
+        frame_rate = 16  # WanVideo frame rate
+        num_frames = calculate_frames(video_duration, frame_rate)
+        
+        print(f"Sending request to Modal backend: {MODAL_ENDPOINT}/fal-effects")
+        
+        # Make POST request to the modal backend
+        response = requests.post(f"{MODAL_ENDPOINT}/fal-effects", 
+            headers=get_modal_auth_headers(),
+            json={
+                "user_id": "anonymous",  # Since we don't have user auth in this app
+                "image_url": image_url,
+                "subject": prompt,  # Use the built prompt as subject
+                "aspect_ratio": "16:9",  # Default aspect ratio for effects
+                "num_frames": 81,
+                "frames_per_second": frame_rate,
+                "length": str(5),
+                "enhance_prompt": False,
+                "lora_scale": 1.0,
+                "turbo_mode": False,
+                "lora_id": selected_index,
+                "lora_strength": 1.0,
+                "generation_ids": [generation_id]
+            }
+        )
+
+        if not response.ok:
+            error_text = response.text
+            try:
+                error_json = response.json()
+                error_message = error_json.get('detail') or error_json.get('error') or 'Failed to create generation'
+            except:
+                error_message = f'Failed to create generation: {error_text}'
+            raise ValueError(error_message)
+
+        result = response.json()
+        print(f"Modal backend response: {result}")
+        
+        # Extract generation ID from response
+        if 'generation_id' in result:
+            return result['generation_id']
+        elif 'id' in result:
+            return result['id']
+        else:
+            # Fallback to our generated ID if the response doesn't contain one
+            return generation_id
         
     except Exception as e:
         print(f"Error in generate_video: {e}")
@@ -579,7 +576,7 @@ def update_selection(evt: gr.SelectData):
   sentence = f"Selected LoRA: {selected_lora['title']}"
   return selected_lora['id'], sentence
 
-async def handle_generation(image_input, subject, duration, selected_index, progress=gr.Progress(track_tqdm=True)):
+async def handle_generation(image_input, subject, selected_index, progress=gr.Progress(track_tqdm=True)):
     try:
         if selected_index is None:
             raise gr.Error("You must select a LoRA before proceeding.")
@@ -588,28 +585,36 @@ async def handle_generation(image_input, subject, duration, selected_index, prog
         prompt_moderation = await moderate_prompt(subject)
         print(f"Prompt moderation result: {prompt_moderation}")
         if prompt_moderation["isNSFW"]:
+            # Re-enable button on error
+            yield None, None, gr.update(visible=False), gr.update(value="Generate", interactive=True)
             raise gr.Error(f"Content moderation failed: {prompt_moderation['reason']}")
         
         # Then, moderate the image
         image_moderation = await moderate_image(image_input)
         print(f"Image moderation result: {image_moderation}")
         if image_moderation["isNSFW"]:
+            # Re-enable button on error
+            yield None, None, gr.update(visible=False), gr.update(value="Generate", interactive=True)
             raise gr.Error(f"Content moderation failed: {image_moderation['reason']}")
         
         # Finally, check the combination
         combined_moderation = await moderate_combined(subject, image_input)
         print(f"Combined moderation result: {combined_moderation}")
         if combined_moderation["isNSFW"]:
+            # Re-enable button on error
+            yield None, None, gr.update(visible=False), gr.update(value="Generate", interactive=True)
             raise gr.Error(f"Content moderation failed: {combined_moderation['reason']}")
             
         # Generate the video and get generation ID
-        generation_id = await generate_video(image_input, subject, duration, selected_index)
+        generation_id = await generate_video(image_input, subject, selected_index)
         
         # Poll for status updates
         while True:
             generation = poll_generation_status(generation_id)
             
             if not generation:
+                # Re-enable button on error
+                yield None, None, gr.update(visible=False), gr.update(value="Generate", interactive=True)
                 raise ValueError(f"Generation {generation_id} not found")
                 
             # Update progress
@@ -619,20 +624,24 @@ async def handle_generation(image_input, subject, duration, selected_index, prog
                 
                 # Check status
                 if generation['status'] == 'completed':
-                    # Final yield with completed video
-                    yield generation['output_url'], generation_id, gr.update(visible=False)
+                    # Final yield with completed video and re-enabled button
+                    yield generation['output_url'], generation_id, gr.update(visible=False), gr.update(value="Generate", interactive=True)
                     break  # Exit the loop
                 elif generation['status'] == 'error':
+                    # Re-enable button on error
+                    yield None, None, gr.update(visible=False), gr.update(value="Generate", interactive=True)
                     raise ValueError(f"Generation failed: {generation.get('error')}")
                 else:
-                    # Yield progress update
-                    yield None, generation_id, gr.update(value=progress_bar, visible=True)
+                    # Yield progress update with button still disabled
+                    yield None, generation_id, gr.update(value=progress_bar, visible=True), gr.update(value="Generating...", interactive=False)
             
             # Wait before next poll
             await asyncio.sleep(2)
             
     except Exception as e:
         print(f"Error in handle_generation: {e}")
+        # Re-enable button on any error
+        yield None, None, gr.update(visible=False), gr.update(value="Generate", interactive=True)
         raise e
 
 css = '''
@@ -711,7 +720,7 @@ css = '''
     pointer-events: none;
 }
 .discord-locked::after {
-    content: "🔒 Discord members only";
+    content: "🔒 Remade Canvas exclusive";
     position: absolute;
     top: 50%;
     left: 50%;
@@ -818,25 +827,20 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
     selected_index = gr.State(None)
     current_generation_id = gr.State(None)
     
-    # Updated title with April Fool's theme
-    gr.Markdown("# Remade AI - April Fool's Edition: Wan 2.1 I2V Effects")
-    
-    # Insert an April Fool's themed banner at the top
-    april_banner = gr.HTML(
-        """
-        <div style="background-color: #8a0ee3; padding: 15px; border-radius: 8px; text-align: center; font-size: 1.5em; margin-bottom: 15px;">
-            🎉 Happy April Fool's Day! Enjoy some playful pranks and fun effects! 🎉
-        </div>
-        """
-    )
+    # Updated title with Remade Canvas theme
+    gr.Markdown("# Remade AI - Open Source Camera Controls")
     
-    # Optionally, update the Discord banner for an April Fool's twist
-    discord_banner = gr.HTML(
+    # Updated Remade Canvas callout
+    gr.HTML(
         """
         <div class="discord-banner">
-            <h3>✨ Unlock Premium April Fool's Pranks! ✨</h3>
-            <p>Join our Discord community for exclusive prank effects, surprise features, and more playful fun!</p>
-            <a href="https://remade.ai/join-discord?utm_source=Huggingface&utm_medium=Social&utm_campaign=hugginface_space&utm_content=april_fools" target="_blank">Join Discord Now</a>
+            <h3>🚀 Unlock 100s of AI Video Effects! 🎬</h3>
+            <p>Access Remade Canvas with Veo, Kling, and hundreds of professional video effects. Create cinematic content with the most advanced AI video models!</p>
+            <a href="https://app.remade.ai?utm_source=Huggingface&utm_medium=Social&utm_campaign=hugginface_space&utm_content=canvas_effects" target="_blank">Try Remade Canvas</a>
+            <div style="margin-top: 15px; padding-top: 15px; border-top: 1px solid rgba(255,255,255,0.7);">
+                <p style="font-size: 0.9em; margin-bottom: 10px;">Join our community for updates and tips:</p>
+                <a href="https://remade.ai/join-discord?utm_source=Huggingface&utm_medium=Social&utm_campaign=hugginface_space&utm_content=canvas_effects" target="_blank" style="background-color: rgba(255,255,255,0.2); font-size: 0.9em; padding: 8px 16px;">Discord Community</a>
+            </div>
         </div>
         """
     )
@@ -856,11 +860,11 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
                 object_fit="contain"
             )
             
-            # Updated Discord/prank callout
+            # Updated Discord/camera controls callout
             gr.HTML(
                 """
                 <div class="discord-feature">
-                    <span class="discord-feature-title">✨ Discord Members:</span> Get access to even more mischievous effects beyond these samples!
+                    <span class="discord-feature-title">🎬 Remade Canvas:</span> Access 100s of effects including Veo, Kling, and advanced camera controls beyond these samples!
                 </div>
                 """
             )
@@ -888,21 +892,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
 
             subject = gr.Textbox(label="Describe your subject", placeholder="Cat toy")
             
-            duration = gr.Radio(
-                ["Short (3s)"],
-                label="Duration",
-                value="Short (3s)"
-            )
-            
-            # Updated Discord feature callout for additional playful messaging
-            gr.HTML(
-                """
-                <div class="discord-feature">
-                    <span class="discord-feature-title">⏱️ Discord Members:</span> Enjoy extended pranks and video durations on our Discord!
-                </div>
-                """
-            )
-            
             with gr.Row():
                 button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
                 audio_button = gr.Button("Add Audio 🔒", interactive=False)
@@ -911,7 +900,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
             warning_message = gr.HTML(
                 """
                 <div class="warning-message">
-                    ⚠️ Please DO NOT refresh the page during generation. Our pranksters are hard at work!
+                    ⚠️ Please DO NOT refresh the page during generation. Processing camera controls takes time for best quality!
                 </div>
                 """,
                 visible=True
@@ -920,7 +909,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
             gr.HTML(
                 """
                 <div class="discord-feature">
-                    <span class="discord-feature-title">⚡ Discord Members:</span> Get faster (and prankier) generation speeds!
+                    <span class="discord-feature-title">⚡ Remade Canvas:</span> Get faster generation speeds and access to Veo, Kling, and 100s of premium effects!
                 </div>
                 """
             )
@@ -983,24 +972,42 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
         if image_input is None:
             raise gr.Error("Please upload an image or select an example image.")
     
+    # Function to immediately disable button
+    def start_generation():
+        return gr.update(value="Generating...", interactive=False)
+    
     # Use gr.on for the button click with validation
     button.click(
         fn=check_inputs,
         inputs=[subject, image_input, selected_index],
         outputs=None,
+    ).success(
+        fn=start_generation,
+        inputs=None,
+        outputs=[button]
     ).success(
         fn=handle_generation,
-        inputs=[image_input, subject, duration, selected_index],
-        outputs=[output, current_generation_id, progress_bar]
+        inputs=[image_input, subject, selected_index],
+        outputs=[output, current_generation_id, progress_bar, button]
     )
     
     # Add a click handler for the disabled audio button
     audio_button.click(
-        fn=lambda: gr.Info("Join our Discord to unlock audio generation features!"),
+        fn=lambda: gr.Info("Try Remade Canvas to unlock audio generation and 100s of other effects!"),
         inputs=None,
         outputs=None
     )
 
+def get_modal_auth_headers():
+    """Get authentication headers for Modal API requests"""
+    if not MODAL_AUTH_TOKEN:
+        raise ValueError("MODAL_AUTH_TOKEN environment variable not found")
+    
+    return {
+        'Authorization': f'Bearer {MODAL_AUTH_TOKEN}',
+        'Content-Type': 'application/json'
+    }
+
 if __name__ == "__main__":
     demo.queue(default_concurrency_limit=20)
     demo.launch(ssr_mode=False, share=True)