Update app.py

app.py

@@ -23,7 +23,7 @@ ensure_flash_attn()
 from inferencer import UniPicV2Inferencer
 
 
-model_path = snapshot_download(repo_id="Skywork/UniPic2-Metaquery-GRPO-9B")
+model_path = snapshot_download(repo_id="Skywork/UniPic2-Metaquery-9B")
 qwen_vl_path = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct-AWQ")
 
 inferencer = UniPicV2Inferencer(
@@ -76,79 +76,79 @@ def on_submit(history, user_msg, img_path, mode, infer_steps, cfg_scale, model_v
     # seed = int(seed)
     # set_seed(seed)  # 设置随机种子以确保可重复性
 
-    # try:
-    if mode == "Understand Image":
-        if img_path is None:
-            updated_history.append([None, "⚠️ Please upload or generate an image first."])
-            return updated_history, "", img_path
-            
-        raw_output = (
-            inferencer.understand_image(Image.open(img_path), user_msg)
-            if img_path
-            else inferencer.query_text(user_msg)
-        )
-        clean_output = raw_output
-        return (
-            updated_history + [(None, clean_output)],
-            "",
-            img_path,
-        )  # 保持 img_path 不变
-
-    if mode == "Generate Image":
-        if not user_msg:
+    try:
+        if mode == "Understand Image":
+            if img_path is None:
+                updated_history.append([None, "⚠️ Please upload or generate an image first."])
+                return updated_history, "", img_path
+                
+            raw_output = (
+                inferencer.understand_image(Image.open(img_path), user_msg)
+                if img_path
+                else inferencer.query_text(user_msg)
+            )
+            clean_output = raw_output
             return (
-                updated_history
-                + [(None, "⚠️ Please enter your message for generating.")],
+                updated_history + [(None, clean_output)],
                 "",
                 img_path,
-            )
-        imgs = inferencer.generate_image(user_msg, num_inference_steps=infer_steps, seed=seed, guidance_scale=cfg_scale)
-        path = save_temp_image(imgs[0])
-        return (
-            updated_history
-            + [
-                (None, (path,)),
-                (
-                    None,
-                    "✅ You can continue editing this image by switching the mode to 'Edit Image' and entering your instruction. ",
-                ),
-            ],
-            "",
-            path,
-        )  # 更新 img_state
-
-    elif mode == "Edit Image":
-        if img_path is None:
+            )  # 保持 img_path 不变
+    
+        if mode == "Generate Image":
+            if not user_msg:
+                return (
+                    updated_history
+                    + [(None, "⚠️ Please enter your message for generating.")],
+                    "",
+                    img_path,
+                )
+            imgs = inferencer.generate_image(user_msg, num_inference_steps=infer_steps, seed=seed, guidance_scale=cfg_scale)
+            path = save_temp_image(imgs[0])
             return (
                 updated_history
-                + [(None, "⚠️ Please upload or generate an image first.")],
+                + [
+                    (None, (path,)),
+                    (
+                        None,
+                        "✅ You can continue editing this image by switching the mode to 'Edit Image' and entering your instruction. ",
+                    ),
+                ],
                 "",
-                img_path,
-            )
-        if not user_msg:
+                path,
+            )  # 更新 img_state
+    
+        elif mode == "Edit Image":
+            if img_path is None:
+                return (
+                    updated_history
+                    + [(None, "⚠️ Please upload or generate an image first.")],
+                    "",
+                    img_path,
+                )
+            if not user_msg:
+                return (
+                    updated_history
+                    + [(None, "⚠️ Please enter your message for editing.")],
+                    "",
+                    img_path,
+                )
+            edited_img = inferencer.edit_image(Image.open(img_path), user_msg, num_inference_steps=infer_steps, seed=seed, guidance_scale=cfg_scale)[0]
+            path = save_temp_image(edited_img)
             return (
                 updated_history
-                + [(None, "⚠️ Please enter your message for editing.")],
+                + [
+                    (None, (path,)),
+                    (
+                        None,
+                        "✅ You can continue editing this image by entering your instruction. ",
+                    ),
+                ],
                 "",
-                img_path,
-            )
-        edited_img = inferencer.edit_image(Image.open(img_path), user_msg, num_inference_steps=infer_steps, seed=seed, guidance_scale=cfg_scale)[0]
-        path = save_temp_image(edited_img)
-        return (
-            updated_history
-            + [
-                (None, (path,)),
-                (
-                    None,
-                    "✅ You can continue editing this image by entering your instruction. ",
-                ),
-            ],
-            "",
-            path,
-        )  # 更新 img_state
+                path,
+            )  # 更新 img_state
 
-    # except Exception as e:
-    #     return updated_history + [(None, f"⚠️ Failed to process: {e}")], "", img_path
+    except Exception as e:
+        return updated_history + [(None, f"⚠️ Failed to process: {e}")], "", img_path
 
 
 # 定义CSS样式（兼容旧版 Gradio）