Update app.py

gradio
replicate

app.py

@@ -1,15 +1,30 @@
 import gradio as gr
+import replicate
+import os
+import uuid
+
+# Set your Replicate API token
+os.environ["REPLICATE_API_TOKEN"] = "r8_FqBcLY8OR6F3Fsf4D4fT415tb1CN3790UGzYE"
 
 def generate_3d(prompt):
-    # Simulate generation with a placeholder model
-    # Replace this later with actual model generation logic
-    return "sci_fi_sword.glb"
+    output = replicate.run(
+        "openai/shap-e:9c9ecb8d35d0cc0ecde68b8a0d4a2f94f2bc6ee9393e8c3d76c44d7c0d75b1f9",
+        input={"prompt": prompt, "guidance_scale": 15, "num_inference_steps": 64}
+    )
+
+    # output = [url_to_glb_file]
+    glb_url = output[0]
+
+    # Download the .glb file and return local filename
+    file_name = f"{uuid.uuid4()}.glb"
+    os.system(f"wget {glb_url} -O {file_name}")
+    return file_name
 
 with gr.Blocks() as demo:
     gr.Markdown("## 3D Model Generator (Text to 3D)")
     prompt = gr.Textbox(label="Prompt", placeholder="e.g. A sci-fi sword")
-    output = gr.Model3D(label="Generated 3D Model", clear_color=[1.0, 1.0, 1.0, 1.0])
-
+    output = gr.Model3D(label="Generated 3D Model")
+    
     btn = gr.Button("Generate")
     btn.click(fn=generate_3d, inputs=prompt, outputs=output)