first try to get the .glb file

main.py

@@ -1,13 +1,11 @@
-# app.py
 import os
 import shutil
 import tempfile
 import zipfile
 from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import JSONResponse
+from fastapi.responses import FileResponse
 from typing import List
 import subprocess
-# import torch
 
 app = FastAPI()
 
@@ -57,7 +55,15 @@ async def upload_zip(file: UploadFile = File(...)):
         except subprocess.CalledProcessError as e:
             raise HTTPException(status_code=500, detail=f"3D model generation failed: {str(e)}")
 
-        return JSONResponse(content={"detail": "3D model generation completed successfully", "output_dir": output_dir})
+        # Find the .glb file in the output directory
+        glb_files = [os.path.join(output_dir, f) for f in os.listdir(output_dir) if f.endswith('.glb')]
+        if not glb_files:
+            raise HTTPException(status_code=500, detail="No .glb file generated")
+
+        glb_file_path = glb_files[0]
+
+        # Return the .glb file as a response
+        return FileResponse(glb_file_path, media_type='application/octet-stream', filename=os.path.basename(glb_file_path))
 
 if __name__ == '__main__':
     import uvicorn