Create main.py

main.py

@@ -0,0 +1,33 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from gradio_client import Client
+
+# Configure your Gradio Space ID and default endpoint
+SPACE_ID = "prithivMLmods/SAMBANOVA"
+DEFAULT_API = "/chat"
+
+client = Client(SPACE_ID)
+
+def chat_with_gradio(message: str, api_name: str = DEFAULT_API):
+    """
+    Send a chat message to the Gradio API and return the response.
+    """
+    try:
+        return client.predict(message=message, api_name=api_name)
+    except Exception as e:
+        return {"error": str(e)}
+
+app = FastAPI()
+
+class ChatRequest(BaseModel):
+    message: str
+    api_name: str = DEFAULT_API
+
+@app.post("/chat")
+async def chat_endpoint(req: ChatRequest):
+    return {"reply": chat_with_gradio(req.message, req.api_name)}
+
+if __name__ == "__main__":
+    import uvicorn
+    print(f"Starting server on http://0.0.0.0:80 using {SPACE_ID}{DEFAULT_API}")
+    uvicorn.run(app, host="0.0.0.0", port=80)