Update app.py

app.py

@@ -1,5 +1,7 @@
-import gradio as gr
 import os
+import requests
+import gradio as gr
+
 
 loras = [
   {
@@ -19,8 +21,16 @@ loras = [
 ### To not have the code/API public, you can go on "Settings" and add a Secret. The secrets appear here as environemnt variables, e.g.
 ### os.getenv('BACKEND_URL')
 ### os.getenv('API_KEY')
-def generate_video(input_image, prompt, duration, selected_index):
-  pass
+def generate_video(input_image, subject, duration, selected_index):
+    if selected_index == "squish":
+        #I suggest not exposing this to the user for simplicity
+        prompt = f"In the video, a miniature {subject} is presented. The {subject} is held in a person's hands. The person then presses on the {subject}, causing a sq41sh squish effect. The person keeps pressing down on the {subject}, further showing the sq41sh squish effect."
+
+        #do all the things to submit the request
+        #url and heads can come from the env variable
+        response = requests.post(url, data=data, files=files, headers=headers)
+    
+    pass
 
 def update_selection(evt: gr.SelectData):
   selected_lora = loras[evt.index]
@@ -34,19 +44,19 @@ with gr.Blocks() as demo:
     with gr.Row():
       with gr.Column():
         gallery = gr.Gallery(
-          [(item["image"], item["title"]) for item in loras],
-          label="Select LoRA",
-          allow_preview=False,
-          columns=4,
-          elem_id="gallery",
-          show_share_button=False,
-          height=350
+            [(item["image"], item["title"]) for item in loras],
+            label="Select LoRA",
+            allow_preview=False,
+            columns=4,
+            elem_id="gallery",
+            show_share_button=False,
+            height=350
         )
-        input_image = gr.Image()
-        prompt = gr.Textbox(label="Describe your object", placeholder="Cat toy")
+        input_image = gr.Image(type="filepath")
+        subject = gr.Textbox(label="Describe your subject", placeholder="Cat toy")
         duration = gr.Radio(["Short (3s)", "Long (5s)"], label="Duration")
-        
         button = gr.Button("Generate")
+          
       with gr.Column():
         output = gr.Video(interactive=False, label="Output video")
         
@@ -56,7 +66,7 @@ with gr.Blocks() as demo:
     )  
     button.click(
       generate_video,
-      inputs=[input_image, prompt, duration, selected_index],
+      inputs=[input_image, subject, duration, selected_index],
       outputs=[output]
     )