Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import pandas as pd
-import numpy as np
 from datetime import datetime, timedelta
 import networkx as nx
 import plotly.graph_objs as go
@@ -8,49 +7,42 @@ import plotly.graph_objs as go
 class TaskDependencyManager:
     def __init__(self):
         self.dependency_graph = nx.DiGraph()
-    
-    def add_task_dependency(self, tasks_df, task_name, predecessor=None, successor=None):
+
+    def add_task(self, task_data):
         """
-        Tambahkan dependency antar tugas dengan validasi komprehensif
+        Tambahkan tugas baru ke dalam graph dependency
         """
-        # Validasi input
-        if not task_name or task_name.strip() == '':
-            return False
+        task_name = task_data['Task Name']
 
         # Tambahkan node tugas ke graph
         if task_name not in self.dependency_graph.nodes():
             self.dependency_graph.add_node(task_name)
-        
+
         # Tambahkan dependency predecessor
-        if predecessor and predecessor != '':
+        predecessor = task_data['Predecessor']
+        if predecessor and predecessor != '-':
             if predecessor not in self.dependency_graph.nodes():
                 self.dependency_graph.add_node(predecessor)
-            
-            # Cegah dependency siklik
             if not self._is_cyclic_dependency(predecessor, task_name):
                 self.dependency_graph.add_edge(predecessor, task_name)
-        
+
         # Tambahkan dependency successor
-        if successor and successor != '':
+        successor = task_data['Successor']
+        if successor and successor != '-':
             if successor not in self.dependency_graph.nodes():
                 self.dependency_graph.add_node(successor)
-            
-            # Cegah dependency siklik
             if not self._is_cyclic_dependency(task_name, successor):
                 self.dependency_graph.add_edge(task_name, successor)
-        
-        return True
-    
+
     def _is_cyclic_dependency(self, source, target):
         """
         Cek apakah menambahkan edge akan menciptakan siklus dalam graph
         """
         try:
-            # Coba cek apakah path sudah ada dari target ke source
             return nx.has_path(self.dependency_graph, target, source)
         except nx.NetworkXError:
             return False
