Update app.py

app.py

@@ -25,7 +25,7 @@ sum_llm_chain = tag_chain.chain(llm=local_llm)
 pexels_api_key = os.getenv('pexels_api_key')
 
 def pred(query):
-    folder_name, sentences = pexels.generate_videos(query, pexel_api_key, 1920, 1080, llm_chain, sum_llm_chain)
+    folder_name, sentences = pexels.generate_videos(query, pexel_api_key, dimension, llm_chain, sum_llm_chain)
     utils.combine_videos(folder_name)
     return {
         'video':folder_name, 
@@ -33,11 +33,14 @@ def pred(query):
     }
 
 with gr.Blocks() as demo:
-    textbox = gr.Textbox(placeholder="Product Name")
+    dimension = gr.Dropdown(
+            ["Potrait", "Landscape", "Square"], label="Video Dimension", info="Choose dimension"
+        )
+    product_name = gr.Textbox(placeholder="Product Name")
     captions = gr.Textbox()
     video = gr.Video()
     btn = gr.Button("Submit")
-    btn.click(pred, inputs=textbox, outputs=[captions,video])
+    btn.click(pred, inputs=[product_name, dimension], outputs=[captions,video])