Update app.py

app.py

@@ -209,12 +209,20 @@ def stream_chat(input_image: Image.Image, caption_type: str, caption_tone: str,
     prompt_str = CAPTION_TYPE_MAP[prompt_key][0].format(length=length, word_count=length)
     
     if caption_type == "style_prompt":
-        prompt_str += f" Lens type: {lens_type} ({lens_types_info[lens_type]}). "
-        prompt_str += f"Film stock: {film_stocks_info[film_stock]}). "
-        prompt_str += f"Composition style: {composition_styles_info[composition_style]}). "
-        prompt_str += f"Lighting aspect: {lighting_aspects_info[lighting_aspect]}). "
-        prompt_str += f"Special technique: {special_techniques_info[special_technique]}). "
-        prompt_str += f"Color effect: {color_effects_info[color_effect]})."
+        # Extract the key from the dropdown value
+        lens_type_key = lens_type.split(":")[0].strip() if lens_type else ""
+        film_stock_key = film_stock.split(":")[0].strip() if film_stock else ""
+        composition_style_key = composition_style.split(":")[0].strip() if composition_style else ""
+        lighting_aspect_key = lighting_aspect.split(":")[0].strip() if lighting_aspect else ""
+        special_technique_key = special_technique.split(":")[0].strip() if special_technique else ""
+        color_effect_key = color_effect.split(":")[0].strip() if color_effect else ""
+
+        prompt_str += f" Lens type: {lens_type_key} ({lens_types_info[lens_type_key]}). " if lens_type_key else ""
+        prompt_str += f"Film stock: {film_stock_key} ({film_stocks_info[film_stock_key]}). " if film_stock_key else ""
+        prompt_str += f"Composition style: {composition_style_key} ({composition_styles_info[composition_style_key]}). " if composition_style_key else ""
+        prompt_str += f"Lighting aspect: {lighting_aspect_key} ({lighting_aspects_info[lighting_aspect_key]}). " if lighting_aspect_key else ""
+        prompt_str += f"Special technique: {special_technique_key} ({special_techniques_info[special_technique_key]}). " if special_technique_key else ""
+        prompt_str += f"Color effect: {color_effect_key} ({color_effects_info[color_effect_key]})." if color_effect_key else ""
 
     # Debugging: Print the constructed prompt string
     print(f"Constructed Prompt: {prompt_str}")
@@ -444,10 +452,10 @@ with gr.Blocks(theme="Hev832/Applio", css=css) as demo:
 
                 gr.Markdown("**Note:** Caption tone doesn't affect `rng-tags`, `training_prompt`, and `style_prompt`.")
 
-                run_button = gr.Button("Make My Caption!")
                 
             with gr.Column():
                 output_caption = gr.Textbox(label="Generated Caption")
+                run_button = gr.Button("Make My Caption!")
 
                 # Container for advanced options
                 with gr.Column(visible=False) as advanced_options: