Update components/llm_ocr_api.py

components/llm_ocr_api.py

@@ -1,21 +1,15 @@
-import requests
-import io
-
-def extract_text_from_space(image):
-    api_url = "https://huggingface.co/spaces/Hammedalmodel/handwritten_to_text/api/predict"
-    
-    image_bytes = io.BytesIO()
-    image.save(image_bytes, format="PNG")
-    image_bytes.seek(0)
-
-    files = {
-        "data": ("image.png", image_bytes, "image/png")
-    }
+from gradio_client import Client, handle_file
 
+def extract_text_from_space(image_path):
     try:
-        response = requests.post(api_url, files=files)
-        response.raise_for_status()
-        result = response.json()
-        return result["data"][0] if "data" in result else "❌ Unexpected response format."
+        # Create client for Hugging Face Space
+        client = Client("Hammedalmodel/handwritten_to_text")
+        
+        # Send image file path to the Space
+        result = client.predict(
+            image=handle_file(image_path),  # handle_file needs a path
+            api_name="/predict"
+        )
+        return result  # Expected to be a single string output
     except Exception as e:
         return f"❌ Error: {str(e)}"