Update app.py

app.py

@@ -1,25 +1,26 @@
-
 import gradio as gr
 import base64
-import mimetypes
 import json
-import mistralai
+import io
 
-print(mistralai)
-print(dir(mistralai))from PIL import Image
+from PIL import Image
 
-import io
+from mistralai import Mistral
+from mistralai.models import ImageURLChunk
 
-client = Mistral(api_key="RJIqm5OvwoMvLeWrFdv5JBx26tLsSSK")
+client = Mistral(api_key="YOUR_API_KEY")
 
 def process_image(image):
     try:
-        # نحول الصورة من PIL إلى base64
+        if image.mode == "RGBA":
+            image = image.convert("RGB")
+
         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}"
+
+        data_url = f"data:image/jpeg;base64,{encoded}"
 
         response = client.ocr.process(
             document=ImageURLChunk(image_url=data_url),
@@ -27,14 +28,14 @@ def process_image(image):
         )
 
         response_dict = json.loads(response.model_dump_json())
-        json_string = json.dumps(response_dict, indent=4, ensure_ascii=False)
-        return json_string
+
+        return json.dumps(response_dict, indent=4, ensure_ascii=False)
 
     except Exception as e:
-        return {"error": str(e)}
+        return str(e)
 
 gr.Interface(
     fn=process_image,
-    inputs=gr.Image(type="pil"), 
-    outputs="text"
-).launch(share=True, show_error=True)
+    inputs=gr.Image(type="pil"),
+    outputs="text",
+).launch()