main.py

#!/usr/bin/env python3
"""
AutoGen Framework - Main Entry Point

Proof of Concept demonstrating AutoGen-based business analysis framework.
Shows memory integration, agent workflows, and cross-requirement context handling.
This serves as the foundation for building specialized multi-agent workflows.
"""
import asyncio
from dotenv import load_dotenv
from ba_copilot import BASession

# Load environment variables from .env file
load_dotenv()


async def demo_ba_workflow():
    """Demonstrate BA agent workflow with memory integration and context handling"""
    print("AutoGen Framework - Business Analysis PoC")
    print("=" * 60)
    print("Demonstrating: BA Agent + Memory Integration + Context Handling")
    print()
    
    # Create and start BA session
    session = BASession()
    await session.start()
    
    try:
        # First requirement - establishes baseline context
        print("=== FIRST REQUIREMENT ===")
        requirement1 = """
        We need a user login system for our web application. Users should be able to register 
        with email and password, log in securely, and reset their password if they forget it.
        """
        print("Requirement 1:", requirement1.strip())
        print("\n" + "-" * 40)
        
        await session.analyze_requirement(requirement1, max_stories=2)
        
        print("\n" + "=" * 60)
        
        # Second requirement - tests cross-requirement memory and context
        print("=== SECOND REQUIREMENT (Tests context retention) ===")
        requirement2 = """
        Now we also need user profile management. Users should be able to update their 
        profile information, upload a profile picture, and manage their account settings.
        """
        print("Requirement 2:", requirement2.strip())
        print("\n" + "-" * 40)
        
        await session.analyze_requirement(requirement2, max_stories=2)
        
    finally:
        # Always clean up
        await session.close()
    
    print("\n" + "=" * 60)
    print("🎉 Framework PoC Complete!")
    print("✓ Memory integration working")
    print("✓ Agent maintained context across requirements")
    print("✓ Clean modular architecture")
    print("✓ Ready for extension with additional agent types")


async def main():
    """Main entry point for AutoGen framework demonstration"""
    await demo_ba_workflow()


if __name__ == "__main__":
    asyncio.run(main())