Update graph.py

graph.py

@@ -24,6 +24,20 @@ from docx import Document
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
 from reportlab.lib.styles import getSampleStyleSheet
 
+# Add after other imports
+from multi_language_support import (
+    detect_language,
+    extract_code_blocks_multi_lang,
+    execute_code,
+    detect_requested_output_types_enhanced,
+    write_script_multi_lang,
+    LANGUAGES
+)
+
+# Replace existing functions
+detect_requested_output_types = detect_requested_output_types_enhanced
+write_script = write_script_multi_lang
+
 # --- Configuration ---
 OUT_DIR = os.environ.get("OUT_DIR", "/tmp")
 os.makedirs(OUT_DIR, exist_ok=True)
@@ -220,23 +234,23 @@ def parse_json_from_llm(llm_output: str) -> Optional[dict]:
 # --- Artifact detection ---
 KNOWN_ARTIFACT_TYPES = {"notebook","excel","word","pdf","image","repo","script"}
 
-def detect_requested_output_types(text: str) -> Dict:
-    if not text:
-        return {"requires_artifact": False, "artifact_type": None, "artifact_hint": None}
-    t = text.lower()
-    if any(k in t for k in ["jupyter notebook", "jupyter", "notebook", "ipynb"]):
-        return {"requires_artifact": True, "artifact_type": "notebook", "artifact_hint": "jupyter notebook"}
-    if any(k in t for k in ["excel", ".xlsx", "spreadsheet", "csv"]):
-        return {"requires_artifact": True, "artifact_type": "excel", "artifact_hint": "Excel file"}
-    if any(k in t for k in ["word document", ".docx", "docx"]):
-        return {"requires_artifact": True, "artifact_type": "word", "artifact_hint": "Word document"}
-    if any(k in t for k in ["pdf", "pdf file"]):
-        return {"requires_artifact": True, "artifact_type": "pdf", "artifact_hint": "PDF document"}
-    if any(k in t for k in ["repo", "repository", "app repo", "backend", "codebase"]):
-        return {"requires_artifact": True, "artifact_type": "repo", "artifact_hint": "application repository"}
-    if any(k in t for k in [".py", "python script", "script"]):
-        return {"requires_artifact": True, "artifact_type": "script", "artifact_hint": "Python script"}
-    return {"requires_artifact": False, "artifact_type": None, "artifact_hint": None}
+#def detect_requested_output_types(text: str) -> Dict:
+#    if not text:
+#        return {"requires_artifact": False, "artifact_type": None, "artifact_hint": None}
+#    t = text.lower()
+#    if any(k in t for k in ["jupyter notebook", "jupyter", "notebook", "ipynb"]):
+#        return {"requires_artifact": True, "artifact_type": "notebook", "artifact_hint": "jupyter notebook"}
+#    if any(k in t for k in ["excel", ".xlsx", "spreadsheet", "csv"]):
+#        return {"requires_artifact": True, "artifact_type": "excel", "artifact_hint": "Excel file"}
+#    if any(k in t for k in ["word document", ".docx", "docx"]):
+#        return {"requires_artifact": True, "artifact_type": "word", "artifact_hint": "Word document"}
+#    if any(k in t for k in ["pdf", "pdf file"]):
+#        return {"requires_artifact": True, "artifact_type": "pdf", "artifact_hint": "PDF document"}
+#    if any(k in t for k in ["repo", "repository", "app repo", "backend", "codebase"]):
+#        return {"requires_artifact": True, "artifact_type": "repo", "artifact_hint": "application repository"}
+#    if any(k in t for k in [".py", "python script", "script"]):
+#        return {"requires_artifact": True, "artifact_type": "script", "artifact_hint": "Python script"}
+#    return {"requires_artifact": False, "artifact_type": None, "artifact_hint": None}
 
 def normalize_experiment_type(exp_type: Optional[str], goal_text: str) -> str:
     if not exp_type:
@@ -284,25 +298,25 @@ def write_notebook_from_text(llm_text: str, out_dir: Optional[str]=None) -> str:
     nbformat.write(nb, filename)
     return filename
 
-def write_script(code_text: str, language_hint: Optional[str]=None, out_dir: Optional[str]=None) -> str:
-    out_dir = out_dir or OUT_DIR
-    os.makedirs(out_dir, exist_ok=True)
-    ext = ".txt"
-    if language_hint:
-        l = language_hint.lower()
-        if "python" in l:
-            ext = ".py"
-        elif "r" in l:
-            ext = ".R"
-        elif "java" in l:
-            ext = ".java"
-        elif "javascript" in l:
-            ext = ".js"
-    uid = uuid.uuid4().hex[:10]
-    filename = os.path.join(out_dir, f"generated_script_{uid}{ext}")
-    with open(filename, "w", encoding="utf-8") as f:
-        f.write(code_text)
-    return filename
+#def write_script(code_text: str, language_hint: Optional[str]=None, out_dir: Optional[str]=None) -> str:
+#    out_dir = out_dir or OUT_DIR
+#    os.makedirs(out_dir, exist_ok=True)
+#    ext = ".txt"
+#    if language_hint:
+#        l = language_hint.lower()
+#        if "python" in l:
+#            ext = ".py"
+#        elif "r" in l:
+#            ext = ".R"
+#        elif "java" in l:
+#            ext = ".java"
+#        elif "javascript" in l:
+#            ext = ".js"
+#    uid = uuid.uuid4().hex[:10]
+#    filename = os.path.join(out_dir, f"generated_script_{uid}{ext}")
+#    with open(filename, "w", encoding="utf-8") as f:
+#        f.write(code_text)
+#    return filename
 
 def write_docx_from_text(text: str, out_dir: Optional[str]=None) -> str:
     out_dir = out_dir or OUT_DIR