Update app.py

app.py

@@ -1,9 +1,35 @@
 import os
 import sys
-import logging
+import shutil
 
 # ==========================================
-# 1. 基础环境净化 (屏蔽显卡)
+# 1. 物理层攻击：篡改模型文件
+# ==========================================
+print("🔨 正在执行物理覆盖手术...")
+
+cwd = os.getcwd()
+cpu_model_path = os.path.join(cwd, "AR", "models", "t2s_model.py")
+gpu_model_path = os.path.join(cwd, "AR", "models", "t2s_model_flash_attn.py")
+
+if os.path.exists(cpu_model_path) and os.path.exists(gpu_model_path):
+    try:
+        # 读取 CPU 代码
+        with open(cpu_model_path, "r", encoding="utf-8") as f:
+            cpu_code = f.read()
+        
+        # 写入 GPU 文件 (鸠占鹊巢)
+        with open(gpu_model_path, "w", encoding="utf-8") as f:
+            f.write(cpu_code)
+            
+        print(f"✅ 覆盖成功！\n源: {cpu_model_path}\n目标: {gpu_model_path}")
+        print("现在 't2s_model_flash_attn.py' 里的代码实际上是 CPU 版代码。")
+    except Exception as e:
+        print(f"❌ 覆盖失败 (权限不足?): {e}")
+else:
+    print("⚠️ 找不到模型文件路径，跳过覆盖步骤 (可能目录结构变了)")
+
+# ==========================================
+# 2. 基础环境设置
 # ==========================================
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 import torch
@@ -16,41 +42,15 @@ torch.nn.Module.cuda = no_op
 print("💉 CUDA 已屏蔽。")
 
 # ==========================================
-# 2. 核心魔法：内存类替换 (Class Swapping)
+# 3. 导入业务逻辑
 # ==========================================
-sys.path.append(os.getcwd())
+sys.path.append(cwd)
 
-try:
-    print("🧠 开始执行内存调包手术...")
-    
-    # 1. 先导入 CPU 版的模型类 (这是我们想要的)
-    from AR.models.t2s_model import Text2SemanticDecoder as CPU_Decoder
-    print("✅ 成功提取 CPU 版模型类")
-    
-    # 2. 再导入 GPU 版的模块 (这是我们要覆盖的)
-    import AR.models.t2s_model_flash_attn as GPU_Module
-    
-    # 3. 【关键】直接把 GPU 模块里的类，换成 CPU 版的类
-    # 这样后续任何代码(包括 inference_webui)去引用 GPU_Module.Text2SemanticDecoder
-    # 实际上都会拿到 CPU_Decoder
-    GPU_Module.Text2SemanticDecoder = CPU_Decoder
-    
-    print("💉 手术成功：FlashAttn 模块已被 CPU 内核接管！")
-    
-except ImportError as e:
-    print(f"⚠️ 手术失败 (可能路径不对): {e}")
-    print(f"当前目录文件: {os.listdir('.')}")
-    if os.path.exists("AR"):
-        print(f"AR目录文件: {os.listdir('AR')}")
-
-# ==========================================
-# 3. 导入业务逻辑 (必须在手术后进行)
-# ==========================================
 try:
     import inference_webui as core
     print("✅ 成功导入 inference_webui")
     
-    # 强制修改全局配置
+    # 双重保险：配置修改
     if hasattr(core, "is_half"): core.is_half = False
     if hasattr(core, "device"): core.device = "cpu"
         
@@ -83,12 +83,14 @@ sovits_path = find_real_model("s2Gv2ProPlus.pth") or find_real_model("s2G")
 # ==========================================
 try:
     if gpt_path and sovits_path:
-        core.is_half = False # 再次确保
+        # 三重保险
+        core.is_half = False
+        
         if hasattr(core, "change_gpt_weights"):
             core.change_gpt_weights(gpt_path=gpt_path)
         if hasattr(core, "change_sovits_weights"):
             core.change_sovits_weights(sovits_path=sovits_path)
-        print(f"🎉 模型加载成功！(CPU Mode)")
+        print(f"🎉 模型加载成功！(HARDWARE HACKED)")
     else:
         print("❌ 未找到模型文件")
 except Exception as e:
@@ -141,10 +143,10 @@ def run_predict(text):
 # 7. 界面
 # ==========================================
 with gr.Blocks() as app:
-    gr.Markdown(f"### GPT-SoVITS V2 (CPU Class Swapped)")
+    gr.Markdown(f"### GPT-SoVITS V2 (File System Override)")
     
     with gr.Row():
-        inp = gr.Textbox(label="文本", value="这下真的真的真的要成功了。")
+        inp = gr.Textbox(label="文本", value="别报错了，求求了。")
         btn = gr.Button("生成")
     
     with gr.Row():