Update app.py fourthly

app.py

@@ -31,16 +31,16 @@ class GenerateRequest(BaseModel):
     prompt: str                     # mô tả cần sinh code
     language: str | None = "Python"
     max_new_tokens: int = 128
-    num_beams: int = 4
-    temperature: float = 0.7
+    num_beams: int = 1              # ít beam hơn cho ổn định
+    temperature: float = 0.3        # giảm randomness
 
 
 class FixRequest(BaseModel):
     code: str                       # code bị lỗi
     language: str | None = "Python"
     max_new_tokens: int = 128
-    num_beams: int = 4
-    temperature: float = 0.3        # thấp để sửa lỗi ổn định hơn
+    num_beams: int = 1
+    temperature: float = 0.2        # thấp để sửa lỗi ổn định hơn
 
 
 class CompleteRequest(BaseModel):
@@ -79,6 +79,44 @@ def run_model(prompt: str,
     return text.strip()
 
 
+def clean_code(raw: str, lang: str) -> str:
+    """
+    Dọn mấy dòng rác đầu output (vd: ':', 'program:', ...) cho ra code “sạch” hơn.
+    Không đụng gì phần giữa & cuối.
+    """
+    lines = raw.splitlines()
+    if not lines:
+        return raw.strip()
+
+    lang_low = (lang or "").lower()
+
+    def looks_like_code(s: str) -> bool:
+        s = s.strip()
+        if not s:
+            return False
+
+        if lang_low == "python":
+            # thường bắt đầu bằng import/def/class/# comment
+            prefixes = ("def ", "class ", "import ", "from ", "#", "@")
+            return s.startswith(prefixes)
+        elif lang_low in ("c", "c++", "cpp"):
+            prefixes = ("#include", "int ", "void ", "char ", "float ",
+                        "double ", "struct ", "typedef ")
+            return s.startswith(prefixes)
+        else:
+            # fallback cho ngôn ngữ khác
+            return any(ch in s for ch in (";", "{", "}", "=", "function ", "public ", "private "))
+
+    start = 0
+    for i, line in enumerate(lines):
+        if looks_like_code(line):
+            start = i
+            break
+
+    cleaned = "\n".join(lines[start:]).strip()
+    return cleaned if cleaned else raw.strip()
+
+
 # ==== ENDPOINT 1: TẠO CODE TỪ PROMPT ====
 
 
@@ -91,12 +129,13 @@ You are a helpful coding assistant.
 
 Generate ONLY valid {lang} source code for the task below.
 Do NOT add any explanations, comments in natural language, or markdown.
-Return only raw {lang} code.
+Do NOT repeat the task description.
+Return only raw {lang} code that can be run.
 
 Task:
 {req.prompt}
 
-{lang} code:
+Begin {lang} code now:
 """.strip()
 
     output = run_model(
@@ -106,6 +145,8 @@ Task:
         temperature=req.temperature,
     )
 
+    output = clean_code(output, lang)
+
     return CodeResponse(output=output)
 
 
@@ -120,6 +161,7 @@ def fix_code(req: FixRequest):
 The following {lang} code contains bugs.
 Fix all bugs and return ONLY the corrected {lang} code.
 Do NOT add any explanations or comments in natural language.
+Do NOT change the language or rewrite the task.
 
 Buggy {lang} code:
 {req.code}
@@ -134,6 +176,8 @@ Corrected {lang} code:
         temperature=req.temperature,
     )
 
+    output = clean_code(output, lang)
+
     return CodeResponse(output=output)
 
 
@@ -175,7 +219,8 @@ Missing {lang} code:
         temperature=req.temperature,
     )
 
-    return CodeResponse(output=output)
+    # completion thường là snippet ngắn, không clean để tránh cắt nhầm
+    return CodeResponse(output=output.strip())
 
 
 # ==== HEALTHCHECK ====