main.py

import subprocess
import sys
import os

def main():
    tasks = {
        "1": ("Chat (Enter exit to stop)", "task_1/chat.py"),
        "2": ("RAG QA (Enter exit to stop)", "task_2/rag_qa.py"),
        "3": ("Planning Agent (Enter exit to stop)", "task_3/agent.py"),
        "4": ("Coding Assistant (Enter exit to stop)", "task_4/coding_assistant.py"),
        "5": ("RAGAS Eval", "task_2/ragas_eval.py"),
        "6": ("Load Document", "task_2/load_document.py"),
    }

    print("Select a task:")
    for key, (name, filename) in tasks.items():
        print(f"  {key}. {name}")

    choice = input("\nEnter number: ").strip()

    if choice in tasks:
        file, filename = tasks[choice]
        env = os.environ.copy()
        env["PYTHONPATH"] = os.path.dirname(os.path.abspath(__file__))
        subprocess.run([sys.executable, filename], env=env)

    else:
        print("Invalid choice.")

if __name__ == "__main__":
    main()