fixing async issue

demo.py

@@ -2,9 +2,10 @@ import gradio as gr
 import os
 import tempfile
 import base64
-from main import process
+import asyncio
+from main import process_async
 
-def process_image_and_generate_video(image, prompt):
+async def process_image_and_generate_video_async(image, prompt):
     # Create a temporary directory for intermediate files
     with tempfile.TemporaryDirectory() as temp_dir:
         # Save the uploaded image to a temporary file
@@ -17,25 +18,32 @@ def process_image_and_generate_video(image, prompt):
         data_url = f"data:image/png;base64,{encoded_image}"
         
         # Process the image and generate video
-        # The process function now handles saving to numbered directories
-        result, generated_image_path, video_path = process(data_url, prompt, temp_dir)
+        result, generated_image_path, video_path = await process_async(data_url, prompt, temp_dir)
         
         if result and video_path:
             return video_path
         else:
             return None
 
+def process_image_and_generate_video(image, prompt):
+    return asyncio.run(process_image_and_generate_video_async(image, prompt))
+
 # Create the Gradio interface
 with gr.Blocks(title="Character Video Generation") as demo:
     gr.Markdown("# Character Video Generation")
     gr.Markdown("""
-    * Upload a high-quality image of a person
+    * Upload a high-quality image of a person (PNG, JPG, or JPEG only)
     * Enter a prompt to generate a video
     """)
     
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(type="pil", label="Upload Reference Image")
+            input_image = gr.Image(
+                type="pil",
+                label="Upload Reference Image (PNG, JPG, or JPEG only)",
+                height=512,
+                width=512
+            )
             prompt = gr.Textbox(label="Enter your prompt")
             generate_btn = gr.Button("Generate")