Add application file

app.py

@@ -2,10 +2,25 @@ import gradio as gr
 import base64
 import requests
 import json
+import re
 
 API_KEY = "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"
 IMAGE_MODEL = "opengvlab/internvl3-14b:free"
 
+def extract_json_from_code_block(text):
+    try:
+        # Match content between triple backticks with 'json'
+        match = re.search(r"```json\s*(\{.*?\})\s*```", text, re.DOTALL)
+        if not match:
+            return {"error": "No JSON block found."}
+        
+        json_str = match.group(1)
+
+        # Convert to Python dictionary
+        return json.loads(json_str)
+    except json.JSONDecodeError as e:
+        return {"error": f"Invalid JSON: {str(e)}"}
+
 def process_passport(image):
     try:
         with open(image, "rb") as f:
@@ -63,7 +78,7 @@ def process_passport(image):
         response = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload)
         result = response.json()
 
-        return json.dumps(result["choices"][0]["message"]["content"], indent=2)
+        return json.dumps(extract_json_from_code_block(result["choices"][0]["message"]["content"]), indent=2)
 
     except Exception as e:
         return f"⚠️ Error: {str(e)}"