Update app.py

app.py

@@ -3,26 +3,25 @@ import base64
 import mimetypes
 import json
 from mistralai import Mistral, ImageURLChunk
+from PIL import Image
+import io
 
 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:{mime_type};base64,{encoded}"
-
+def process_image(image):
     try:
+        # نحول الصورة من PIL إلى base64
+        buffered = io.BytesIO()
+        image.save(buffered, format="JPEG")
+        encoded = base64.b64encode(buffered.getvalue()).decode()
+        mime_type = "image/jpeg"
+        data_url = f"data:{mime_type};base64,{encoded}"
+
         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
@@ -32,6 +31,6 @@ def process_image(image_path):
 
 gr.Interface(
     fn=process_image,
-    inputs=gr.Image(type="filepath"),
-    outputs="text"  # لأننا هلأ عم نرجع json_string كسلسلة نصية
+    inputs=gr.Image(type="pil"),  # بدل filepath بـ pil
+    outputs="text"
 ).launch(share=True, show_error=True)