Update app.py

app.py

@@ -69,14 +69,37 @@ def smoldocling_readimage(image: Image.Image, prompt_text: str) -> str:
 # -----------------------------
 #  Wrapper for MCP schema compatibility
 # -----------------------------
-def smoldocling_entry(image: object, prompt_text: str) -> str:
+def smoldocling_entry(image: str, prompt_text: str) -> str:
     """
-    Entry point for MCP tool.
-    Accepts any of:
-      - base64 string
-      - dict {"type": "image", "data": "data:image/png;base64,..."}
-      - file path
+    Entry point for the SmolDocling MCP tool.
+
+    Expected input formats:
+    - **Base64 string**: "data:image/png;base64,...."
+    - **Object** (Perplexity/Claude style): {"type": "image", "data": "data:image/png;base64,..."}
+    - **Local file path** (for internal testing)
+
+    Parameters
+    ----------
+    image : str
+        A base64-encoded image string (with or without data: prefix) OR
+        a JSON-encoded object containing image data.
+    prompt_text : str
+        Instruction text for how to process the document (e.g., "Convert this page to docling.")
+
+    Returns
+    -------
+    str
+        Structured or textual content extracted from the image.
     """
+    # Handle Perplexity-style dicts encoded as JSON strings
+    try:
+        import json
+        maybe_json = json.loads(image)
+        if isinstance(maybe_json, dict) and "data" in maybe_json:
+            image = maybe_json
+    except Exception:
+        pass
+
     pil_image = convert_to_pil(image)
     return smoldocling_readimage(pil_image, prompt_text)