Update app.py

app.py

@@ -25,7 +25,19 @@ sum_llm_chain = tag_chain.chain(llm=local_llm)
 pexels_api_key = os.getenv('pexels_api_key')
 
 def pred(product_name, orientation):
-    folder_name, sentences = pexels.generate_videos(product_name, pexels_api_key, orientation, llm_chain, sum_llm_chain)
+    if orientation == "Shorts/Reels/TikTok (1080 x 1920)":
+        orientation = "potrait"
+        height = 1080
+        width = 1920
+    elif orientation == "Youtube Videos (1920 x 1080)":
+        orientation = "landscape"
+        height = 1920
+        width = 1080
+    else :
+        orientation = "square"
+        height = 1080
+        width = 1080
+    folder_name, sentences = pexels.generate_videos(product_name, pexels_api_key, orientation, height, width, llm_chain, sum_llm_chain)
     gc.collect()
     utils.combine_videos(folder_name)
     return ["\n".join(sentences), os.path.join(folder_name, "Final_Ad_Video.mp4")]
@@ -35,9 +47,10 @@ def pred(product_name, orientation):
 
 with gr.Blocks() as demo:
     dimension = gr.Dropdown(
-            ["Potrait", "Landscape", "Square"], label="Video Dimension", info="Choose orientation"
+            ["Shorts/Reels/TikTok (1080 x 1920)", "Facebook/Youtube Videos (1920 x 1080)", "Square (1080 x 1080)"], 
+            label="Video Dimension", info="Choose dimension"
         )
-    product_name = gr.Textbox(label="Product Name")
+    product_name = gr.Textbox(label="product name")
     captions = gr.Textbox(label="captions")
     video = gr.Video()
     btn = gr.Button("Submit")