Update app.py

app.py

@@ -111,7 +111,7 @@ class BasicModel:
                  content_text = content
             prompt.append(f"<|im_start|>{role}\n\n{content_text}<|im_end|>") # Qwen3 format
 
-        print(f"{role}: {content_text}") ## <- Print the last message in log
+        # print(f"{role}: {content_text}") ## <- Print the last message in log
         
         # Add the assistant prompt start
         prompt.append("<|im_start|>assistant\n\n")
@@ -160,6 +160,9 @@ class BasicModel:
                 json.loads(json_substring)
                 # If it's valid, we replace the original response with just the clean JSON part.
                 response = json_substring  
+
+                print("\n -------------JSON string is successfully extracted!-------\n----------------\n")
+                
         # If no valid JSON is found, we proceed with the original response string.
         except (json.JSONDecodeError, TypeError):
             # Pass silently and let the framework handle the potentially malformed string.
@@ -185,13 +188,11 @@ class BasicModel:
                 
                 # This is the core fix from the evaluation report's advice.
                 # We take the raw code string and wrap it in the required markdown format.
-                raw_code = agent_output_dict["code"]
+                raw_code = agent_output_dict["code"].strip()
                 
                 # We also clean up potential markdown fences the LLM might have added itself
-                if raw_code.strip().startswith("```"):
-                    raw_code = raw_code.strip()
-                    # raw_code = regex.sub(r"^(?:```(?:py|python)?\n)?(.*?)(?:\n```)?$", r"\1", raw_code, flags=regex.DOTALL)
-                    cleaned_code = regex.sub(r"^\s*```(?:py|python)?\n(.*?)\n```\s*$", r"\1", raw_code, flags=regex.DOTALL)
+                # raw_code = regex.sub(r"^(?:```(?:py|python)?\n)?(.*?)(?:\n```)?$", r"\1", raw_code, flags=regex.DOTALL)
+                cleaned_code = regex.sub(r"^\s*```(?:py|python)?\n(.*?)\n```\s*$", r"\1", raw_code, flags=regex.DOTALL)
     
                 # formatted_code = f"```py\n{raw_code}\n```"