Update app.py

app.py

@@ -1,50 +1,47 @@
 import gradio as gr
 import replicate
 import requests
-import os
 import uuid
+import os
 
-# Set your Replicate API token
+# Set your Replicate API token securely
 os.environ["REPLICATE_API_TOKEN"] = "r8_FqBcLY8OR6F3Fsf4D4fT415tb1CN3790UGzYE"
 
 def generate_3d(prompt):
-    print("Prompt received:", prompt)  # Check if button works
-def generate_3d(prompt):
-    print("Prompt received:", prompt)
     try:
-        output = output = replicate.run(
-    "lucataco/text-to-3d:e461273fe25ac0b2b3fe14b7551636ee58b78f12fe25aa0f141e4cd32b2f3c2d",
-    input={"prompt": prompt}
+        print("Prompt received:", prompt)
+
+        output = replicate.run(
+            "lucataco/text-to-3d:e461273fe25ac0b2b3fe14b7551636ee58b78f12fe25aa0f141e4cd32b2f3c2d",
+            input={"prompt": prompt}
         )
-        print("Replicate output:", output)
 
-        if not output or not isinstance(output, list):
-            return "Error: No model output."
+        print("Model output:", output)
 
-        glb_url = output[0]
-        print("GLB URL:", glb_url)
+        if not output or not isinstance(output, str):
+            return None
 
-        response = requests.get(glb_url)
+        response = requests.get(output)
         if response.status_code != 200:
-            print("Download failed with status:", response.status_code)
-            return f"Download failed: {response.status_code}"
+            print("Download failed:", response.status_code)
+            return None
 
         file_name = f"{uuid.uuid4()}.glb"
         with open(file_name, 'wb') as f:
             f.write(response.content)
 
-        print("Saved to:", file_name)
         return file_name
 
     except Exception as e:
-        print("Exception occurred:", str(e))
-        return f"Generation failed: {str(e)}"
-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")
-
-    btn = gr.Button("Generate")
-    btn.click(fn=generate_3d, inputs=prompt, outputs=output)
-
-demo.launch()
+        print("Exception:", str(e))
+        return None
+
+demo = gr.Interface(
+    fn=generate_3d,
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs=gr.Model3D(label="Generated 3D Model"),
+    title="Text to 3D Model Viewer",
+    description="Enter a prompt to generate and view a 3D model",
+)
+
+demo.launch(share=True)