fix langsegment

app.py

@@ -1,5 +1,7 @@
 import os
 import sys
+import importlib.util
+import site
 import json
 import torch
 import gradio as gr
@@ -9,37 +11,87 @@ from huggingface_hub import snapshot_download, hf_hub_download
 import subprocess
 import re
 
-# 修复LangSegment包导入问题
-def fix_langsegment_imports():
+def patch_langsegment_init():
     try:
-        import LangSegment
-        langsegment_init_path = os.path.join(os.path.dirname(LangSegment.__file__), '__init__.py')
-        
-        if os.path.exists(langsegment_init_path):
-            with open(langsegment_init_path, 'r') as f:
-                content = f.read()
-            
-            # 检查是否存在已废弃的导入
-            if 'setLangfilters' in content or 'getLangfilters' in content:
-                print("正在修复LangSegment包中的导入问题...")
-                # 替换导入语句，删除setLangfilters和getLangfilters
-                fixed_content = re.sub(
-                    r'from \.LangSegment import LangSegment,([^,]*),setLangfilters,getLangfilters',
-                    r'from .LangSegment import LangSegment,\1',
-                    content
-                )
-                
-                # 写回文件
-                with open(langsegment_init_path, 'w') as f:
-                    f.write(fixed_content)
-                print("LangSegment包导入修复完成")
+        # 尝试找到 LangSegment 包的位置
+        spec = importlib.util.find_spec("LangSegment")
+        if spec is None or spec.origin is None:
+            print("无法定位 LangSegment 包。")
+            return
+
+        # 构建 __init__.py 的路径
+        init_path = os.path.join(os.path.dirname(spec.origin), '__init__.py')
+        
+        if not os.path.exists(init_path):
+            print(f"未找到 LangSegment 的 __init__.py 文件于: {init_path}")
+            # 尝试在 site-packages 中查找，适用于某些环境
+            for site_pkg_path in site.getsitepackages():
+                potential_path = os.path.join(site_pkg_path, 'LangSegment', '__init__.py')
+                if os.path.exists(potential_path):
+                    init_path = potential_path
+                    print(f"在 site-packages 中找到 __init__.py: {init_path}")
+                    break
+            else: # 如果循环正常结束（没有 break）
+                 print(f"在 site-packages 中也未找到 __init__.py")
+                 return
+
+
+        print(f"尝试读取 LangSegment __init__.py: {init_path}")
+        with open(init_path, 'r') as f:
+            lines = f.readlines()
+
+        modified = False
+        new_lines = []
+        target_line_prefix = "from .LangSegment import"
+
+        for line in lines:
+            stripped_line = line.strip()
+            if stripped_line.startswith(target_line_prefix):
+                if 'setLangfilters' in stripped_line or 'getLangfilters' in stripped_line:
+                    print(f"发现需要修改的行: {stripped_line}")
+                    # 移除 setLangfilters 和 getLangfilters
+                    modified_line = stripped_line.replace(',setLangfilters', '')
+                    modified_line = modified_line.replace(',getLangfilters', '')
+                    # 确保逗号处理正确 (例如，如果它们是末尾的项)
+                    modified_line = modified_line.replace('setLangfilters,', '')
+                    modified_line = modified_line.replace('getLangfilters,', '')
+                    # 如果它们是唯一的额外导入，移除可能多余的逗号
+                    modified_line = modified_line.rstrip(',') 
+                    new_lines.append(modified_line + '\n')
+                    modified = True
+                    print(f"修改后的行: {modified_line.strip()}")
+                else:
+                    new_lines.append(line) # 行没问题，保留原样
             else:
-                print("LangSegment包无需修复")
+                new_lines.append(line) # 非目标行，保留原样
+
+        if modified:
+            print(f"尝试写回已修改的 LangSegment __init__.py 到: {init_path}")
+            try:
+                with open(init_path, 'w') as f:
+                    f.writelines(new_lines)
+                print("LangSegment __init__.py 修改成功。")
+                # 尝试重新加载模块以使更改生效（可能无效，取决于导入链）
+                try:
+                    import LangSegment
+                    importlib.reload(LangSegment)
+                    print("LangSegment 模块已尝试重新加载。")
+                except Exception as reload_e:
+                     print(f"重新加载 LangSegment 时出错（可能无影响）: {reload_e}")
+            except PermissionError:
+                print(f"错误：权限不足，无法修改 {init_path}。请考虑修改 requirements.txt。")
+            except Exception as write_e:
+                print(f"写入 LangSegment __init__.py 时发生其他错误: {write_e}")
+        else:
+            print("LangSegment __init__.py 无需修改。")
+
+    except ImportError:
+         print("未找到 LangSegment 包，无法进行修复。")
     except Exception as e:
-        print(f"修复LangSegment包时出错: {e}")
+        print(f"修复 LangSegment 包时发生意外错误: {e}")
 
-# 在导入其他模块之前执行修复
-fix_langsegment_imports()
+# 在所有其他导入（尤其是可能触发 LangSegment 导入的 Amphion）之前执行修复
+patch_langsegment_init()
 
 # 克隆Amphion仓库
 if not os.path.exists("Amphion"):