Update app.py

app.py

@@ -24,6 +24,7 @@ from openai import OpenAI
 # Load Hugging Face token if needed
 hf_token = os.getenv("HF_TOKEN")
 openai_api_key = os.getenv("OPENAI_API_KEY")
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
 
 system_prompt_t2v_path = "assets/system_prompt_t2v.txt"
 system_prompt_i2v_path = "assets/system_prompt_i2v.txt"
@@ -204,6 +205,7 @@ pipeline = XoraVideoPipeline(
 
 
 def generate_video_from_text(
+    secret_token="",
     prompt="",
     enhance_prompt_toggle=False,
     negative_prompt="",
@@ -216,6 +218,10 @@ def generate_video_from_text(
     num_frames=121,
     progress=gr.Progress(),
 ):
+    if secret_token != SECRET_TOKEN:
+        raise gr.Error(
+            f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
+
     if len(prompt.strip()) < 50:
         raise gr.Error(
             "Prompt must be at least 50 characters long. Please provide more details for the best results.",
@@ -422,64 +428,13 @@ def create_advanced_options():
 
 # Define the Gradio interface with tabs
 with gr.Blocks(theme=gr.themes.Soft()) as iface:
-    with gr.Row(elem_id="title-row"):
-        gr.Markdown(
-            """
-        <div style="text-align: center; margin-bottom: 1em">
-            <h1 style="font-size: 2.5em; font-weight: 600; margin: 0.5em 0;">Video Generation with LTX Video</h1>
-        </div>
-        """
-        )
-    with gr.Row(elem_id="title-row"):
-        gr.HTML(  # add technical report link
-            """
-        <div style="display:flex;column-gap:4px;">
-            <a href="https://github.com/Lightricks/LTX-Video">
-                <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
-            </a>
-            <a href="https://github.com/Lightricks/ComfyUI-LTXVideo">
-                <img src='https://img.shields.io/badge/GitHub-ComfyUI-blue'>
-            </a>
-            <a href="http://www.lightricks.com/ltxv">
-                <img src="https://img.shields.io/badge/Project-Page-green" alt="Follow me on HF">
-            </a>
-            <a href="https://huggingface.co/spaces/Lightricks/LTX-Video-Playground?duplicate=true">
-                <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-sm.svg" alt="Duplicate this Space">
-            </a>
-            <a href="https://huggingface.co/Lightricks">
-                <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/follow-me-on-HF-sm-dark.svg" alt="Follow me on HF">
-            </a>
+    gr.HTML("""
+        <div style="z-index: 100; position: fixed; top: 0px; right: 0px; left: 0px; bottom: 0px; width: 100%; height: 100%; background: white; display: flex; align-items: center; justify-content: center; color: black;">
+        <div style="text-align: center; color: black;">
+        <p style="color: black;">This space is a headless component of the cloud rendering engine used by AiTube.</p>
+        <p style="color: black;">It is not available for public use, but you can use the <a href="https://huggingface.co/spaces/Lightricks/LTX-Video-Playground" target="_blank">original space</a>.</p>
         </div>
-        """
-        )
-    with gr.Accordion(
-        " 📖 Tips for Best Results", open=False, elem_id="instructions-accordion"
-    ):
-        gr.Markdown(
-            """
-        📝 Prompt Engineering
-
-        When writing prompts, focus on detailed, chronological descriptions of actions and scenes. Include specific movements, appearances, camera angles, and environmental details - all in a single flowing paragraph. Start directly with the action, and keep descriptions literal and precise. Think like a cinematographer describing a shot list. Keep within 200 words.
-        For best results, build your prompts using this structure:
-
-        - Start with main action in a single sentence
-        - Add specific details about movements and gestures
-        - Describe character/object appearances precisely
-        - Include background and environment details
-        - Specify camera angles and movements
-        - Describe lighting and colors
-        - Note any changes or sudden events
-
-        See examples for more inspiration.
-
-        🎮 Parameter Guide
-
-        - Resolution Preset: Higher resolutions for detailed scenes, lower for faster generation and simpler scenes
-        - Seed: Save seed values to recreate specific styles or compositions you like
-        - Guidance Scale: 3-3.5 are the recommended values
-        - Inference Steps: More steps (40+) for quality, fewer steps (20-30) for speed
-        """
-        )
+        </div>""")
 
     with gr.Tabs():
         # Text to Video Tab
@@ -556,6 +511,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         with gr.TabItem("Image to Video"):
             with gr.Row():
                 with gr.Column():
+                    secret_token = gr.Textbox()
                     img2vid_image = gr.Image(
                         type="filepath",
                         label="Step 1: Upload Input Image",
@@ -640,6 +596,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     txt2vid_generate.click(
         fn=generate_video_from_text,
         inputs=[
+            secret_token,
             txt2vid_prompt,
             txt2vid_enhance_toggle,
             txt2vid_negative_prompt,
@@ -673,6 +630,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     )
 
 if __name__ == "__main__":
-    iface.queue(max_size=64, default_concurrency_limit=1, api_open=False).launch(
-        share=True, show_api=False
+    iface.queue(max_size=32, default_concurrency_limit=1, api_open=False).launch(
+        share=True, show_api=True
     )