Update app.py

app.py

@@ -40,8 +40,14 @@ Instagram Caption:
 
 hf_api_key = os.environ["HF_API_KEY"]
 
-def generate_caption(image_file_path, model_name):
+def generate_caption(image_file_path, model_name, user_prompt):
     image_to_text = ImageCaptioner(model_name="Salesforce/blip-image-captioning-base")
+    # รวม prompt หลัก + user input
+    full_prompt_text = base_prompt
+    if user_prompt.strip():
+        full_prompt_text += f"\nMake it themed like: {user_prompt.strip()}"
+
+    prompt_template = [ChatMessage.from_user(full_prompt_text)]
     prompt_builder = ChatPromptBuilder(template=prompt_template, required_variables="*")
     generator = HuggingFaceAPIChatGenerator(
         api_type="serverless_inference_api", 
@@ -69,6 +75,11 @@ with gr.Blocks(theme="soft") as demo:
                 value="deepseek-ai/DeepSeek-V3.1-Terminus", 
                 label="Choose your model!"
                 )
+             # 🆕 ช่องให้ผู้ใช้พิมพ์ธีมของ Caption
+            user_prompt = gr.Textbox(
+                label="Add a theme or tone (e.g. romantic, funny, luxury...)",
+                placeholder="Type here to customize the caption style"
+                
             gr.Examples(["./whale.png", "./rainbow.jpeg", "./selfie.png"], inputs=image, label="Click on any example") 
     submit_btn = gr.Button("✨ Captionate ✨")
     caption = gr.Textbox(label="Caption", show_copy_button=True)