Update app.py

app.py

@@ -4,7 +4,10 @@ import time
 import sys
 import shutil
 from pathlib import Path
+from unittest.mock import MagicMock
 
+# ==========================================================
+# 1. SETUP & BUGFIXES
 # ==========================================================
 if os.environ.get("SPACE_ID"):
     # Fix für Rendering: Xvfb nur starten, wenn noch nicht aktiv
@@ -21,11 +24,21 @@ if os.environ.get("SPACE_ID"):
         "MESA_GL_VERSION_OVERRIDE": "4.1"
     })
 
-# Fix für Gradio Schema-Fehler
+# Fix A: Gradio Schema-Fehler
 import gradio_client.utils as client_utils
 client_utils._json_schema_to_python_type = lambda *args, **kwargs: "Any"
 client_utils.json_schema_to_python_type = lambda *args, **kwargs: "Any"
 
+# Fix B: Flash Attention Mocking (verhindert ImportError)
+try:
+    import flash_attn
+except ImportError:
+    mock = MagicMock()
+    sys.modules["flash_attn"] = mock
+    sys.modules["flash_attn.modules"] = mock
+    sys.modules["flash_attn.modules.mha"] = mock
+    print("Flash Attention not found, using Mock for compatibility.")
+
 # ==========================================================
 # 2. CORE IMPORTS (Erst nach dem Setup!)
 # ==========================================================