Spaces:

maxjski
/

3dai-backend

Paused

App Files Files Community

Vnmrsharma commited on May 29, 2025

Commit

b834f10

1 Parent(s): f3fa14d

Updated Simplified API Calls

Browse files

Files changed (4) hide show

__pycache__/main.cpython-313.pyc +0 -0
frontend.html +130 -0
main.py +178 -0
requirements.txt +7 -0

__pycache__/main.cpython-313.pyc ADDED Viewed

Binary file (9.35 kB). View file

frontend.html ADDED Viewed

	@@ -0,0 +1,130 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8" />
+  <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+  <title>Meshy Image-to-3D Proxy Test</title>
+  <style>
+    body { font-family: Arial, sans-serif; margin: 2rem; }
+    section { margin-bottom: 2rem; }
+    pre { background: #f4f4f4; padding: 1rem; overflow: auto; }
+    input, button { margin-top: 0.5rem; }
+  </style>
+</head>
+<body>
+  <h1>Meshy Image-to-3D Proxy Test UI</h1>
+  <section>
+    <h2>1) Create Image-to-3D Task</h2>
+    <input type="file" id="imageInput" accept="image/png, image/jpeg" /><br />
+    <button id="submitImage">Submit Image</button>
+    <pre id="imageResult">✔ No result yet</pre>
+  </section>
+  <section>
+    <h2>2) Retrieve Task Result</h2>
+    <input type="text" id="taskIdInput" placeholder="Enter task ID" size="50" /><br />
+    <button id="getResult">Fetch Result</button>
+    <pre id="resultOutput">✔ No result yet</pre>
+  </section>
+  <section>
+    <h2>3) List All Tasks</h2>
+    <button id="listTasks">List Tasks</button>
+    <pre id="listOutput">✔ No result yet</pre>
+  </section>
+  <section>
+    <h2>4) Create Text-to-3D Task</h2>
+    <textarea id="textInput" rows="4" cols="50" placeholder="Enter description..."></textarea><br />
+    <button id="submitText">Submit Text</button>
+    <pre id="textResult">✔ No result yet</pre>
+  </section>
+  <section>
+    <h2>5) Retrieve Text-to-3D Result</h2>
+    <input type="text" id="textTaskIdInput" placeholder="Enter text task ID" size="50" /><br />
+    <button id="getTextResult">Fetch Text Result</button>
+    <pre id="textResultOutput">✔ No result yet</pre>
+  </section>
+  <section>
+    <h2>6) List Text-to-3D Tasks</h2>
+    <button id="listTextTasks">List Text Tasks</button>
+    <pre id="listTextOutput">✔ No result yet</pre>
+  </section>
+  <script>
+    // Base URL of your FastAPI service
+    const API_BASE = 'http://localhost:8000';
+    // 1) Create Image-to-3D Task
+    document.getElementById('submitImage').addEventListener('click', async () => {
+      const fileInput = document.getElementById('imageInput');
+      if (!fileInput.files.length) return alert('Please select an image first.');
+      const form = new FormData();
+      form.append('image', fileInput.files[0]);
+      const res = await fetch(`${API_BASE}/req-img-to-3d`, {
+        method: 'POST',
+        body: form,
+      });
+      const data = await res.json();
+      document.getElementById('imageResult').textContent = JSON.stringify(data, null, 2);
+    });
+    // 2) Retrieve Task Result
+    document.getElementById('getResult').addEventListener('click', async () => {
+      const taskId = document.getElementById('taskIdInput').value.trim();
+      if (!taskId) return alert('Enter a task ID');
+      const res = await fetch(`${API_BASE}/req-result-img-to-3d/${taskId}`);
+      const data = await res.json();
+      document.getElementById('resultOutput').textContent = JSON.stringify(data, null, 2);
+    });
+    // 3) List All Tasks
+    document.getElementById('listTasks').addEventListener('click', async () => {
+      const res = await fetch(`${API_BASE}/list-img-to-3d-req`);
+      const data = await res.json();
+      document.getElementById('listOutput').textContent = JSON.stringify(data, null, 2);
+    });
+    // 4) Create Text-to-3D Task
+    document.getElementById('submitText').addEventListener('click', async () => {
+      const textInput = document.getElementById('textInput');
+      if (!textInput.value.trim()) return alert('Please enter text');
+      const res = await fetch(`${API_BASE}/req-text-to-3d`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({
+          text: textInput.value.trim(),
+        }),
+      });
+      const data = await res.json();
+      document.getElementById('textResult').textContent = JSON.stringify(data, null, 2);
+    });
+    // 5) Retrieve Text-to-3D Result
+    document.getElementById('getTextResult').addEventListener('click', async () => {
+      const textTaskId = document.getElementById('textTaskIdInput').value.trim();
+      if (!textTaskId) return alert('Enter a text task ID');
+      const res = await fetch(`${API_BASE}/req-result-text-to-3d/${textTaskId}`);
+      const data = await res.json();
+      document.getElementById('textResultOutput').textContent = JSON.stringify(data, null, 2);
+    });
+    // 6) List Text-to-3D Tasks
+    document.getElementById('listTextTasks').addEventListener('click', async () => {
+      const res = await fetch(`${API_BASE}/list-text-to-3d-req`);
+      const data = await res.json();
+      document.getElementById('listTextOutput').textContent = JSON.stringify(data, null, 2);
+    });
+  </script>
+</body>
+</html>

main.py ADDED Viewed

	@@ -0,0 +1,178 @@