-    
+
     def get_task_dependencies(self, task_name):
         """
         Dapatkan predecessor dan successor suatu tugas
@@ -58,7 +50,6 @@ class TaskDependencyManager:
         try:
             predecessors = list(self.dependency_graph.predecessors(task_name))
             successors = list(self.dependency_graph.successors(task_name))
-            
             return {
                 'predecessors': ', '.join(predecessors) if predecessors else '-',
                 'successors': ', '.join(successors) if successors else '-'
@@ -68,49 +59,20 @@ class TaskDependencyManager:
                 'predecessors': '-',
                 'successors': '-'
             }
-    
-    def update_task_dependencies(self, tasks_df):
-        """
-        Update dependency untuk seluruh task di dataframe
-        """
-        # Reset graph
-        self.dependency_graph = nx.DiGraph()
-        
-        # Tambahkan dependency untuk setiap task
-        for _, task in tasks_df.iterrows():
-            task_name = task['Task Name']
-            predecessor = task['Predecessor']
-            successor = task['Successor']
-            
-            # Tambahkan node
-            if task_name not in self.dependency_graph.nodes():
-                self.dependency_graph.add_node(task_name)
-            
-            # Tambahkan predecessor
-            if predecessor and predecessor != '-':
-                if predecessor not in self.dependency_graph.nodes():
-                    self.dependency_graph.add_node(predecessor)
-                self.dependency_graph.add_edge(predecessor, task_name)
-            
-            # Tambahkan successor
-            if successor and successor != '-':
-                if successor not in self.dependency_graph.nodes():
-                    self.dependency_graph.add_node(successor)
-                self.dependency_graph.add_edge(task_name, successor)
-    
+
     def create_dependency_visualization(self):
         """
         Buat visualisasi graph dependency dengan posisi node yang optimal
         """
         if not self.dependency_graph.nodes():
             return None
-        
+
         # Layout untuk positioning node
         try:
             pos = nx.kamada_kawai_layout(self.dependency_graph)
         except:
             pos = nx.spring_layout(self.dependency_graph, k=0.5, seed=42)
-        
+
         # Trace untuk edges
         edge_x, edge_y = [], []
         for edge in self.dependency_graph.edges():
@@ -118,23 +80,23 @@ class TaskDependencyManager:
             x1, y1 = pos[edge[1]]
             edge_x.extend([x0, x1, None])
             edge_y.extend([y0, y1, None])
-        
+
         edge_trace = go.Scatter(
             x=edge_x, y=edge_y,
             line=dict(width=1.5, color='#888', dash='dot'),
             hoverinfo='none',
             mode='lines')
-        
+
         # Trace untuk nodes
         node_x, node_y, node_text = [], [], []
         node_degrees = dict(self.dependency_graph.degree())
-        
+
         for node in self.dependency_graph.nodes():
             x, y = pos[node]
             node_x.append(x)
             node_y.append(y)
             node_text.append(f"{node} (Koneksi: {node_degrees[node]})")
-        
+
         node_trace = go.Scatter(
             x=node_x, y=node_y,
             text=node_text,
@@ -154,62 +116,56 @@ class TaskDependencyManager:
                 line_width=2
             )
         )
-        
+
         layout = go.Layout(
             title='Graph Dependency Proyek',
             titlefont_size=16,
             showlegend=False,
             hovermode='closest',
-            margin=dict(b=20,l=5,r=5,t=40),
+            margin=dict(b=20, l=5, r=5, t=40),
             xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
             yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
         )
-        
+
         return go.Figure(data=[edge_trace, node_trace], layout=layout)
 
 class StreamlitAdvancedProjectApp:
     def __init__(self):
-        if 'dependency_manager' not in st.session_state:
-            st.session_state.dependency_manager = TaskDependencyManager()
-        self.dependency_manager = st.session_state.dependency_manager
-    
+        if 'tasks' not in st.session_state:
+            st.session_state.tasks = pd.DataFrame(columns=[
+                'Task Name', 'Start Date', 'End Date',
+                'Duration', 'Complexity', 'Predecessor', 'Successor',
+                'Progress', 'Cost'
+            ])
+        self.dependency_manager = TaskDependencyManager()
+
     def run(self):
         st.set_page_config(page_title="Project Dependency Optimizer", page_icon="🚀")
-        
         st.title("🚀 Advanced Project Dependency Optimizer")
-        
+
         # Sidebar untuk input tugas
         st.sidebar.header("Input Tugas Terperinci")
-        
-        # Inisialisasi session state untuk tasks
-        if 'tasks' not in st.session_state:
-            st.session_state.tasks = pd.DataFrame(columns=[
-                'Task Name', 'Start Date', 'End Date', 
-                'Duration', 'Complexity', 'Predecessor', 'Successor', 
-                'Progress', 'Cost'
-            ])
-        
-        # Form input tugas
+
         with st.sidebar.form("task_input"):
-            task_name = st.text_input("Nama Tugas", key="task_name_input")
-            start_date = st.date_input("Tanggal Mulai", key="start_date_input")
-            duration = st.number_input("Durasi (Hari)", min_value=1, value=5, key="duration_input")
-            complexity = st.selectbox("Kompleksitas", ['Low', 'Medium', 'High'], key="complexity_input")
-            
+            task_name = st.text_input("Nama Tugas")
+            start_date = st.date_input("Tanggal Mulai")
+            duration = st.number_input("Durasi (Hari)", min_value=1, value=5)
+            complexity = st.selectbox("Kompleksitas", ['Low', 'Medium', 'High'])
+
             # Pilih predecessor dan successor dari tugas yang sudah ada
             existing_tasks = [''] + list(st.session_state.tasks['Task Name'])
-            predecessor = st.selectbox("Predecessor (Opsional)", existing_tasks, index=0, key="predecessor_input")
-            successor = st.selectbox("Successor (Opsional)", existing_tasks, index=0, key="successor_input")
-            
+            predecessor = st.selectbox("Predecessor (Opsional)", existing_tasks, index=0)
+            successor = st.selectbox("Successor (Opsional)", existing_tasks, index=0)
+
             submit = st.form_submit_button("Tambah Tugas")
-            
+
             if submit and task_name:
                 # Cek duplikasi nama tugas
                 if task_name in st.session_state.tasks['Task Name'].values:
                     st.sidebar.error("Tugas dengan nama ini sudah ada!")
                 else:
                     # Siapkan data tugas baru
-                    new_task = pd.DataFrame([{
+                    new_task = {
                         'Task Name': task_name,
                         'Start Date': start_date,
                         'End Date': start_date + timedelta(days=duration),
@@ -219,47 +175,36 @@ class StreamlitAdvancedProjectApp:
                         'Successor': successor if successor else '-',
                         'Progress': 0,
                         'Cost': duration * 500
-                    }])
-                    
-                    # Tambahkan ke dataframe
-                    st.session_state.tasks = pd.concat([
-                        st.session_state.tasks, 
-                        new_task
-                    ], ignore_index=True)
-                    
-                    # Update dependency graph
-                    self.dependency_manager.update_task_dependencies(st.session_state.tasks)
-        
+                    }
+                    st.session_state.tasks = st.session_state.tasks.append(new_task, ignore_index=True)
+                    self.dependency_manager.add_task(new_task)
+
         # Tampilkan daftar tugas
         if not st.session_state.tasks.empty:
             st.subheader("Daftar Tugas")
-            tasks_display = st.session_state.tasks.copy()
-            st.dataframe(tasks_display)
-            
+            st.dataframe(st.session_state.tasks, use_container_width=True)
+
             # Analisis Dependency
             st.subheader("Analisis Hubungan Tugas")
             selected_task = st.selectbox(
-                "Pilih Tugas untuk Melihat Dependency", 
-                tasks_display['Task Name']
+                "Pilih Tugas untuk Melihat Dependency",
+                st.session_state.tasks['Task Name']
             )
-            
+
             if selected_task:
                 dependencies = self.dependency_manager.get_task_dependencies(selected_task)
-                
                 col1, col2 = st.columns(2)
                 with col1:
                     st.write("Predecessor:")
                     st.code(dependencies['predecessors'])
-                
                 with col2:
                     st.write("Successor:")
                     st.code(dependencies['successors'])
-            
+
             # Visualisasi Graph Dependency
             st.subheader("Visualisasi Graph Dependency")
             try:
                 dependency_graph_fig = self.dependency_manager.create_dependency_visualization()
-                
                 if dependency_graph_fig:
                     st.plotly_chart(dependency_graph_fig, use_container_width=True)
                 else: