Update backend/main.py

backend/main.py

@@ -0,0 +1,52 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import StreamingResponse
+from fastapi.middleware.cors import CORSMiddleware
+import gradio_client as grc
+from PIL import Image
+import io
+
+app = FastAPI(title="Face Aging + White Hair & Beard API")
+
+# CORS allow karo Android app ke liye
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Production mein specific origin daalna (jaise http://10.0.2.2 etc.)
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Apna Hugging Face Space ka URL yahan daal do
+HF_SPACE_URL = "https://huggingface.co/spaces/Seniordev22/OldServer"   # ← apna actual space URL daal do
+
+client = grc.Client(HF_SPACE_URL)
+
+@app.post("/age-face")
+async def age_face(file: UploadFile = File(...)):
+    if not file.content_type.startswith("image/"):
+        raise HTTPException(400, detail="Only image files allowed")
+
+    contents = await file.read()
+
+    try:
+        # Gradio Space ko call kar rahe hain (image input)
+        result = client.predict(
+            image=grc.handle_file(io.BytesIO(contents)),   # ya direct bytes bhi try kar sakte ho
+            api_name="/predict"      # ← yeh important hai! Apne Space mein "Use via API" check karo exact api_name ke liye
+        )
+
+        # Agar result mein image hai (usually tuple hota hai)
+        if isinstance(result, tuple):
+            output_image = result[0]   # pehla output image assume kar rahe hain
+        else:
+            output_image = result
+
+        # Image ko bytes mein convert karke return karo
+        img_byte_arr = io.BytesIO()
+        output_image.save(img_byte_arr, format='PNG')
+        img_byte_arr.seek(0)
+
+        return StreamingResponse(img_byte_arr, media_type="image/png")
+
+    except Exception as e:
+        raise HTTPException(500, detail=f"Error processing image: {str(e)}")