Update app.py

app.py

@@ -1,5 +1,8 @@
 from smolagents import CodeAgent, HfApiModel, tool
 import yaml
+import math
+from tools.final_answer import FinalAnswerTool
+from Gradio_UI import GradioUI
 
 @tool
 def udp_advisor(development_phase: str, team_size: int, project_complexity: str) -> str:
@@ -34,72 +37,72 @@ def udp_advisor(development_phase: str, team_size: int, project_complexity: str)
     udp_tools = {
         "planning": {
             "low": [
-                {"name": "Trello", "category": "Project Management", "complexity": "Low", "description": "Simple kanban board for task tracking"},
-                {"name": "GitHub Projects", "category": "Project Management", "complexity": "Low", "description": "Basic project management integrated with GitHub"}
+                {"name": "Trello", "category": "Project Management", "complexity": "Low", "min_team_size": 1, "description": "Simple kanban board for task tracking"},
+                {"name": "GitHub Projects", "category": "Project Management", "complexity": "Low", "min_team_size": 1, "description": "Basic project management integrated with GitHub"}
             ],
             "medium": [
-                {"name": "Jira", "category": "Project Management", "complexity": "Medium", "description": "Comprehensive agile project management tool"},
-                {"name": "Azure DevOps Boards", "category": "Project Management", "complexity": "Medium", "description": "Project tracking with Azure integration"}
+                {"name": "Jira", "category": "Project Management", "complexity": "Medium", "min_team_size": 5, "description": "Comprehensive agile project management tool"},
+                {"name": "Azure DevOps Boards", "category": "Project Management", "complexity": "Medium", "min_team_size": 5, "description": "Project tracking with Azure integration"}
             ],
             "high": [
-                {"name": "Monday.com", "category": "Project Management", "complexity": "High", "description": "Customizable workflow platform for complex projects"},
-                {"name": "Rally", "category": "Project Management", "complexity": "High", "description": "Enterprise-scale agile management and portfolio planning"}
+                {"name": "Monday.com", "category": "Project Management", "complexity": "High", "min_team_size": 10, "description": "Customizable workflow platform for complex projects"},
+                {"name": "Rally", "category": "Project Management", "complexity": "High", "min_team_size": 20, "description": "Enterprise-scale agile management and portfolio planning"}
             ]
         },
         "development": {
             "low": [
-                {"name": "GitHub", "category": "Version Control", "complexity": "Low", "description": "Popular code hosting platform with basic CI/CD"},
-                {"name": "VS Code", "category": "IDE", "complexity": "Low", "description": "Lightweight, extensible code editor"}
+                {"name": "GitHub", "category": "Version Control", "complexity": "Low", "min_team_size": 1, "description": "Popular code hosting platform with basic CI/CD"},
+                {"name": "VS Code", "category": "IDE", "complexity": "Low", "min_team_size": 1, "description": "Lightweight, extensible code editor"}
             ],
             "medium": [
-                {"name": "GitLab", "category": "DevOps Platform", "complexity": "Medium", "description": "Integrated DevOps platform with built-in CI/CD"},
-                {"name": "JetBrains IDEs", "category": "IDE", "complexity": "Medium", "description": "Language-specific professional IDEs"}
+                {"name": "GitLab", "category": "DevOps Platform", "complexity": "Medium", "min_team_size": 5, "description": "Integrated DevOps platform with built-in CI/CD"},
+                {"name": "JetBrains IDEs", "category": "IDE", "complexity": "Medium", "min_team_size": 1, "description": "Language-specific professional IDEs"}
             ],
             "high": [
-                {"name": "Bitbucket + Bamboo", "category": "Enterprise DevOps", "complexity": "High", "description": "Atlassian's integrated code and CI/CD solution"},
-                {"name": "GitHub Enterprise", "category": "Enterprise DevOps", "complexity": "High", "description": "Self-hosted, compliant GitHub with advanced security"}
+                {"name": "Bitbucket + Bamboo", "category": "Enterprise DevOps", "complexity": "High", "min_team_size": 10, "description": "Atlassian's integrated code and CI/CD solution"},
+                {"name": "GitHub Enterprise", "category": "Enterprise DevOps", "complexity": "High", "min_team_size": 20, "description": "Self-hosted, compliant GitHub with advanced security"}
             ]
         },
         "testing": {
             "low": [
-                {"name": "Jest", "category": "Testing Framework", "complexity": "Low", "description": "JavaScript testing framework with minimal configuration"},
-                {"name": "GitHub Actions", "category": "CI/CD", "complexity": "Low", "description": "Simple workflow automation integrated with GitHub"}
+                {"name": "Jest", "category": "Testing Framework", "complexity": "Low", "min_team_size": 1, "description": "JavaScript testing framework with minimal configuration"},
+                {"name": "GitHub Actions", "category": "CI/CD", "complexity": "Low", "min_team_size": 1, "description": "Simple workflow automation integrated with GitHub"}
             ],
             "medium": [
-                {"name": "Jenkins", "category": "CI/CD", "complexity": "Medium", "description": "Flexible automation server for building, testing, and deployment"},
-                {"name": "Selenium", "category": "Testing", "complexity": "Medium", "description": "Browser automation for web application testing"}
+                {"name": "Jenkins", "category": "CI/CD", "complexity": "Medium", "min_team_size": 3, "description": "Flexible automation server for building, testing, and deployment"},
+                {"name": "Selenium", "category": "Testing", "complexity": "Medium", "min_team_size": 2, "description": "Browser automation for web application testing"}
             ],
             "high": [
-                {"name": "Azure DevOps Pipelines", "category": "CI/CD", "complexity": "High", "description": "Enterprise-grade pipelines with extensive integration"},
-                {"name": "SonarQube", "category": "Code Quality", "complexity": "High", "description": "Advanced code quality and security platform"}
+                {"name": "Azure DevOps Pipelines", "category": "CI/CD", "complexity": "High", "min_team_size": 5, "description": "Enterprise-grade pipelines with extensive integration"},
+                {"name": "SonarQube", "category": "Code Quality", "complexity": "High", "min_team_size": 5, "description": "Advanced code quality and security platform"}
             ]
         },
         "deployment": {
             "low": [
-                {"name": "Netlify", "category": "Hosting", "complexity": "Low", "description": "Simple deployment platform for frontend applications"},
-                {"name": "Heroku", "category": "PaaS", "complexity": "Low", "description": "Platform-as-a-Service with simplified deployment"}
+                {"name": "Netlify", "category": "Hosting", "complexity": "Low", "min_team_size": 1, "description": "Simple deployment platform for frontend applications"},
+                {"name": "Heroku", "category": "PaaS", "complexity": "Low", "min_team_size": 1, "description": "Platform-as-a-Service with simplified deployment"}
             ],
             "medium": [
-                {"name": "Docker", "category": "Containerization", "complexity": "Medium", "description": "Container platform for application packaging"},
-                {"name": "AWS Elastic Beanstalk", "category": "PaaS", "complexity": "Medium", "description": "Easy-to-use service for deploying applications"}
+                {"name": "Docker", "category": "Containerization", "complexity": "Medium", "min_team_size": 2, "description": "Container platform for application packaging"},
+                {"name": "AWS Elastic Beanstalk", "category": "PaaS", "complexity": "Medium", "min_team_size": 3, "description": "Easy-to-use service for deploying applications"}
             ],
             "high": [
-                {"name": "Kubernetes", "category": "Container Orchestration", "complexity": "High", "description": "Production-grade container orchestration"},
-                {"name": "Terraform", "category": "IaC", "complexity": "High", "description": "Infrastructure as Code for multi-cloud deployment"}
+                {"name": "Kubernetes", "category": "Container Orchestration", "complexity": "High", "min_team_size": 5, "description": "Production-grade container orchestration"},
+                {"name": "Terraform", "category": "IaC", "complexity": "High", "min_team_size": 3, "description": "Infrastructure as Code for multi-cloud deployment"}
             ]
         },
         "monitoring": {
             "low": [
-                {"name": "Uptime Robot", "category": "Monitoring", "complexity": "Low", "description": "Simple uptime monitoring for websites"},
-                {"name": "Sentry", "category": "Error Tracking", "complexity": "Low", "description": "Application error monitoring"}
+                {"name": "Uptime Robot", "category": "Monitoring", "complexity": "Low", "min_team_size": 1, "description": "Simple uptime monitoring for websites"},
+                {"name": "Sentry", "category": "Error Tracking", "complexity": "Low", "min_team_size": 1, "description": "Application error monitoring"}
             ],
             "medium": [
-                {"name": "Grafana", "category": "Observability", "complexity": "Medium", "description": "Metrics visualization and dashboards"},
-                {"name": "New Relic", "category": "APM", "complexity": "Medium", "description": "Application performance monitoring"}
+                {"name": "Grafana", "category": "Observability", "complexity": "Medium", "min_team_size": 2, "description": "Metrics visualization and dashboards"},
+                {"name": "New Relic", "category": "APM", "complexity": "Medium", "min_team_size": 3, "description": "Application performance monitoring"}
             ],
             "high": [
-                {"name": "Prometheus", "category": "Monitoring", "complexity": "High", "description": "Advanced metrics collection and alerting system"},
-                {"name": "Datadog", "category": "Observability", "complexity": "High", "description": "Full-stack observability platform"}
+                {"name": "Prometheus", "category": "Monitoring", "complexity": "High", "min_team_size": 3, "description": "Advanced metrics collection and alerting system"},
+                {"name": "Datadog", "category": "Observability", "complexity": "High", "min_team_size": 5, "description": "Full-stack observability platform"}
             ]
         }
     }
