Update app.py

app.py

@@ -1356,7 +1356,90 @@ def create_ui():
         ############# this part is for text to video #############
         ###############################################################################
         
+        with gr.Row(variant="panel") as text2video_model:
+            with gr.Group():
+                with gr.Row():
+                    prompt = gr.Textbox(
+                        label='Prompt'
+                    )
+                with gr.Row():
+                    select_base = gr.Dropdown(
+                        label='Base model',
+                        choices=[
+                            "Cartoon", 
+                            "Realistic",
+                            "3d",
+                            "Anime",
+                        ],
+                        value=base_loaded,
+                        interactive=True
+                    )
+                    select_motion = gr.Dropdown(
+                        label='Motion',
+                        choices=[
+                            ("Default", ""),
+                            ("Zoom in", "guoyww/animatediff-motion-lora-zoom-in"),
+                            ("Zoom out", "guoyww/animatediff-motion-lora-zoom-out"),
+                            ("Tilt up", "guoyww/animatediff-motion-lora-tilt-up"),
+                            ("Tilt down", "guoyww/animatediff-motion-lora-tilt-down"),
+                            ("Pan left", "guoyww/animatediff-motion-lora-pan-left"),
+                            ("Pan right", "guoyww/animatediff-motion-lora-pan-right"),
+                            ("Roll left", "guoyww/animatediff-motion-lora-rolling-anticlockwise"),
+                            ("Roll right", "guoyww/animatediff-motion-lora-rolling-clockwise"),
+                        ],
+                        value="guoyww/animatediff-motion-lora-zoom-in",
+                        interactive=True
+                    )
+                    select_step = gr.Dropdown(
+                        label='Inference steps',
+                        choices=[
+                            ('1-Step', 1), 
+                            ('2-Step', 2),
+                            ('4-Step', 4),
+                            ('8-Step', 8),
+                        ],
+                        value=4,
+                        interactive=True
+                    )
+                    submit = gr.Button(
+                        scale=1,
+                        variant='primary'
+                    )
+            video = gr.Video(
+                label='AnimateDiff-Lightning',
+                autoplay=True,
+                height=512,
+                width=512,
+                elem_id="video_output"
+            )
+        
+            prompt.submit(
+                fn=generate_image,
+                inputs=[prompt, select_base, select_motion, select_step],
+                outputs=video,
+            )
+            submit.click(
+                fn=generate_image,
+                inputs=[prompt, select_base, select_motion, select_step],
+                outputs=video,
+            )
         
+            gr.Examples(
+                examples=[
+                ["Focus: Eiffel Tower (Animate: Clouds moving)"], #Atmosphere Movement Example
+                ["Focus: Trees In forest (Animate: Lion running)"], #Object Movement Example
+                ["Focus: Astronaut in Space"], #Normal
+                ["Focus: Group of Birds in sky (Animate:  Birds Moving) (Shot From distance)"], #Camera distance
+                ["Focus:  Statue of liberty (Shot from Drone) (Animate: Drone coming toward statue)"], #Camera Movement
+                ["Focus: Panda in Forest (Animate: Drinking Tea)"], #Doing Something
+                ["Focus: Kids Playing (Season: Winter)"], #Atmosphere or Season
+                {"Focus: Cars in Street (Season: Rain, Daytime) (Shot from Distance) (Movement: Cars running)"} #Mixture
+            ], 
+                fn=generate_image,
+                inputs=[prompt],
+                outputs=video,
+                cache_examples=True,
+        )
         
         
         ###############################################################################