Spaces:
Sleeping
Sleeping
File size: 1,053 Bytes
9d8aa3f 8bbb2b3 9d8aa3f ec96aa7 9d8aa3f 6f912b7 9d8aa3f ec96aa7 8bbb2b3 ec96aa7 8bbb2b3 9d8aa3f bf3a2bf ec96aa7 fae1e29 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import gradio as gr
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):
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"
)
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="pil"), # بدل filepath بـ pil
outputs="text"
).launch(share=True, show_error=True)
|