fix some bugs

.gitignore

@@ -0,0 +1 @@
+__pycache__

Rodin.py

@@ -103,21 +103,22 @@ class Generator:
         _, self.token = login(user_id, password)
         self.task_uuid = None
         
-    def preprocess(self, prompt, image_path):
+    def preprocess(self, prompt, image_path, cache_image_base64, task_uuid=""):
+        if cache_image_base64 and prompt and task_uuid != "":
+            return prompt, cache_image_base64
+        
         image_file = open(image_path, 'rb')
-
         if image_file == None:
             print("Invalid image file.")
-
         try:
-            if not prompt:
-                preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name="images.jpeg", token=self.token)
-            else:
+            if prompt and task_uuid:
                 preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name="images.jpeg", token=self.token)
+            else:
+                preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name="images.jpeg", token=self.token)
             if 'error' in preprocess_response:
                 print("Error in image preprocessing:", preprocess_response['error'])
             else:
-                if not prompt:
+                if not (prompt and task_uuid):
                     prompt = preprocess_response.get('prompt', 'Default prompt if none returned')                
                 processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
         finally:
@@ -149,8 +150,9 @@ class Generator:
             subscription_key = update_response['job']['subscription_key']
             checker = JobStatusChecker(BASE_URL, subscription_key)
             checker.start()
-            
-        preview_image = rodin_history(task_uuid, self.token)["v1"]["preview_image"]
+        
+        preview_image = next(reversed(rodin_history(task_uuid, self.token).items()))[1]["preview_image"]
+        print(f"Preview image URL: {rodin_history(task_uuid, self.token)}")
         response = requests.get(preview_image, stream=True)
         if response.status_code == 200:
             # 创建一个PIL Image对象

app.py

@@ -1,5 +1,5 @@
 import os
-# os.system('pip uninstall -y gradio_fake3d')
+os.system('pip uninstall -y gradio_fake3d')
 os.system('pip install gradio_fake3d-0.0.2-py3-none-any.whl')
 
 import gradio as gr
@@ -152,7 +152,7 @@ with gr.Blocks() as demo:
     cache_raw_image = gr.Image(visible=False, type="pil")
     cache_image_base64 = gr.Text(visible=False)
     cacha_empty = gr.Text(visible=False)
-    cache_task_uuid = gr.Text(value="", visible=False)
+    cache_task_uuid = gr.Text(value="")
 
     # button_generate.click(fn=return_render, inputs=[block_image], outputs=[raw_image, fake3d])
     block_image.change(
@@ -164,7 +164,7 @@ with gr.Blocks() as demo:
     
     button_generate.click(
         fn=generator.preprocess, 
-        inputs=[block_prompt, block_image], 
+        inputs=[block_prompt, block_image, cache_image_base64, cache_task_uuid], 
         outputs=[block_prompt, cache_image_base64],
         show_progress="minimal"
     ).success(