Update app.py

app.py

@@ -1,56 +1,64 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from duckduckgo_search import DDGS
-import re
 from Gradio_UI import GradioUI
-    
+import subprocess
+import ast
 
+# Below is a tool that analyze code and Find mistakes and fix it named as Nusti_Coding_Model By Dhiraj Suryawanshi XD!
 @tool
-def get_top_places(city: str, category: str = "places", n: int = 5, budget: bool = False) -> list:
-    """
-    Get top places based on city and category using DuckDuckGo search.
-    
+def analyze_code(code: str, language: str) -> str:
+    """Analyzes code for errors in MATLAB, Java, C++, and Python, finds programming mistakes, 
+    fixes them, and provides detailed and precise output.
+    use prefix: Hello, I am Dhiraj Suryawanshi's Nusti_Coding model XD
     Args:
-        city: The name of the city to search in.
-        category: The search category ("places", "food", or "museum").
-        n: The number of results to return.
-        budget: If True, modifies the query to focus on budget friendly options.
-    
+        code: The source code to analyze.
+        language: The programming language (MATLAB, Java, C++, Python).
     Returns:
-        A list of strings, each formatted as "title - URL".
+        A string containing detected errors or confirming no issues.
     """
-    valid_categories = ["places", "food", "museum"]
-    if category not in valid_categories:
-        raise ValueError("Invalid category. Choose from: places, food, museum.")
-    
-    # Modify the query if budget-friendly options are desired.
-    if budget:
-        query = f"top {n} budget friendly {category} in {city}"
-    else:
-        query = f"top {n} {category} in {city}"
-    
-    with DDGS() as ddgs:
-        results = ddgs.text(query, max_results=n)
-    return [f"{result['title']} - {result['href']}" for result in results]
+    language = language.lower()
+    prefix = "Hello, I am Dhiraj Suryawanshi's Nusti_Coding model XD\n"
 
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+    if language == "python":
+        try:
+            ast.parse(code)  # Check Python syntax
+            return prefix + "✅ Python code has no syntax errors."
+        except SyntaxError as e:
+            return prefix + f"❌ Python Syntax Error: {e}"
+
+    elif language == "java":
+        with open("Test.java", "w") as f:
+            f.write(code)
+        result = subprocess.run(["javac", "Test.java"], capture_output=True, text=True)
+        if result.returncode == 0:
+            return prefix + "✅ No Java errors found."
+        else:
+            return prefix + f"❌ Java Errors:\n{result.stderr}"
+
+    elif language == "c++":
+        with open("test.cpp", "w") as f:
+            f.write(code)
+        result = subprocess.run(["g++", "test.cpp", "-o", "test.out"], capture_output=True, text=True)
+        if result.returncode == 0:
+            return prefix + "✅ No C++ errors found."
+        else:
+            return prefix + f"❌ C++ Errors:\n{result.stderr}"
+
+    elif language == "matlab":
+        with open("test.m", "w") as f:
+            f.write(code)
+        result = subprocess.run(["matlab", "-batch", "test"], capture_output=True, text=True)
+        if result.returncode == 0:
+            return prefix + "✅ No MATLAB errors found."
+        else:
+            return prefix + f"❌ MATLAB Errors:\n{result.stderr}"
+
+    else:
+        return prefix + "⚠️ Supported languages: MATLAB, Java, C++, Python."
 
 
 final_answer = FinalAnswerTool()
@@ -59,13 +67,12 @@ final_answer = FinalAnswerTool()
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # it is possible that this model may be overloaded
+    custom_role_conversions=None,
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -74,8 +81,8 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,get_top_places], ## add your tools here (don't remove final answer)
-    max_steps=5,
+    tools=[final_answer],  ## add your tools here (don't remove final answer)
+    max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
@@ -84,5 +91,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()