Update app.py

app.py

@@ -14,23 +14,22 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Global variables
-CURRENT_UTC_TIME = datetime.strptime("2025-02-04 19:22:08", "%Y-%m-%d %H:%M:%S")
+CURRENT_UTC_TIME = datetime.strptime("2025-02-04 19:27:44", "%Y-%m-%d %H:%M:%S")
 CURRENT_USER = "ChadSaglam"
 
 @st.cache_resource
 def load_ai_model():
-    """Initialize the AI model for task generation"""
     try:
         model_name = "EleutherAI/gpt-neo-125M"
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForCausalLM.from_pretrained(model_name)
         
-        # Create pipeline
         pipe = pipeline(
             "text-generation",
             model=model,
             tokenizer=tokenizer,
             max_new_tokens=512,
+            do_sample=True,
             temperature=0.7,
             top_p=0.95,
             repetition_penalty=1.15
@@ -59,77 +58,122 @@ class AIProjectManager:
             "Retrospective"
         ]
 
+    def _generate_structured_tasks(self, project_name: str, project_type: str) -> List[Dict]:
+        """Fallback method for generating structured tasks"""
+        base_tasks = {
+            "E-commerce Website": [
+                {
+                    "title": "User Authentication System",
+                    "priority": "P0",
+                    "assigned_to": "Backend Team",
+                    "estimated_hours": 40,
+                    "description": "Implement secure authentication system",
+                    "acceptance_criteria": ["Secure login", "Password recovery", "OAuth integration"],
+                    "subtasks": [
+                        {"title": "Setup Auth API", "assigned_to": "Backend Dev", "estimated_hours": 16},
+                        {"title": "Implement OAuth", "assigned_to": "Backend Dev", "estimated_hours": 24}
+                    ]
+                },
+                {
+                    "title": "Product Catalog",
+                    "priority": "P0",
+                    "assigned_to": "Frontend Team",
+                    "estimated_hours": 60,
+                    "description": "Create product listing and details pages",
+                    "acceptance_criteria": ["Product grid", "Search functionality", "Filtering"],
+                    "subtasks": [
+                        {"title": "Product Grid UI", "assigned_to": "Frontend Dev", "estimated_hours": 24},
+                        {"title": "Search Implementation", "assigned_to": "Frontend Dev", "estimated_hours": 36}
+                    ]
+                }
+            ],
+            "Mobile Application": [
+                {
+                    "title": "App Architecture Setup",
+                    "priority": "P0",
+                    "assigned_to": "Mobile Team",
+                    "estimated_hours": 40,
+                    "description": "Setup initial app architecture",
+                    "acceptance_criteria": ["Project structure", "Navigation flow", "State management"],
+                    "subtasks": [
+                        {"title": "Project Setup", "assigned_to": "Mobile Dev", "estimated_hours": 16},
+                        {"title": "Navigation Setup", "assigned_to": "Mobile Dev", "estimated_hours": 24}
+                    ]
+                }
+            ]
+        }
+        
+        tasks = base_tasks.get(project_type, [])
+        return [self._format_single_task(task) for task in tasks]
+
+    def _format_single_task(self, task: Dict) -> Dict:
+        """Format a single task with required fields"""
+        return {
+            "id": str(uuid.uuid4()),
+            "title": task.get("title", "Untitled Task"),
+            "priority": task.get("priority", "P2"),
+            "status": "TO_DO",
+            "assigned_to": task.get("assigned_to", "Unassigned"),
+            "estimated_hours": task.get("estimated_hours", 0),
+            "description": task.get("description", ""),
+            "acceptance_criteria": task.get("acceptance_criteria", []),
+            "subtasks": [
+                {
+                    "id": str(uuid.uuid4()),
+                    "title": subtask.get("title", "Untitled Subtask"),
+                    "status": "TO_DO",
+                    "assigned_to": subtask.get("assigned_to", "Unassigned"),
+                    "estimated_hours": subtask.get("estimated_hours", 0)
+                }
+                for subtask in task.get("subtasks", [])
+            ]
+        }
+
     def generate_project_tasks(self, project_name: str, project_type: str) -> List[Dict]:
-        """AI-driven task generation based on project type and name"""
         try:
