Update app.py

app.py

@@ -1,5 +1,4 @@
 
-
 # app.py
 
 # AI Agent Framework Imports
@@ -83,15 +82,38 @@ def toggle_auto_execution(enable: bool) -> str:
 
 # --------------------------------------------
 # Configure the AI Model.
+# If the agent does not answer because the model is overloaded, you can switch to the alternative endpoint.
 model = HfApiModel(
     max_tokens=2096,                       # Maximum response length.
     temperature=0.5,                       # Controls response randomness.
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # Selected model.
+    # Use the following model_id if the primary model is overloaded:
+    model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+    # Alternatively, you can revert to the original model by using:
+    # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None
 )
 
 # Load prompt templates from a YAML configuration file.
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-# 
 
+# Create the AI agent with the defined tools.
+agent = CodeAgent(
+    model=model,
+    tools=[
+        FinalAnswerTool(),
+        detect_ambiguity,
+        explain_assumed_knowledge,
+        highlight_elements,
+        explain_code_line,
+        teacher_box_query,
+        toggle_auto_execution
+    ],
+    max_steps=6,             # Maximum number of reasoning steps.
+    verbosity_level=1,       # Level of detail in agent responses.
+    prompt_templates=prompt_templates
+)
+
+# Launch the interactive UI using Gradio.
+if __name__ == "__main__":
+    GradioUI(agent).launch(server_name="0.0.0.0", server_port=7860)