Update app.py

app.py

@@ -3,33 +3,41 @@ from fastapi.responses import StreamingResponse
 import io
 from bald_processor import make_realistic_bald
 
-app = FastAPI(title="Make Me Bald API 😎")
+app = FastAPI(title="Make Me Bald API 😎", 
+              description="Upload photo → Get realistic bald version! 🧑‍🦲",
+              version="1.0")
 
 @app.post("/make-bald/")
 async def bald_endpoint(file: UploadFile = File(...)):
     if not file.content_type.startswith("image/"):
-        raise HTTPException(400, detail="Sirf image file upload kar bhai!")
+        raise HTTPException(status_code=400, detail="Sirf image file upload kar bhai! (jpeg/png etc.)")
     
     try:
         contents = await file.read()
         bald_bytes = make_realistic_bald(contents)
         
+        return StreamingResponse(
+            io.BytesIO(bald_bytes),
+            media_type="image/jpeg",
+            headers={"Content-Disposition": "attachment; filename=bald_version.jpg"}
+        )
+    
     except ValueError as ve:
         error_detail = str(ve)
         if error_detail == "NO_HAIR_DETECTED":
             raise HTTPException(status_code=400, detail="NO_HAIR")
         else:
             raise HTTPException(status_code=400, detail=error_detail)
-        
-    except Exception as e:
-        raise HTTPException(500, detail=f"Processing mein error: {str(e)}")
     
-    return StreamingResponse(
-        io.BytesIO(bald_bytes),
-        media_type="image/jpeg",
-        headers={"Content-Disposition": "attachment; filename=bald_version.jpg"}
-    )
+    except Exception as e:
+        # Optional: print for logs
+        print(f"Processing error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Processing mein error: {str(e)}")
 
 @app.get("/")
 def home():
-    return {"message": "Bald banne aaya? POST /make-bald/ pe image daal!"}
+    return {
+        "message": "Bald banne aaya? 😏",
+        "how_to_use": "POST request bhejo /make-bald/ pe with form-data key 'file' aur image attach karo.",
+        "example": "cURL: curl -X POST -F 'file=@your_photo.jpg' https://your-space-url.hf.space/make-bald/"
+    }