Update app.py

app.py

@@ -19,6 +19,10 @@ from datetime import datetime
 import gradio as gr
 from huggingface_hub import InferenceClient
 
+# Added missing imports
+import tempfile
+import zipfile
+
 # ---------- Config ----------
 PYTHON_MODEL = "Nxcode-CQ-7B-orpo"
 OTHER_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
@@ -73,20 +77,21 @@ def get_token_from_env_or_manual(manual_token: Optional[str]) -> Optional[str]:
         return manual_token.strip()
     return os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN")
 
-def detect_language(goal: str, code: str) -> str:
-    """Detects if the project is primarily Python or another language."""
-    combined = (goal + " " + code).lower()
+def detect_language(goal: str, code: str) -> bool:
+    """Detects whether the project is primarily Python.
+    Returns True for Python, False otherwise.
+    """
+    combined = (goal + " " + (code or "")).lower()
     python_kw = ["python", "django", "flask", "fastapi", "pytest", "def ", "import ", "pip"]
-    if any(kw in combined for kw in python_kw):
-        return "python"
-    return "other"
+    return any(kw in combined for kw in python_kw)
 
 def run_cmd(cmd: List[str], cwd: Optional[str] = None, timeout: int = COMMAND_TIMEOUT) -> Tuple[int, str]:
     """Runs a command in a subprocess with a timeout and captures output."""
     try:
         # Sandboxed execution: disable network for pytest
         env = os.environ.copy()
-        if "pytest" in cmd[0]:
+        # If command references pytest explicitly, disable network for safety
+        if any("pytest" in str(part) for part in cmd):
             env["ALLOW_NETWORK"] = "0"
 
         proc = subprocess.run(
@@ -575,7 +580,8 @@ def create_ui():
                             with gr.TabItem("tests/test_main.py"):
                                 test_file = gr.Code(language="python", lines=15)
                             with gr.TabItem("requirements.txt"):
-                                req_file = gr.Code(language="text", lines=10)
+                                # Changed language from "text" (unsupported) to "ini"
+                                req_file = gr.Code(language="ini", lines=10)
                             with gr.TabItem("README.md"):
                                 readme_file = gr.Code(language="markdown", lines=15)
                             with gr.TabItem("Other Files"):
@@ -640,4 +646,4 @@ if __name__ == "__main__":
         demo.queue().launch(server_name="0.0.0.0", server_port=7860)
     except Exception as e:
         print(f"Failed to launch Gradio app: {e}", file=sys.stderr)
-        sys.exit(1)
+        sys.exit(1)