Update definitions.py

definitions.py

@@ -1,4 +1,5 @@
 import platform
+import streamlit as st
 import psutil
 from typing import List, Dict, Optional, Any, Tuple
 from dataclasses import dataclass
@@ -531,30 +532,27 @@ if __name__ == "__main__":
             try:
                 # Generate tests
                 test_generation = await self.tool_manager.execute_tool(
-                    "test_generator",
-                    context.get("generated_code", "")
+                    "test_generator", context.get("generated_code", "")
                 )
 
                 # Run tests
                 test_results = await self.tool_manager.execute_tool(
-                    "test_runner",
-                    test_generation["result"]
+                    "test_runner", test_generation["result"]
                 )
 
                 # Generate coverage report
                 coverage_report = await self.tool_manager.execute_tool(
-                    "coverage_analyzer",
-                    test_results["result"]
+                    "coverage_analyzer", test_results["result"]
                 )
 
                 return {
                     "test_cases": test_generation["result"],
                     "test_results": test_results["result"],
+                    "coverage_report": coverage_report["result"]
                 }
             except Exception as e:
                 raise Exception(f"Testing stage failed: {str(e)}")
                 
-
         def _validate_stage_output(self, stage: PipelineStage, result: Dict[str, Any]):
             """Validate stage output against defined criteria"""
             validation_rules = self._get_validation_rules(stage)
@@ -1259,6 +1257,10 @@ if __name__ == "__main__":
             st.write(f"Available Memory: {psutil.virtual_memory().available / (1024**3):.1f} GB free")
 
 
+# Main entry point defined
+def main():
+    app = AutonomousAgentApp()
+    app.run()
+
 if __name__ == "__main__":
-    import streamlit as st
     main()