Update app.py

app.py

@@ -21,7 +21,7 @@ import tempfile, gc, os, shutil, json, time, re
 from urllib.parse import urlparse
 from typing import Optional
 
-# ===== BUILT-IN UTILITY FUNCTIONS =====
+# ===== UTILITIES =====
 def fix_image_url(raw_url_or_path: str, host: Optional[str] = None) -> str:
     if not raw_url_or_path:
         return raw_url_or_path
@@ -170,7 +170,7 @@ def safe_generate_caption_direct(image, system_prompt, user_prompt, max_chars=12
         gc.collect()
         return f"❌ Error: {str(e)[:200]}"
 
-# ===== GENERATION FUNCTIONS =====
+# ===== CAPTION FUNCTIONS =====
 @spaces.GPU(duration=60)
 @torch.no_grad()
 def generate_caption_1(image, system1, user1):
@@ -263,7 +263,7 @@ def export_joycaption_data(keywords, custom_instructions, avoid, question, cap1,
     except Exception as e:
         return f"❌ Export failed: {str(e)}", None
 
-# ===== GRADIO UI =====
+# ===== UI =====
 with gr.Blocks(title="JoyCaption Advanced Prompting System", theme=gr.themes.Soft()) as demo:
     gr.HTML(TITLE)
     insert_key, insert_que, insert_use, insert_not = create_template_functions()
@@ -278,7 +278,8 @@ with gr.Blocks(title="JoyCaption Advanced Prompting System", theme=gr.themes.Sof
             ask_btn = gr.Button("Ask", variant="secondary")
             qa_output = gr.Textbox(label="Answer", lines=3, show_copy_button=True)
 
-            gr.Markdown("---")
+        with gr.Column(scale=1):
+            # Template buttons moved above tabs
             gr.Markdown("**Insert Template**")
             with gr.Row():
                 key_btn = gr.Button("key", size="sm")
@@ -286,20 +287,22 @@ with gr.Blocks(title="JoyCaption Advanced Prompting System", theme=gr.themes.Sof
                 use_btn = gr.Button("use", size="sm")
                 not_btn = gr.Button("not", size="sm")
 
-        with gr.Column(scale=1):
+            # Tabs with flexible Textboxes
             with gr.Tab("📝 Casual"):
-                system1 = gr.Textbox(label="System", lines=2, value=DEFAULT_PROMPTS["casual"]["system"])
-                user1 = gr.Textbox(label="User", lines=2, value=DEFAULT_PROMPTS["casual"]["user"])
+                system1 = gr.Textbox(label="System", show_label=True, placeholder="", value=DEFAULT_PROMPTS["casual"]["system"], lines=1, max_lines=5)
+                user1 = gr.Textbox(label="User", show_label=True, placeholder="", value=DEFAULT_PROMPTS["casual"]["user"], lines=1, max_lines=8)
                 gen1_btn = gr.Button("Generate Casual", variant="primary")
                 out1 = gr.Textbox(lines=5, show_copy_button=True)
+
             with gr.Tab("🤝 Friendly"):
-                system2 = gr.Textbox(label="System", lines=2, value=DEFAULT_PROMPTS["friendly"]["system"])
-                user2 = gr.Textbox(label="User", lines=2, value=DEFAULT_PROMPTS["friendly"]["user"])
+                system2 = gr.Textbox(label="System", show_label=True, placeholder="", value=DEFAULT_PROMPTS["friendly"]["system"], lines=1, max_lines=5)
+                user2 = gr.Textbox(label="User", show_label=True, placeholder="", value=DEFAULT_PROMPTS["friendly"]["user"], lines=1, max_lines=8)
                 gen2_btn = gr.Button("Generate Friendly", variant="primary")
                 out2 = gr.Textbox(lines=5, show_copy_button=True)
+
             with gr.Tab("🔥 Erotic"):
-                system3 = gr.Textbox(label="System", lines=2, value=DEFAULT_PROMPTS["erotic"]["system"])
-                user3 = gr.Textbox(label="User", lines=2, value=DEFAULT_PROMPTS["erotic"]["user"])
+                system3 = gr.Textbox(label="System", show_label=True, placeholder="", value=DEFAULT_PROMPTS["erotic"]["system"], lines=1, max_lines=5)
+                user3 = gr.Textbox(label="User", show_label=True, placeholder="", value=DEFAULT_PROMPTS["erotic"]["user"], lines=1, max_lines=8)
                 gen3_btn = gr.Button("Generate Erotic", variant="primary")
                 out3 = gr.Textbox(lines=5, show_copy_button=True)
 
@@ -308,13 +311,12 @@ with gr.Blocks(title="JoyCaption Advanced Prompting System", theme=gr.themes.Sof
             export_out = gr.Textbox(visible=False)
             export_file = gr.File(visible=False)
 
-    # === Event Bindings ===
+    # === EVENTS ===
     gen1_btn.click(generate_caption_1, [image_input, system1, user1], out1)
     gen2_btn.click(generate_caption_2, [image_input, system2, user2], out2)
     gen3_btn.click(generate_caption_3, [image_input, system3, user3], out3)
     ask_btn.click(answer_question, [image_input, question_input], qa_output)
 
-    # Shared template bar
     key_btn.click(lambda s1,u1,k: insert_key(s1,u1,k), [system1,user1,keywords_input], [system1,user1])
     que_btn.click(lambda s1,u1,q: insert_que(s1,u1,q), [system1,user1,question_input], [system1,user1])
     use_btn.click(lambda s1,u1,c: insert_use(s1,u1,c), [system1,user1,custom_instruction_input], [system1,user1])