Optimizing Classroom Time: Using Lyzr-Automata to Generate Effective Timetables

lyzr-automata==0.1.2streamlit==1.33.0python-dotenv==1.0.1

import streamlit as stfrom lyzr_functions import generate_basic_timetable

# Time Slotstimeslot_input = st.text_area(    "TIMESLOTS",    '''08:00 AM - 09:00 AM09:15 AM - 10:15 AM10:30 AM - 11:30 AM11:45 AM - 12:45 PM01:00 PM - 02:00 PM02:15 PM - 03:15 PM03:30 PM - 04:30 PM04:45 PM - 05:45 PM''')# Subjectssubjects_input = st.text_area(    "SUBJECTS",    '''MathematicsScienceEnglishSocial StudiesHistoryGeographyPhysical EducationComputerArtMusic''')# Constraintsconstraints_input = st.text_area(    "CONSTRAINTS",    '''1. Mathematics - 5 classes a week2. Science - 5 classes a week3. English - 4 classes a week4. Social Studies - 4 classes a week5. History - 3 classes a week6. Geography - 3 classes a week7. Physical Education - 4 classes a week8. Computer - 5 classes a week9. Art - 2 class a week10. Music - 2 class a week11. Each classroom is allocated once per time slot on any given day.12. Same subjects cannot be assigned consecutively''')# Submit buttonsubmit_inputs = st.button("Submit", type="primary")

if submit_inputs:    # Save inputs to a file    with open("example_slot.txt", "w") as my_file:        my_file.write("TIMESLOTS")        my_file.write("
 
")        my_file.write(timeslot_input)        my_file.write("
 
")        my_file.write("SUBJECTS")        my_file.write("
 
")        my_file.write(subjects_input)        my_file.write("
 
")        my_file.write("CONSTRAINTS")        my_file.write("
 
")        my_file.write(constraints_input)    # Call timetable generator    generate_basic_timetable()    # Read and display from result file    with open("example_result.txt", "r") as my_file:        content = my_file.read()    st.write(content)

from lyzr_automata.ai_models.openai import OpenAIModelfrom lyzr_automata.memory.open_ai import OpenAIMemoryfrom lyzr_automata import Agent, Taskfrom lyzr_automata.tasks.task_literals import InputType, OutputTypefrom dotenv import load_dotenvimport osload_dotenv()OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

# OpenAI Text Modelopen_ai_model_text = OpenAIModel(api_key=OPENAI_API_KEY,parameters={    "model": "gpt-4-turbo-preview",    "temperature": 0.1,    "max_tokens": 4096,    },)# OpenAI Memory filetimetable_memory = OpenAIMemory(    file_path='example_slot.txt')

def generate_basic_timetable():    # Remove file if exists    if os.path.exists("assistant_ids.json"):        os.remove("assistant_ids.json")    # Timetable generator Agent    timetable_agent = Agent(        prompt_persona="You are an intelligent agent that can create efficient class timetables for a week in a simple, structured format. Do not assign more classes than required, assign free slots instead. Generate timetable for every day from Monday to Friday.",        role="Timetable creator",        memory=timetable_memory    )    # Timetable generator Task    timetable_task = Task(        name="Timetable Creator",        agent=timetable_agent,        output_type=OutputType.TEXT,        input_type=InputType.TEXT,        model=open_ai_model_text,        instructions="Using the time slots, subject details and requirements, create a timetable that satisfies every constraint. Return the timetable in a simple format and a count of number of classes scheduled for each subject.",        log_output=True,        enhance_prompt=False,    ).execute()    # Save output to a file    with open("example_result.txt", "w") as my_file:        my_file.write("# GENERATED TIMETABLE - Timetable Agent")        my_file.write("
 
")        my_file.write(timetable_task)        my_file.write("
")    # Timetable verification Agent    timetable_checker_agent = Agent(        prompt_persona="You are an intelligent agent that can verify if a generated timetable fulfills all the constraints or not. Make sure all classes meet the exact requirements; not more, not less.",        role="Timetable checker",        memory=timetable_memory    )    # Timetable verification Task    timetable_checker_task = Task(        name="Timetable Checker",        agent=timetable_checker_agent,        output_type=OutputType.TEXT,        input_type=InputType.TEXT,        model=open_ai_model_text,        instructions="Verify that the generated timetable input fulfills the requirements mentioned in the file. If VALID, return <!VALID!> and the timetable in a table format; If INVALID, return <!INVALID!> the reason why invalid. Do not return anything else.",        log_output=True,        enhance_prompt=False,        previous_output=timetable_task    ).execute()    # Save output to a file    with open("example_result.txt", "a") as my_file:        my_file.write("
 
")        my_file.write("# GENERATED VERIFICATION - Verification Agent")        my_file.write("
 
")        my_file.write(timetable_checker_task)        my_file.write("
")