update

app.py

@@ -1,7 +1,8 @@
-
 import cv2
 from mtcnn.mtcnn import MTCNN
 from utils import *
+import requests
+from urllib.parse import urlparse
 
 
 cloth_examples = get_cloth_examples(hr=0)
@@ -20,6 +21,28 @@ description = r"""
 """
 
 
+def is_http_resource_accessible(url):
+    """
+    Check if HTTP resource is accessible
+    Returns True if accessible, False otherwise
+    """
+    if not url or not isinstance(url, str):
+        return False
+    
+    try:
+        # Parse URL to check if it's a valid HTTP/HTTPS URL
+        parsed = urlparse(url)
+        if parsed.scheme not in ['http', 'https']:
+            return False
+        
+        # Make a HEAD request to check if resource exists
+        response = requests.head(url, timeout=5, allow_redirects=True)
+        return response.status_code == 200
+    except Exception as e:
+        print(f"Error checking resource accessibility: {e}")
+        return False
+
+
 def onClick(cloth_image, pose_image, high_resolution, request: gr.Request):
     if pose_image is None:
         yield None, "no pose image found !", ""
@@ -78,7 +101,9 @@ def onClick(cloth_image, pose_image, high_resolution, request: gr.Request):
             return None, "fail to public you task", ""
 
         print(client_ip, public_res['mid_result'])
-        yield public_res['mid_result'], f"task is processing, task id: {public_res['id']}, {public_res['msg']}", ""
+        # Check if mid_result resource is accessible
+        mid_result = public_res['mid_result'] if is_http_resource_accessible(public_res['mid_result']) else None
+        yield mid_result, f"task is processing, task id: {public_res['id']}, {public_res['msg']}", ""
 
         max_try = 120*3
         wait_s = 0.5
@@ -87,9 +112,13 @@ def onClick(cloth_image, pose_image, high_resolution, request: gr.Request):
             state = getInfRes(public_res['id'])
             timestamp = int(time.time() * 1000)
             if state is None:
-                yield public_res['mid_result'] + f"?t={timestamp}", "task query failed,", ""
+                mid_result = public_res['mid_result'] if is_http_resource_accessible(public_res['mid_result']) else None
+                result_url = mid_result + f"?t={timestamp}" if mid_result else None
+                yield result_url, "task query failed,", ""
             elif state['status']=='PROCESSING':
-                yield public_res['mid_result'] + f"?t={timestamp}", f"task is processing, query {i}", ""
+                mid_result = public_res['mid_result'] if is_http_resource_accessible(public_res['mid_result']) else None
+                result_url = mid_result + f"?t={timestamp}" if mid_result else None
+                yield result_url, f"task is processing, query {i}", ""
             elif state['status']=='SUCCEED':
                 yield state['output1'] + f"?t={timestamp}", f"task finished, {state['msg']}", ""
                 return state['output1'] + f"?t={timestamp}", f"task finished, {state['msg']}", ""
@@ -97,7 +126,9 @@ def onClick(cloth_image, pose_image, high_resolution, request: gr.Request):
                 yield None, f"task failed, {state['msg']}", ""
                 return None, f"task failed, {state['msg']}", ""
             else:
-                yield public_res['mid_result'] + f"?t={timestamp}", f"task is on processing, query {i}", ""
+                mid_result = public_res['mid_result'] if is_http_resource_accessible(public_res['mid_result']) else None
+                result_url = mid_result + f"?t={timestamp}" if mid_result else None
+                yield result_url, f"task is on processing, query {i}", ""
         return None, "no machine...", ""
     except Exception as e:
         print(e)