Create app.py

app.py

@@ -0,0 +1,96 @@
+import requests
+import gradio as gr
+import time
+import os
+class Model3d_generater:
+
+  def __init__(self):
+    api_key = os.getenv("MESHY_API_KEY")
+    self.headers = {
+      "Authorization": f"Bearer {api_key}"
+      }
+
+  def _get_access_(self,query,style_prompt,resolution):
+
+      payload = {
+      "object_prompt":query,
+      "style_prompt":style_prompt,
+      "enable_pbr": True,
+      "resolution":resolution,
+      "art_style": "realistic",
+      # "negative_prompt": "low quality, low resolution, low poly, ugly"
+      }
+
+      generate_3d = requests.post(
+      "https://api.meshy.ai/v1/text-to-3d",
+      headers=self.headers,
+      json=payload,
+      )
+
+      task_id=generate_3d.json()["result"]
+      return task_id
+
+  def check_status(self,task_id):
+
+    response =requests.get(  f"https://api.meshy.ai/v2/text-to-3d/{task_id}",
+      headers=self.headers,)
+    if response.status_code == 200:
+      content=response.content
+      status=response.json()
+      return content,status
+    else:
+      return "Failed to get status"
+
+  def _generate_3d_(self,query,style_prompt,resolution):
+      task_id=self._get_access_(query,style_prompt,resolution)
+      while True:
+          content, status= self.check_status(task_id)
+          if status['status'] == 'SUCCEEDED':
+              print("Process complete. Model URLs:")
+              download_3d=status["model_urls"]["glb"]
+              response = requests.get(download_3d, allow_redirects=False)
+              short_filename = '3d_asset.glb'  # Change 'my_file.ext' to whatever you prefer, with the correct extension
+
+              with open(short_filename, 'wb') as file:
+                  file.write(response.content)
+              return short_filename
+
+          elif status and status['status'] == 'FAILED':
+              print("An error occurred during processing.")
+              break
+          else:
+              print("Processing... Please wait.")
+
+              time.sleep(30)
+
+  def interface(self):
+    css=""".gradio-container {background: rgb(157,228,255);
+        background: radial-gradient(circle, rgba(157,228,255,1) 0%, rgba(18,115,106,1) 100%);}
+
+        }
+        """
+    with gr.Blocks(css=css) as demo:
+      gr.HTML("""
+            <center><h1 style="color:#fff">3d Model Generater</h1></center>""")
+
+      with gr.Row():
+        prompt=gr.Textbox(label="Prompt")
+        with gr.Column(scale=0.15):
+          resolution=gr.Dropdown(choices=["1024","2048","4096"],label="Resolution")
+
+      with gr.Row():
+        style=gr.Textbox(placeholder="describe the style like HDR,high quality,extrac",label="Style Prompt")
+
+      with gr.Row():
+        output=gr.Model3D()
+
+      with gr.Row():
+        button=gr.Button()
+
+      button.click(self._generate_3d_,[prompt,style,resolution],outputs=output)
+
+    demo.launch()
+
+if __name__=="__main__":
+  result=Model3d_generater()
+  result.interface()