| import gradio as gr |
| import replicate |
| import requests |
| import uuid |
| import os |
|
|
| |
| os.environ["REPLICATE_API_TOKEN"] = "r8_FqBcLY8OR6F3Fsf4D4fT415tb1CN3790UGzYE" |
|
|
| def generate_3d(prompt): |
| try: |
| print("Prompt received:", prompt) |
|
|
| output = replicate.run( |
| "lucataco/text-to-3d:e461273fe25ac0b2b3fe14b7551636ee58b78f12fe25aa0f141e4cd32b2f3c2d", |
| input={"prompt": prompt} |
| ) |
|
|
| print("Model output:", output) |
|
|
| if not output or not isinstance(output, str): |
| return None |
|
|
| response = requests.get(output) |
| if response.status_code != 200: |
| 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) |
|
|
| return file_name |
|
|
| except Exception as e: |
| 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) |