-            # Create prompt for the AI model
             prompt = f"""
-            Generate a detailed project plan for {project_name}, which is a {project_type}.
-            Include essential tasks with the following structure:
-            - Task title
-            - Priority (P0, P1, P2, P3)
-            - Description
-            - Estimated hours
-            - Assigned team
-            - Subtasks with details
-            - Acceptance criteria
-            
-            Focus on modern best practices and industry standards.
-            Return the response in JSON format.
+            Create a detailed project plan for {project_name} ({project_type}) with the following JSON structure:
+            {{
+                "tasks": [
+                    {{
+                        "title": "Task name",
+                        "priority": "P0/P1/P2/P3",
+                        "assigned_to": "Team/Role",
+                        "estimated_hours": number,
+                        "description": "Detailed description",
+                        "acceptance_criteria": ["criteria1", "criteria2"],
+                        "subtasks": [
+                            {{
+                                "title": "Subtask name",
+                                "assigned_to": "Team/Role",
+                                "estimated_hours": number
+                            }}
+                        ]
+                    }}
+                ]
+            }}
             """
             
             # Generate response using the AI model
             response = self.llm(prompt)
             
-            # Parse the response and format tasks
             try:
+                # Try to parse the AI response as JSON
                 tasks_data = json.loads(response)
-                return self._format_tasks(tasks_data)
+                return [self._format_single_task(task) for task in tasks_data.get("tasks", [])]
             except json.JSONDecodeError:
-                # Fallback to structured task generation if JSON parsing fails
+                # Fallback to structured tasks if JSON parsing fails
+                logger.warning("Failed to parse AI response, using structured tasks")
                 return self._generate_structured_tasks(project_name, project_type)
                 
         except Exception as e:
             logger.error(f"Error generating tasks: {str(e)}")
             return self._generate_structured_tasks(project_name, project_type)
 
-    def _format_tasks(self, tasks_data: Dict) -> List[Dict]:
-        """Format AI-generated tasks into standardized structure"""
-        formatted_tasks = []
-        
-        for task in tasks_data.get("tasks", []):
-            formatted_task = {
-                "id": str(uuid.uuid4()),
-                "title": task.get("title", "Untitled Task"),
-                "priority": task.get("priority", "P2"),
-                "status": "TO_DO",
-                "assigned_to": task.get("assigned_to", "Unassigned"),
-                "estimated_hours": task.get("estimated_hours", 0),
-                "description": task.get("description", ""),
-                "acceptance_criteria": task.get("acceptance_criteria", []),
-                "subtasks": []
-            }
-            
-            # Format subtasks
-            for subtask in task.get("subtasks", []):
-                formatted_subtask = {
-                    "id": str(uuid.uuid4()),
-                    "title": subtask.get("title", "Untitled Subtask"),
-                    "status": "TO_DO",
-                    "assigned_to": subtask.get("assigned_to", "Unassigned"),
-                    "estimated_hours": subtask.get("estimated_hours", 0)
-                }
-                formatted_task["subtasks"].append(formatted_subtask)
-            
-            formatted_tasks.append(formatted_task)
-        
-        return formatted_tasks
-
 def render_task_ui():
     st.title("AI-Powered Project Management")
     
-    # Project Setup
-    project_name = st.text_input("Project Name")
+    project_name = st.text_input("Project Name", placeholder="Enter your project name")
     project_type = st.selectbox("Project Type", [
         "E-commerce Website",
         "Mobile Application",
@@ -140,17 +184,15 @@ def render_task_ui():
     ])
     
     if st.button("Generate Tasks") and project_name:
-        ai_manager = AIProjectManager()
-        tasks = ai_manager.generate_project_tasks(project_name, project_type)
-        
-        # Store tasks in session state
-        st.session_state.tasks = tasks
+        with st.spinner("Generating tasks with AI..."):
+            ai_manager = AIProjectManager()
+            tasks = ai_manager.generate_project_tasks(project_name, project_type)
+            st.session_state.tasks = tasks
         
-        # Display tasks
         display_tasks(tasks)
 
 def display_tasks(tasks: List[Dict]):
-    st.subheader("Generated Tasks")
+    st.subheader("Project Tasks")
     
     for task in tasks:
         with st.expander(f"📋 {task['title']} ({task['priority']})"):
@@ -185,11 +227,9 @@ def display_tasks(tasks: List[Dict]):
 def main():
     st.set_page_config(page_title="AI Project Management", layout="wide")
     
-    # Display current user and time
     st.sidebar.info(f"Current User: {CURRENT_USER}")
     st.sidebar.info(f"Current Time (UTC): {CURRENT_UTC_TIME.strftime('%Y-%m-%d %H:%M:%S')}")
     
-    # Initialize session state
     if "tasks" not in st.session_state:
         st.session_state.tasks = []