Update app.py

app.py

@@ -105,7 +105,6 @@ def process_image(image, task_name):
         return "❌ โมเดลยังไม่ได้โหลดหรือเกิดข้อผิดพลาดในการเริ่มต้น"
     if image is None:
         return "กรุณาเลือกรูปภาพ"
-
     try:
         # แปลง Gradio Image เป็น PIL Image
         if not isinstance(image, Image.Image):
@@ -115,7 +114,6 @@ def process_image(image, task_name):
 
         # ดึง Prompt จาก Task ที่เลือก
         prompt = TASK_PROMPTS.get(task_name, "<CAPTION>")
-
         inputs = florence_processor(text=prompt, images=image, return_tensors="pt").to(device)
         
         generated_ids = florence_model.generate(
@@ -126,26 +124,32 @@ def process_image(image, task_name):
             do_sample=False,
             num_beams=3,
         )
-        
         generated_text = florence_processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
         parsed_answer = florence_processor.post_process_generation(
             generated_text, task=prompt, image_size=(image.width, image.height)
         )
+
+        # --- ส่วนแก้ไข: แยกค่าออกจาก Dictionary ---
+        # parsed_answer จะเป็น dict เช่น {'<DETAILED_CAPTION>': 'ข้อความ...'}
+        # เราต้องการแค่ 'ข้อความ...'
+        if isinstance(parsed_answer, dict):
+            # ดึงค่าแรก (Value) จาก Dictionary
+            result_text = next(iter(parsed_answer.values()))
+        else:
+            # ถ้าไม่ใช่ dict (กรณีบาง task อาจ return string โดยตรง)
+            result_text = str(parsed_answer)
         
-        # คืนค่าผลลัพธ์ (อาจเป็น dict หรือ str แล้วแต่ task)
-        result = str(parsed_answer)
-        print(f"\n✅ Task: {task_name} | Prompt: {prompt}")
-        return result
+        print(f"\n✅ Task: {task_name} | Result: {result_text}")
+        return result_text
 
     except Exception as e:
         return f"❌ เกิดข้อผิดพลาดขณะประมวลผล: {str(e)}"
 
 # --- สร้าง UI ---
-with gr.Blocks(title="Image-to-Prompt (Florence-2 NSFW)", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Image-to-Prompt (Florence-2)", theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
-    # 🖼️ Image-to-Prompt (Florence-2 NSFW)
+    # 🖼️ Image-to-Prompt (Florence-2)
     อัปโหลดรูปภาพและเลือกประเภทการวิเคราะห์เพื่อสร้าง Prompt
-    *ใช้โมเดล NSFW: `ljnlonoljpiljm/florence-2-base-nsfw-v2`*
     """)
 
     with gr.Row():