@@ -107,22 +110,32 @@ def udp_advisor(development_phase: str, team_size: int, project_complexity: str)
     # Team size categorization
     team_category = "small" if team_size < 10 else "medium" if team_size < 50 else "large"
     
-    # Get tool recommendations
+    # Find suitable tools based on team size and complexity
+    suitable_tools = []
     tools_for_phase = udp_tools[development_phase][project_complexity]
     
+    # Filter tools by team size
+    for tool in tools_for_phase:
+        if tool["min_team_size"] <= team_size:
+            suitable_tools.append(tool)
+    
     # Format response
     response = f"## UDP Tool Recommendations\n\n"
     response += f"**Development Phase:** {development_phase.capitalize()}\n"
     response += f"**Team Size:** {team_size} ({team_category} team)\n"
     response += f"**Project Complexity:** {project_complexity.capitalize()}\n\n"
     
-    response += "### Recommended Tools\n\n"
-    
-    for tool in tools_for_phase:
-        response += f"**{tool['name']}**\n"
-        response += f"- Category: {tool['category']}\n"
-        response += f"- Complexity: {tool['complexity']}\n"
-        response += f"- Description: {tool['description']}\n\n"
+    if suitable_tools:
+        response += "### Recommended Tools\n\n"
+        
+        for tool in suitable_tools:
+            response += f"**{tool['name']}**\n"
+            response += f"- Category: {tool['category']}\n"
+            response += f"- Complexity: {tool['complexity']}\n"
+            response += f"- Min Team Size: {tool['min_team_size']}\n"
+            response += f"- Description: {tool['description']}\n\n"
+    else:
+        response += "Unfortunately, we cannot provide specific tool recommendations for your team size and complexity requirements. Consider reviewing tools for a different complexity level or adjusting your team size.\n\n"
     
     # Additional phase-specific advice
     if development_phase == "planning":
