Update app.py

app.py

@@ -4,58 +4,64 @@ 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 __init__(self,):
     self.query=''
 
+  def _generate_3d_(self,query,api_key):
+      #api key from meshy website
+      try:
+        headers = {
+        "Authorization": f"Bearer {api_key}"
+        }
+        self.query = query
+        # print(f"query is {self.query}")
+        payload = {
+        "mode": "preview",
+        "prompt":query,
+        "art_style": "realistic",
+        "seed":1565957778,
+        "negative_prompt":"messy hands and feet, blurry mess, three eyes",
+        "texture_richness":"high"
+        }
+
+        generate_3d= requests.post(
+        "https://api.meshy.ai/v2/text-to-3d",
+        headers=headers,
+        json=payload,
+        )
+        task_id=generate_3d.json()["result"]
+       
+        return task_id
+      except Exception as e:
+        print("Error:",e)
+  def _get_task_id_(self,query,api_key):
 
+    try:
 
-  def _generate_3d_(self,query):
-      self.query = query
-      print(f"query is {self.query}")
-      payload = {
-      "mode": "preview",
-      "prompt":query,
-      "art_style": "realistic",
-      "seed":1565957778,
-      "negative_prompt":"messy hands and feet, blurry mess, three eyes",
-      "texture_richness":"high"
+      headers = {
+      "Authorization": f"Bearer {api_key}"
       }
 
-      generate_3d= requests.post(
-      "https://api.meshy.ai/v2/text-to-3d",
-      headers=self.headers,
-      json=payload,
-      )
-      task_id=generate_3d.json()["result"]
-
-      return task_id
-  def _get_task_id_(self,query):
-
-    try:
-      task_id=self._generate_3d_(query)
+      task_id=self._generate_3d_(query,api_key)
 
       time.sleep(50)
 
       get_asset= requests.get(
         f"https://api.meshy.ai/v2/text-to-3d/{task_id}",
-        headers=self.headers,
+        headers=headers,
         )
 
 
       download_3d=get_asset.json()["model_urls"]["glb"]
-
+     
       return download_3d
     except:
       print("error on _get_task_id_")
 
-  def _download_3d(self,query):
+  def _download_3d(self,query,api_key):
     try:
 
-      download_3d=self._get_task_id_(query)
+      download_3d=self._get_task_id_(query,api_key)
       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
 
@@ -71,6 +77,9 @@ class Model3d_generater:
       gr.HTML("""
             <center><h1 style="color:#fff">3d Model Generater</h1></center>""")
 
+      with gr.Row():
+            api_key=gr.Textbox(label="API key",placeholder="API key from meshy website")
+
       with gr.Row():
         prompt=gr.Textbox(label="Prompt")
 
@@ -79,9 +88,9 @@ class Model3d_generater:
       with gr.Row():
         button=gr.Button()
 
-      button.click(self._download_3d,prompt,output)
+      button.click(self._download_3d,[prompt,api_key],output)
 
-    demo.launch()
+    demo.launch(debug=True)
 
 if __name__=="__main__":
   result=Model3d_generater()