Update app.py


app.py

@@ -10,14 +10,36 @@ from gradio_client import Client, handle_file
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def image_to_3d(image_path:str)-> dict: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that uses the HuggingFace Space "frogleo/Image-to-3D" to transform a 2D image into a 3D model
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        image_path: A string with the path to the 2D image that will be used
+        
     """
-    return "What magic will you build ?"
+    
+    client = Client("frogleo/Image-to-3D")
+    result = client.predict(
+    		image=image_path,
+    		steps=5,
+    		guidance_scale=5.5,
+    		seed=1234,
+    		octree_resolution=256,
+    		num_chunks=8000,
+    		target_face_num=10000,
+    		randomize_seed=True,
+    		api_name="/gen_shape"
+    )
+    print(result)
+    
+    return {
+        "glb_path": result[2],
+        "obj_path": result[3],
+        "file_export": result[1],
+        "html_preview": result[0]
+    }
+
+    
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -56,7 +78,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_to_3d], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,