Update app.py

app.py

@@ -2,25 +2,30 @@ from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 import numpy as np
-import cv2
 from PIL import Image
 from io import BytesIO
 import easyocr
 from ultralytics import YOLO
 import os
 
-# ✅ Force EasyOCR & YOLO to use writable cache folders
+# Gradio imports
+import gradio as gr
+import requests
+import io
+
+# Set writable cache folders
 os.environ["EASYOCR_MODULE_PATH"] = "/app/.EasyOCR"
 os.environ["EASYOCR_CACHE_DIR"] = "/app/.EasyOCR"
 os.environ["YOLO_CONFIG_DIR"] = "/app/.yolo"
 
 app = FastAPI()
 
-# ✅ Load EasyOCR and YOLO from local files
+# Load models
 reader = easyocr.Reader(['en'], model_storage_directory="/app/.EasyOCR")
 model = YOLO("/app/yolov8n.pt")
 print("✅ YOLO model loaded from /app/yolov8n.pt")
 
+# CORS middleware
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"], allow_credentials=True,
@@ -52,3 +57,29 @@ async def detect_plate(file: UploadFile = File(...)):
 
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)
+
+# --- Gradio frontend ---
+
+def detect_plate_gradio(image):
+    buf = io.BytesIO()
+    image.save(buf, format="JPEG")
+    buf.seek(0)
+    files = {"file": ("image.jpg", buf, "image/jpeg")}
+    # Call your own deployed API endpoint
+    api_url = "https://huggingface.co/spaces/muddasser/Plate_Reader/api/detect"
+    response = requests.post(api_url, files=files)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return f"Error: {response.status_code}"
+
+iface = gr.Interface(
+    fn=detect_plate_gradio,
+    inputs=gr.Image(type="pil"),
+    outputs="json",
+    title="License Plate Reader",
+    description="Upload an image to detect license plates"
+)
+
+if __name__ == "__main__":
+    iface.launch()