Update app.py

app.py

@@ -1,21 +1,37 @@
 import gradio as gr
 import base64
+import mimetypes
+import json
 from mistralai import Mistral, ImageURLChunk
 
 client = Mistral(api_key="RJIqm5OvwoMvLeWrFdv5JBx26tLsSSK7")
 
 def process_image(image_path):
+    mime_type, _ = mimetypes.guess_type(image_path)
+    if mime_type is None:
+        return {"error": "Unsupported image type or cannot detect file type."}
+    
     with open(image_path, "rb") as image:
         encoded = base64.b64encode(image.read()).decode()
-    data_url = f"data:image/jpeg;base64,{encoded}"
-    response = client.ocr.process(
-        document=ImageURLChunk(image_url=data_url),
-        model="mistral-ocr-latest"
-    )
-    return response.to_dict()
+    
+    data_url = f"data:{mime_type};base64,{encoded}"
+
+    try:
+        response = client.ocr.process(
+            document=ImageURLChunk(image_url=data_url),
+            model="mistral-ocr-latest"
+        )
+
+        # تحويل النتيجة إلى JSON منسق
+        response_dict = json.loads(response.model_dump_json())
+        json_string = json.dumps(response_dict, indent=4, ensure_ascii=False)
+        return json_string
+
+    except Exception as e:
+        return {"error": str(e)}
 
 gr.Interface(
     fn=process_image,
     inputs=gr.Image(type="filepath"),
-    outputs="json"
+    outputs="text"  # لأننا هلأ عم نرجع json_string كسلسلة نصية
 ).launch(share=True)