Update use_template_app.py

use_template_app.py

@@ -20,6 +20,46 @@ TEMP_DIR = pathlib.Path("temp_gradio_app_files") # Renamed for clarity
 # Ensure temp directory exists
 TEMP_DIR.mkdir(parents=True, exist_ok=True) # No need to clean it aggressively on each start
 
+# --- Helper Functions ---
+def fix_internal_urls(url_string: str) -> str:
+    """
+    Replaces internal IP address URLs (like http://10.10.71.201:8002) with the external API base URL.
+    This makes internal URLs accessible externally.
+    """
+    if not url_string or not isinstance(url_string, str):
+        return url_string
+    
+    try:
+        # Parse the URL
+        parts = urllib.parse.urlsplit(url_string)
+        
+        # Check if the netloc (domain/IP) looks like an internal address
+        # This covers internal IPs (10.x.x.x, 172.16-31.x.x, 192.168.x.x) and localhost
+        internal_patterns = ['10.', '172.16.', '172.17.', '172.18.', '172.19.', '172.2', '172.3', '192.168.', 'localhost', '127.0.0.1']
+        is_internal = any(parts.netloc.startswith(pattern) for pattern in internal_patterns)
+        
+        if is_internal and parts.scheme in ['http', 'https']:
+            # Keep the path and query components, but replace the domain with TEMPLATE_API_BASE_URL
+            # First, get the external base URL parts
+            base_parts = urllib.parse.urlsplit(TEMPLATE_API_BASE_URL)
+            
+            # Create new URL with base URL's scheme and netloc, but original path and query
+            new_url = urllib.parse.urlunsplit((
+                base_parts.scheme,
+                base_parts.netloc,
+                parts.path,
+                parts.query,
+                parts.fragment
+            ))
+            
+            print(f"URL transformed: {url_string} -> {new_url}")
+            return new_url
+        
+        return url_string
+    except Exception as e:
+        print(f"Error fixing internal URL '{url_string}': {e}")
+        return url_string
+
 # --- Helper Functions ---
 
 def sanitize_url(url_string: str) -> str:
@@ -72,7 +112,20 @@ def fetch_template_details(m_param, u_param, category_param, name_param):
     try:
         response = requests.get(api_url, timeout=10)
         response.raise_for_status()
-        return response.json() 
+        template_data = response.json()
+        
+        # Fix internal URLs in the template details
+        if isinstance(template_data, dict):
+            if "thumbnail_url" in template_data:
+                template_data["thumbnail_url"] = fix_internal_urls(template_data["thumbnail_url"])
+            if "svg_download_url" in template_data:
+                template_data["svg_download_url"] = fix_internal_urls(template_data["svg_download_url"])
+            # Handle any other URLs that might be in the response
+            for key, value in template_data.items():
+                if isinstance(value, str) and (value.startswith("http://") or value.startswith("https://")):
+                    template_data[key] = fix_internal_urls(value)
+        
+        return template_data
     except requests.exceptions.RequestException as e:
         print(f"Error fetching template details from {api_url}: {e}")
         return None
@@ -80,7 +133,9 @@ def fetch_template_details(m_param, u_param, category_param, name_param):
 def download_svg_content(svg_url):
     """Downloads the raw SVG content from a given URL."""
     try:
-        response = requests.get(svg_url, timeout=10)
+        # Fix any internal URLs before downloading
+        fixed_url = fix_internal_urls(svg_url)
+        response = requests.get(fixed_url, timeout=10)
         response.raise_for_status()
         return response.text # SVG is XML text
     except requests.exceptions.RequestException as e:
@@ -334,11 +389,11 @@ def initial_load_template_info(request: gr.Request):
     u = query_params.get("u")
     category = query_params.get("category")
     name = query_params.get("name") # This is the template name
-
+    
     if not all([m, u, category, name]):
         print("Initial load: URL parameters (m, u, category, name) are incomplete or missing.")
         return None, "无模板信息 (请检查URL参数)", None, "无模板"
-
+        
     details = fetch_template_details(m, u, category, name)
     
     local_thumbnail_path = None
@@ -348,10 +403,11 @@ def initial_load_template_info(request: gr.Request):
     if details and "name" in details: 
         template_display_name = details["name"] # Use name from details for display
         svg_download_url = details.get("svg_download_url")
-
+        
         if details.get("thumbnail_url"):
             try:
                 thumb_url = details["thumbnail_url"]
+                # thumb_url is already fixed in fetch_template_details
                 print(f"Fetching thumbnail from: {thumb_url}")
                 thumb_response = requests.get(thumb_url, timeout=10)
                 thumb_response.raise_for_status()