@@ -151,25 +164,24 @@ def udp_advisor(development_phase: str, team_size: int, project_complexity: str)
         response += "- Establish incident response processes\n"
         response += "- Regularly review logs and performance data\n"
     
-    response += "\n### UDP Learning Resources\n"
-    response += "- [Unified Development Platforms Study Guide](https://ds-an.github.io/udp-study-guide/)\n"
-    response += "- [DevOps Roadmap](https://roadmap.sh/devops)\n"
-    response += "- [GitHub Learning Lab](https://lab.github.com/)\n"
+    response += "\n### Important Information\n"
+    response += "- These are general recommendations based on the information provided.\n"
+    response += "- Before adopting tools, assess your team's specific needs and skills.\n"
+    response += "- Consider integration capabilities with your existing toolchain.\n"
+    response += "- Training requirements should be factored into tool adoption decisions.\n"
     
     return response
 
-# Import final answer tool
-from tools.final_answer import FinalAnswerTool
+# Setup agent
 final_answer = FinalAnswerTool()
 
-# Configure the language model
 model = HfApiModel(
-    max_tokens=1024,
-    temperature=0.7,
-    model_id='Qwen/Qwen2.5-7B-Instruct'  # Using a smaller model for educational purposes
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
 )
 
-# Define prompt templates
+# Configure templates for the UDP advisor
 udp_prompt_templates = {
     "react-system": """You are a UDP (Unified Development Platform) advisor who helps users choose appropriate tools for their software development projects.
 You have access to the following tools:
@@ -177,9 +189,9 @@ You have access to the following tools:
 {tools_string}
 
 You should help the user determine which UDP tools are best for their project based on:
-1. The current development phase they are in
+1. The current development phase they are in (planning, development, testing, deployment, monitoring)
 2. The size of their development team
-3. The complexity of their project
+3. The complexity of their project (low, medium, high)
 
 Use the udp_advisor tool to analyze these factors and provide tool recommendations.
 
@@ -192,6 +204,7 @@ Always use Python syntax for calling tools. After using a tool, analyze the resu
 """,
     "react-human": "{input_string}",
     "react-assistant": "{agent_response}",
+    "final_answer": "Final Answer: {answer}"
 }
 
 # Load prompt templates from file or use our templates
