Update app.py

app.py

@@ -7,33 +7,33 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
+# Initialize the model
 model = OpenAIServerModel(
     model_id='qwen/qwen2.5-coder-32b-instruct',
     api_key=os.getenv("OPENAI_API_KEY", "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"),
     api_base=os.getenv("OPENAI_API_BASE", "https://integrate.api.nvidia.com/v1"),
 )
 
-web_search  = WebSearch()
+# Initialize tools
+web_search = WebSearch()
 final_answer = FinalAnswer()
 
+# Load YAML prompt templates
 with open(os.path.join(CURRENT_DIR, "prompts.yaml"), "r") as f:
     raw_templates = yaml.safe_load(f)
 
-# 🟨 Updated function: recursively unwrap {"type": "text", "text": "..."} or lists of them
+# ✅ FIXED: Function name corrected from unwrap_text_object to unwrap_text_objects
 def unwrap_text_objects(obj):
     if isinstance(obj, dict):
         if obj.get("type") == "text" and "text" in obj:
             return obj["text"]
         return {k: unwrap_text_objects(v) for k, v in obj.items()}
     elif isinstance(obj, list):
-        # If it's a list of strings, join it. If it's list of dicts, unwrap recursively.
         return "\n".join(unwrap_text_objects(i) for i in obj)
     return obj
 
-
-prompt_templates = unwrap_text_object(raw_templates)
-
-
+# Apply the unwrapping to raw templates
+prompt_templates = unwrap_text_objects(raw_templates)
 
 # Create the agent
 agent = CodeAgent(
@@ -49,6 +49,6 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch UI
+# Launch the Gradio UI
 if __name__ == "__main__":
-    GradioUI(agent).launch()
+    GradioUI(agent).launch()