add example

Rodin.py

@@ -7,6 +7,7 @@ import base64
 import io
 from PIL import Image
 from io import BytesIO
+import gradio as gr
 from requests_toolbelt.multipart.encoder import MultipartEncoder
 from constant import *
 
@@ -53,6 +54,8 @@ def rodin_preprocess_image(generate_prompt, image, name, token):
     return response.json()
 
 def crop_image(image, type):
+    if image == None:
+        raise gr.Error("Generate Image First")
     new_image_width = 360 * (11520 // 720)  # 每隔720像素裁切一次，每次裁切宽度为360
     new_image_height = 360  # 新图片的高度
     new_image = Image.new('RGB', (new_image_width, new_image_height))
@@ -135,7 +138,7 @@ class Generator:
         self.processed_image = None
         
     def preprocess(self, prompt, image_path, processed_image , task_uuid=""):
-        if processed_image and prompt and task_uuid != "":
+        if processed_image and prompt and (not task_uuid):
             log("INFO", "Using cached image and prompt...")
             return prompt, processed_image
         log("INFO", "Preprocessing image...")
@@ -205,7 +208,7 @@ class Generator:
         try:
             preview_image = next(reversed(history.items()))[1]["preview_image"]
         except Exception as e:
-            print(history)
+            log("ERROR", f"Error in generating mesh: {history}")
             raise e
         response = requests.get(preview_image, stream=True)
         if response.status_code == 200:

app.py

@@ -1,6 +1,6 @@
 import os
-os.system('pip uninstall -y gradio_fake3d')
-os.system('pip install gradio_fake3d-0.0.3-py3-none-any.whl')
+# os.system('pip uninstall -y gradio_fake3d')
+# os.system('pip install gradio_fake3d-0.0.3-py3-none-any.whl')
 
 import gradio as gr
 import re
@@ -61,16 +61,16 @@ options = [
 ]
 
 example = [
-    "assets/00.png",
-    "assets/07.png",
-    "assets/08.png",
-    "assets/11.png",
-    "assets/13.png",
-    "assets/21.png",
-    "assets/24.PNG",
-    "assets/30.png",
-    "assets/42.png",
-    "assets/46.png",
+    ["assets/00.png"],
+    ["assets/07.png"],
+    ["assets/08.png"],
+    ["assets/11.png"],
+    ["assets/13.png"],
+    ["assets/21.png"],
+    ["assets/24.PNG"],
+    ["assets/30.png"],
+    ["assets/42.png"],
+    ["assets/46.png"]
 ]
 
 def do_nothing(text):
@@ -107,8 +107,11 @@ def handle_prompt_change(prompt):
     return hint_list
 
 def clear_task(task_input=None):
+    """_summary_
+    [cache_task_uuid, block_prompt, block_prompt_hint, fake3d]
+    """
     log("INFO", "Clearing task...")
-    return "", "", [], None
+    return "", "", [], "assets/white_image.png"
 
 def clear_task_id():
     return ""
@@ -160,22 +163,22 @@ with gr.Blocks() as demo:
                             block_normal = gr.Button("Normal", min_width=0)
             
             button_more = gr.Button(value="Download", variant="primary", link=rodin_url)
+            cache_task_uuid = gr.Text(value="", visible=False)
+
             
     cache_raw_image = gr.Image(visible=False, type="pil")
     cacha_empty = gr.Text(visible=False)
     cache_image_base64 = gr.Text(visible=False)
-    cache_task_uuid = gr.Text(value="", visible=False)
-    
-    # block_example = gr.Examples(
-    #     examples=example,
-    #     fn=clear_task, 
-    #     inputs=[block_image], 
-    #     outputs=[cache_task_uuid, block_prompt, block_prompt_hint, fake3d], 
-    #     run_on_click=True,
-    #     label="Examples"
-    # )
-    # blcok_input_example = gr.Gallery(value=example, columns=10, allow_preview=False, interactive=False)
-    
+    block_example = gr.Examples(
+        examples=example,
+        fn=clear_task,
+        inputs=[block_image],
+        outputs=[cache_task_uuid, block_prompt, block_prompt_hint, fake3d],
+        run_on_click=True,
+        cache_examples=True,
+        label="Examples"
+    )  
+
     block_image.upload(
         fn=do_nothing, 
         js=change_button_name_to_generating, 
@@ -208,13 +211,6 @@ with gr.Blocks() as demo:
         show_progress="hidden"
     )
     
-    # block_image.change(
-    #     fn=do_nothing, 
-    #     js=reset_button_name, 
-    #     inputs=[cacha_empty],
-    #     outputs=[cacha_empty]
-    # )
-    
     button_generate.click(
         fn=do_nothing, 
         js=change_button_name_to_generating, 
@@ -261,4 +257,4 @@ with gr.Blocks() as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(show_api=False)
+    demo.launch(show_api=False, show_error=True)