Update app.py

app.py

@@ -58,35 +58,55 @@ def extract_metadata(image):
 
     try:
         # Ensure image is PIL Image
+        print(f"DEBUG: Received image type: {type(image)}")
         if not isinstance(image, PILImage.Image):
             image = PILImage.open(image).convert("RGB")
+        print(f"DEBUG: After conversion, image type: {type(image)}")
+        print(f"DEBUG: Image format before setting: {image.format}")
 
         # Set format (required by Outlines Image class)
         if not image.format:
             image.format = "PNG"
+        print(f"DEBUG: Image format after setting: {image.format}")
+
+        # Wrap in Outlines Image
+        outlines_image = Image(image)
+        print(f"DEBUG: Outlines Image created: {type(outlines_image)}")
+        print(f"DEBUG: Outlines Image.image type: {type(outlines_image.image)}")
 
         # Create Chat prompt with Image
+        content_list = [
+            {"type": "text", "text": EXTRACTION_PROMPT},
+            {"type": "image", "image": outlines_image},
+        ]
+        print(f"DEBUG: Content list created: {len(content_list)} items")
+        print(f"DEBUG: Content[0] type: {content_list[0]['type']}")
+        print(f"DEBUG: Content[1] type: {content_list[1]['type']}, image type: {type(content_list[1]['image'])}")
+
         prompt = Chat(
             messages=[
                 {
                     "role": "user",
-                    "content": [
-                        {"type": "text", "text": EXTRACTION_PROMPT},
-                        {"type": "image", "image": Image(image)},
-                    ],
+                    "content": content_list,
                 }
             ]
         )
+        print(f"DEBUG: Chat prompt created successfully")
 
         # Generate with structured output - guaranteed valid JSON
+        print(f"DEBUG: Starting generation...")
         result = model(prompt, CatalogCardMetadata, max_new_tokens=512)
+        print(f"DEBUG: Generation complete, result type: {type(result)}")
 
         # Parse and format (always valid JSON with Outlines)
         metadata = CatalogCardMetadata.model_validate_json(result)
-        return str(json.dumps(metadata.model_dump(exclude_none=True), indent=2))
+        return json.dumps(metadata.model_dump(exclude_none=True), indent=2)
 
     except Exception as e:
-        return f"Error during extraction: {str(e)}"
+        import traceback
+        error_msg = f"Error during extraction: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
+        print(error_msg)
+        return error_msg
 
 
 # Create Gradio interface
@@ -111,7 +131,7 @@ with gr.Blocks(title="Library Card Metadata Extractor") as demo:
 
         with gr.Column(scale=1):
             gr.Markdown("### 📋 Extracted Metadata (JSON)")
-            output = gr.Text()
+            output = gr.Code(label="Metadata", language="json", lines=15)
 
     submit_btn.click(fn=extract_metadata, inputs=image_input, outputs=output)