Update app.py

app.py

@@ -4,6 +4,10 @@ import os
 
 hf_token = os.environ.get("HF_TKN")
 
+from style_template import styles
+STYLE_NAMES = list(styles.keys())
+DEFAULT_STYLE_NAME = "Watercolor"
+
 def get_instantID(portrait_in, condition_pose, prompt):
     client = Client("https://fffiloni-instantid.hf.space/", hf_token=hf_token)
     negative_prompt = "(lowres, low quality, worst quality:1.2), (text:1.2), watermark, (frame:1.2), deformed, ugly, deformed eyes, blur, out of focus, blurry, deformed cat, deformed, photo, anthropomorphic cat, monochrome, photo, pet collar, gun, weapon, blue, 3d, drones, drone, buildings in background, green"
@@ -99,11 +103,12 @@ with gr.Blocks(css=css) as demo:
             Generate live camera shot from input face
         </p>
         """)
-        with gr.Group():
-            with gr.Row():
-                with gr.Column():
-                    face_in = gr.Image(type="filepath", label="Face to copy")
-                with gr.Column():
+        
+        with gr.Row():
+            with gr.Column():
+                face_in = gr.Image(type="filepath", label="Face to copy")
+            with gr.Column():
+                with gr.Row():
                     camera_shot = gr.Dropdown(
                         label = "Camera Shot", 
                         info = "Use standard camera shots vocabulary, or drop your custom shot as conditional pose (1280*720 ratio is recommended)",
@@ -112,10 +117,12 @@ with gr.Blocks(css=css) as demo:
                         ],
                         value = "custom"
                     )
-                    condition_shot = gr.Image(type="filepath", label="Custom conditional shot (Optional)")
-                    prompt = gr.Textbox(label="Prompt")
-                    chosen_model = gr.Radio(label="Choose a model", choices=["i2vgen-xl", "stable-video"], value="i2vgen-xl", interactive=False, visible=False)
-                
+                    style = gr.Dropdown(label="Style template", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
+
+                condition_shot = gr.Image(type="filepath", label="Custom conditional shot (Optional)")
+                prompt = gr.Textbox(label="Prompt")
+                chosen_model = gr.Radio(label="Choose a model", choices=["i2vgen-xl", "stable-video"], value="i2vgen-xl", interactive=False, visible=False)
+            
         with gr.Column():
             submit_btn = gr.Button("Submit")
             video_out = gr.Video()