Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import json
 from datetime import datetime
 from pathlib import Path
+from PIL import Image
 
 # ============================
 #  USER DATABASE
@@ -33,6 +34,23 @@ else:
         ]
     }
 
+# ============================
+#  IMAGE CAPTIONING MODEL
+# ============================
+from transformers import BlipProcessor, BlipForConditionalGeneration
+
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
+model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
+
+def describe_image(img):
+    if img is None:
+        return ""
+    image = Image.open(img)
+    inputs = processor(image, return_tensors="pt")
+    out = model.generate(**inputs)
+    caption = processor.decode(out[0], skip_special_tokens=True)
+    return caption
+
 # ============================
 #  HELPERS
 # ============================
@@ -112,13 +130,6 @@ def generate_variants(kind, goal, tone, audience, length, constraints, language,
     return variants
 
 def generate_json(username, tier, goal, kind, tone, audience, length, constraints, language, n):
-    goal = normalize(goal, "Describe the task clearly.")
-    tone = normalize(tone, "neutral")
-    audience = normalize(audience, "general")
-    length = normalize(length, "medium")
-    constraints = normalize(constraints, "none")
-    language = normalize(language, "Indonesian")
-
     limits = {"free": 5, "premium": 10, "super": 20}
     n = max(1, min(limits[tier], int(n)))
 
@@ -154,9 +165,7 @@ with gr.Blocks() as demo:
     login_user = gr.State("")
     login_tier = gr.State("free")
 
-    # ============================
-    #  LOGIN UI
-    # ============================
+    # LOGIN UI
     with gr.Group(visible=True) as login_ui:
         gr.Markdown("# 🔐 Login untuk Menggunakan AIPromptLab")
 
@@ -165,14 +174,28 @@ with gr.Blocks() as demo:
         login_btn = gr.Button("Login")
         login_msg = gr.Markdown("")
 
-    # ============================
-    #  MAIN FEATURE UI (HIDDEN FIRST)
-    # ============================
+    # MAIN UI (hidden first)
     with gr.Group(visible=False) as main_ui:
 
         gr.Markdown("# 🌟 AIPromptLab — JSON Output Edition")
 
-        goal = gr.Textbox(label="Goal")
+        input_mode = gr.Radio(["text", "image"], value="text", label="Input Mode")
+
+        text_input = gr.Textbox(label="Text Input", visible=True)
+        image_input = gr.Image(label="Upload Image", type="filepath", visible=False)
+
+        def switch_input(mode):
+            return (
+                gr.update(visible=(mode=="text")),
+                gr.update(visible=(mode=="image"))
+            )
+
+        input_mode.change(
+            switch_input,
+            [input_mode],
+            [text_input, image_input]
+        )
+
         kind = gr.Radio(["text", "image", "code"], value="text")
         tone = gr.Textbox(label="Tone")
         audience = gr.Textbox(label="Audience")
@@ -185,15 +208,13 @@ with gr.Blocks() as demo:
         output_json = gr.Textbox(label="Output JSON", lines=25, elem_id="json-output")
         copy_btn = gr.Button("Copy JSON")
 
-    # ============================
-    #  LOGIN LOGIC
-    # ============================
+    # LOGIN LOGIC
     def handle_login(username, password):
         ok, tier, msg = login(username, password)
         return (
             msg,
-            gr.update(visible=not ok),   # login UI hilang
-            gr.update(visible=ok),       # main UI muncul
+            gr.update(visible=not ok),
+            gr.update(visible=ok),
             ok,
             username if ok else "",
             tier if ok else "free"
@@ -205,27 +226,30 @@ with gr.Blocks() as demo:
         [login_msg, login_ui, main_ui, login_status, login_user, login_tier]
     )
 
-    # ============================
-    #  GENERATE JSON
-    # ============================
+    # GENERATE JSON
     def handle_generate(login_status, login_user, login_tier,
-                        goal, kind, tone, audience, length, constraints, language, n):
+                        input_mode, text_input, image_input,
+                        kind, tone, audience, length, constraints, language, n):
 
         if not login_status:
             return "❌ Anda belum login."
 
+        if input_mode == "text":
+            goal = text_input
+        else:
+            goal = describe_image(image_input)
+
         return generate_json(login_user, login_tier, goal, kind, tone, audience, length, constraints, language, n)
 
     generate_btn.click(
         handle_generate,
         [login_status, login_user, login_tier,
-         goal, kind, tone, audience, length, constraints, language, n],
+         input_mode, text_input, image_input,
+         kind, tone, audience, length, constraints, language, n],
         output_json
     )
 
-    # ============================
-    #  COPY JSON (SAFE FOR GRADIO 6.x)
-    # ============================
+    # COPY JSON
     copy_btn.click(
         None,
         None,
@@ -233,9 +257,7 @@ with gr.Blocks() as demo:
         js="navigator.clipboard.writeText(document.getElementById('json-output').value)"
     )
 
-# ============================
-#  LAUNCH (share=True FIX)
-# ============================
+# LAUNCH
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,