+import os
+import base64
+from fastapi import FastAPI, UploadFile, File, HTTPException, Depends
+import httpx
+from contextlib import asynccontextmanager
+from dotenv import load_dotenv
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+load_dotenv()
+class Settings:
+    def __init__(self):
+        self.mesh_api_key = os.getenv("MESHY_API_KEY")
+        if not self.mesh_api_key:
+            raise RuntimeError("MESHY_API_KEY environment variable not set")
+        self.openai_api_key = os.getenv("OPENAI_API_KEY")
+        if not self.openai_api_key:
+            raise RuntimeError("OPENAI_API_KEY environment variable not set")
+settings = Settings()
+def get_settings():
+    return settings
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    app.state.client = httpx.AsyncClient(timeout=10.0)
+    yield
+    await app.state.client.aclose()
+app = FastAPI(
+    title="3DAI API Module",
+    version="1.0.0",
+    description="Proxy endpoints for the APIs",
+    lifespan=lifespan
+)
+# Add CORS for all origins (for testing)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+# Pydantic model for incoming text prompt
+class TextPrompt(BaseModel):
+    text: str
+@app.post("/req-img-to-3d", tags=["Image-to-3D"])
+async def req_img_to_3d(
+    image: UploadFile = File(...),
+    settings: Settings = Depends(get_settings)
+):
+    """
+    Create a new Image-to-3D task by uploading an image.
+    """
+    content = await image.read()
+    if not content:
+        raise HTTPException(status_code=400, detail="Empty file upload")
+    data_uri = (f"data:{image.content_type};base64," +
+                base64.b64encode(content).decode())
+    payload = {"image_url": data_uri}
+    headers = {"Authorization": f"Bearer {settings.mesh_api_key}"}
+    resp = await app.state.client.post(
+        "https://api.meshy.ai/openapi/v1/image-to-3d",
+        json=payload,
+        headers=headers,
+    )
+    if resp.status_code not in (200, 201):
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+@app.get("/req-result-img-to-3d/{task_id}", tags=["Image-to-3D"])
+async def req_result_img_to_3d(
+    task_id: str,
+    settings: Settings = Depends(get_settings)
+):
+    """
+    Retrieve the status and result of a Meshy Image-to-3D task by ID.
+    """
+    headers = {"Authorization": f"Bearer {settings.mesh_api_key}"}
+    url = f"https://api.meshy.ai/openapi/v1/image-to-3d/{task_id}"
+    resp = await app.state.client.get(url, headers=headers)
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+@app.get("/list-img-to-3d-req", tags=["Image-to-3D"])
+async def list_img_to_3d_req(
+    settings: Settings = Depends(get_settings)
+):
+    """
+    List all Image-to-3D tasks.
+    """
+    headers = {"Authorization": f"Bearer {settings.mesh_api_key}"}
+    resp = await app.state.client.get(
+        "https://api.meshy.ai/openapi/v1/image-to-3d",
+        headers=headers
+    )
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+@app.post("/req-text-to-3d", tags=["Text-to-3D"] )
+async def req_text_to_3d(
+    prompt: TextPrompt,
+    settings: Settings = Depends(get_settings)
+):
+    """
+    Reframe user text via OpenAI, then create a Text-to-3D preview task.
+    """
+    # Reframe prompt
+    openai_resp = await app.state.client.post(
+        "https://api.openai.com/v1/chat/completions",
+        headers={"Authorization": f"Bearer {settings.openai_api_key}"},
+        json={
+            "model": "gpt-3.5-turbo",
+            "messages": [
+                {"role": "system", "content": "Rephrase the user description to be an ideal prompt for 3D model generation."},
+                {"role": "user", "content": prompt.text}
+            ]
+        }
+    )
+    if openai_resp.status_code != 200:
+        raise HTTPException(status_code=openai_resp.status_code, detail=openai_resp.text)
+    reframed = openai_resp.json()["choices"][0]["message"]["content"]
+    # Create Meshy Text-to-3D task
+    meshy_resp = await app.state.client.post(
+        "https://api.meshy.ai/openapi/v2/text-to-3d",
+        headers={"Authorization": f"Bearer {settings.mesh_api_key}"},
+        json={"mode": "preview", "prompt": reframed}
+    )
+    if meshy_resp.status_code not in (200, 201):
+        raise HTTPException(status_code=meshy_resp.status_code, detail=meshy_resp.text)
+    return meshy_resp.json()
+@app.get("/req-result-text-to-3d/{task_id}", tags=["Text-to-3D"])
+async def req_result_text_to_3d(
+    task_id: str,
+    settings: Settings = Depends(get_settings)
+):
+    """
+    Retrieve the status/result of a Text-to-3D task by ID.
+    """
+    resp = await app.state.client.get(
+        f"https://api.meshy.ai/openapi/v2/text-to-3d/{task_id}",
+        headers={"Authorization": f"Bearer {settings.mesh_api_key}"}
+    )
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+@app.get("/list-text-to-3d-req", tags=["Text-to-3D"])
+async def list_text_to_3d_req(
+    settings: Settings = Depends(get_settings)
+):
+    """
+    List all Text-to-3D tasks.
+    """
+    resp = await app.state.client.get(
+        "https://api.meshy.ai/openapi/v2/text-to-3d",
+        headers={"Authorization": f"Bearer {settings.mesh_api_key}"}
+    )
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+@app.get("/", tags=["Health"])
+async def health_check():
+    """Basic health-check endpoint returning service status."""
+    return {"status": "ok"}
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run("main:app", host="0.0.0.0", port=8000, log_level="info")

requirements.txt ADDED Viewed

	@@ -0,0 +1,7 @@

+fastapi[standard]>=0.70.0
+httpx>=0.23.0
+uvicorn[standard]>=0.18.0
+pydantic>=2.0.0
+pydantic-settings>=2.0.0
+python-dotenv>=0.21.0
+openai