@@ -212,59 +225,42 @@ agent = CodeAgent(
 )
 
 # Create a custom Gradio UI
-def create_udp_advisor_ui():
-    import gradio as gr
-    
-    with gr.Blocks(theme=gr.themes.Base()) as demo:
-        gr.Markdown("# UDP Advisor for Software Development Projects")
-        gr.Markdown("Get tool recommendations for your Unified Development Platform")
-        
-        with gr.Row():
-            with gr.Column():
-                phase_input = gr.Radio(
-                    ["Planning", "Development", "Testing", "Deployment", "Monitoring"], 
-                    label="Development Phase"
-                )
-                team_input = gr.Slider(
-                    minimum=1, 
-                    maximum=100, 
-                    step=1, 
-                    value=5,
-                    label="Team Size"
-                )
-                complexity_input = gr.Radio(
-                    ["Low", "Medium", "High"], 
-                    label="Project Complexity"
-                )
-                submit_btn = gr.Button("Get Recommendations", variant="primary")
+class UDPAdvisorUI(GradioUI):
+    def launch(self, **kwargs):
+        import gradio as gr
         
-        with gr.Row():
-            with gr.Column():
-                recommendation_output = gr.Markdown(label="UDP Recommendations")
-        
-        def get_recommendations(phase, team_size, complexity):
-            if not phase:
-                return "Please select a development phase."
-            if not team_size or team_size < 1:
-                return "Please provide a valid team size (at least 1)."
-            if not complexity:
-                return "Please select a project complexity level."
+        with gr.Blocks(theme=gr.themes.Soft()) as demo:
+            gr.Markdown("# UDP Advisor for Software Development")
+            gr.Markdown("Get personalized UDP tool recommendations based on your project profile")
+            
+            with gr.Row():
+                with gr.Column():
+                    phase_input = gr.Radio(["Planning", "Development", "Testing", "Deployment", "Monitoring"], label="Development Phase")
+                    team_input = gr.Number(label="Team Size", minimum=1)
+                    complexity_input = gr.Radio(["Low", "Medium", "High"], label="Project Complexity")
+                    submit_btn = gr.Button("Get Recommendations", variant="primary")
             
-            try:
-                recommendations = udp_advisor(phase.lower(), int(team_size), complexity.lower())
-                return recommendations
-            except Exception as e:
-                return f"Error generating recommendations: {str(e)}"
+            with gr.Row():
+                with gr.Column():
+                    recommendation_output = gr.Markdown(label="UDP Recommendations")
+            
+            def get_recommendations(phase, team_size, complexity):
+                if not phase:
+                    return "Please select a development phase."
+                if not team_size or team_size < 1:
+                    return "Please provide a valid team size (at least 1)."
+                if not complexity:
+                    return "Please select a project complexity level."
+                
+                try:
+                    recommendations = udp_advisor(phase.lower(), int(team_size), complexity.lower())
+                    return recommendations
+                except Exception as e:
+                    return f"Error generating recommendations: {str(e)}"
+            
+            submit_btn.click(get_recommendations, inputs=[phase_input, team_input, complexity_input], outputs=recommendation_output)
         
-        submit_btn.click(
-            get_recommendations, 
-            inputs=[phase_input, team_input, complexity_input], 
-            outputs=recommendation_output
-        )
-    
-    return demo
+        demo.launch(**kwargs)
 
-# Launch the application
-if __name__ == "__main__":
-    demo = create_udp_advisor_ui()
-    demo.launch()
+# Launch the interface
+UDPAdvisorUI(agent).launch()