Update definitions.py

definitions.py

@@ -32,6 +32,19 @@ def main():
     autonomous_agent_app = AutonomousAgentApp()
     app.run()
 
+class AutonomousAgentApp:
+    """Main application class for the Autonomous Agent System"""
+    
+    def __init__(self):
+        self.autonomous_agent = AutonomousAgent(self)
+        self.workspace_manager = self.autonomous_agent.workspace_manager
+        self.refinement_loop = self.autonomous_agent.refinement_loop
+        self.interface = self.autonomous_agent.interface
+
+    def run(self):
+        """Main entry point for the application"""
+        self.interface.render_main_interface()
+
 class CodeMetricsAnalyzer:
     """Analyzes code metrics using various tools"""
 
@@ -288,21 +301,6 @@ class QualityMetrics:
             "performance": 0.80
         }
 
-class ChatSystem:
-    """Manages the chat interaction between users and the autonomous system."""
-    def __init__(self, agent: 'AutonomousAgent'):
-        self.agent = agent
-        self.chat_history = []
-        self.active_tasks = {}
-        self.command_handlers = {
-            '/task': self.handle_task_command,
-            '/status': self.handle_status_command,
-            '/stop': self.handle_stop_command,
-            '/help': self.handle_help_command,
-            '/modify': self.handle_modify_command
-        }
-        self.logger = logging.getLogger(__name__)
-
 class AutonomousAgent:
     """Autonomous agent for the system."""
     def __init__(self, app):
@@ -312,7 +310,7 @@ class AutonomousAgent:
         self.refinement_loop = RefinementLoop(pipeline=self.pipeline)
         self.interface = self.StreamlitInterface(self)
         self.tools_repository = self._initialize_tool_repository()
-        self.chat_system = ChatSystem(self)
+        self.chat_system = ChatSystem(self) 
 
     def initialize_tool_repository(self, tool_repository: object) -> None:
         """
@@ -354,13 +352,14 @@ class AutonomousAgent:
     class StreamlitInterface:
         """Streamlit UI integration for the Autonomous Agent system."""
     
-    def __init__(self, app):
-        self.app = app
-        self.chat_system = self.app.chat_system
+        def __init__(self, app):
+            self.app = app
+            self.chat_system = self.app.chat_system
 
     
     class ChatSystem:
         """Manages the chat interaction between users and the autonomous system."""
+
         def __init__(self, agent):
             self.agent = agent
             self.chat_history = []
@@ -373,7 +372,7 @@ class AutonomousAgent:
                 '/modify': self.handle_modify_command
             }
             self.logger = logging.getLogger(__name__)
-            
+                
         def render_chat_interface(self):
             """Render the chat interface in Streamlit sidebar."""
             with st.sidebar:
@@ -401,12 +400,13 @@ class AutonomousAgent:
             self.app.refinement_loop.run_refinement_cycle(task)
             return "Task command handled"
 
-        def handle_status_command(self, input_data: Dict):
+        def handle_status_command(self, input_data: Dict):  # Fixed indentation
             """Handle status command."""
             self.logger.info("Handling status command")
-            status = input_data["status"]
-            self.app.refinement_loop.get_refinement_history()
-            return "Status command handled"
+            return {
+                "status": "success",
+                "history": self.agent.app.refinement_loop.get_refinement_history()
+            }
 
         def handle_stop_command(self, input_data: Dict):
             """Handle stop command."""
@@ -595,7 +595,7 @@ class RefinementLoop:
 
 
     def __init__(self, app):
-        """Autonomous agent for the system."""
+         """Autonomous agent for the system."""
         self.app = app
         self.workspace_manager = WorkspaceManager(workspace_dir=os.getenv('WORKSPACE_DIR', 'workspace'))
         self.pipeline = self._initialize_pipeline()
@@ -603,7 +603,7 @@ class RefinementLoop:
         self.interface = self.StreamlitInterface(self)
         self.tools_repository = self._initialize_tool_repository()
         self.chat_system = ChatSystem(self) 
-    
+
     class ChatSystem:
         """Manages the chat interaction between users and the autonomous system."""
         def __init__(self, agent):