Create app.py

app.py

@@ -1,26 +1,31 @@
 import gradio as gr
 import datetime
+import random
+from fpdf import FPDF
+from docx import Document
+import os
 
-# Function to generate a simple, unique timetable
-def generate_timetable(days, subjects, start_hour=8, slot_duration=1):
-    """
-    days: list of days of week
-    subjects: list of subjects
-    start_hour: starting hour in 24h format
-    slot_duration: duration of each slot in hours
-    """
-    import random
+# Function to generate a timetable
+def generate_timetable(days, subjects, teachers, start_hour=8, slot_duration=1):
     timetable = {}
     num_slots = len(subjects)
-    
     for day in days:
-        random.shuffle(subjects)  # shuffle subjects for uniqueness
         day_schedule = []
-        for i, subject in enumerate(subjects):
+        # shuffle subjects for uniqueness
+        subject_order = subjects.copy()
+        teacher_order = teachers.copy()
+        random.shuffle(subject_order)
+        random.shuffle(teacher_order)
+        for i, subject in enumerate(subject_order):
+            teacher = teacher_order[i % len(teachers)]
             slot_start = datetime.time(hour=start_hour + i*slot_duration)
             slot_end_hour = start_hour + (i+1)*slot_duration
             slot_end = datetime.time(hour=slot_end_hour)
-            day_schedule.append(f"{slot_start.strftime('%H:%M')} - {slot_end.strftime('%H:%M')}: {subject}")
+            day_schedule.append({
+                "time": f"{slot_start.strftime('%H:%M')} - {slot_end.strftime('%H:%M')}",
+                "subject": subject,
+                "teacher": teacher
+            })
         timetable[day] = day_schedule
     return timetable
 
@@ -30,31 +35,84 @@ def format_timetable(timetable_dict):
     for day, slots in timetable_dict.items():
         formatted += f"### {day}\n"
         for slot in slots:
-            formatted += f"- {slot}\n"
+            formatted += f"- {slot['time']}: {slot['subject']} (Teacher: {slot['teacher']})\n"
         formatted += "\n"
     return formatted
 
-# Gradio interface function
-def timetable_generator(days_input, subjects_input):
+# Export timetable to PDF
+def export_pdf(timetable_dict, filename="timetable.pdf"):
+    pdf = FPDF()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.add_page()
+    pdf.set_font("Arial", "B", 16)
+    pdf.cell(0, 10, "Timetable", ln=True, align="C")
+    pdf.ln(5)
+    pdf.set_font("Arial", "", 12)
+    for day, slots in timetable_dict.items():
+        pdf.set_font("Arial", "B", 14)
+        pdf.cell(0, 10, day, ln=True)
+        pdf.set_font("Arial", "", 12)
+        for slot in slots:
+            pdf.multi_cell(0, 8, f"{slot['time']}: {slot['subject']} (Teacher: {slot['teacher']})")
+        pdf.ln(2)
+    pdf.output(filename)
+    return filename
+
+# Export timetable to Word
+def export_word(timetable_dict, filename="timetable.docx"):
+    doc = Document()
+    doc.add_heading("Timetable", 0)
+    for day, slots in timetable_dict.items():
+        doc.add_heading(day, level=1)
+        for slot in slots:
+            doc.add_paragraph(f"{slot['time']}: {slot['subject']} (Teacher: {slot['teacher']})")
+    doc.save(filename)
+    return filename
+
+# Main function for Gradio
+def timetable_app(days_input, subjects_input, teachers_input, export_format):
     days = [d.strip().capitalize() for d in days_input.split(",") if d.strip()]
     subjects = [s.strip().capitalize() for s in subjects_input.split(",") if s.strip()]
-    if not days or not subjects:
-        return "Please enter valid days and subjects separated by commas."
-    tt = generate_timetable(days, subjects)
-    return format_timetable(tt)
+    teachers = [t.strip().capitalize() for t in teachers_input.split(",") if t.strip()]
+    
+    if not days or not subjects or not teachers:
+        return "Please enter valid days, subjects, and teacher names."
+    
+    timetable = generate_timetable(days, subjects, teachers)
+    formatted = format_timetable(timetable)
+    
+    # Export file if requested
+    file_path = None
+    if export_format == "PDF":
+        file_path = export_pdf(timetable, "timetable.pdf")
+    elif export_format == "Word":
+        file_path = export_word(timetable, "timetable.docx")
+    elif export_format == "Text":
+        with open("timetable.txt", "w") as f:
+            f.write(formatted)
+        file_path = "timetable.txt"
+    
+    return formatted, file_path
 
-# Gradio app
+# Gradio UI
 with gr.Blocks(title="Automatic Time Table Generation Agent (Improved)") as demo:
     gr.Markdown("# 🗓 Automatic Time Table Generation Agent (Improved)")
-    gr.Markdown("Enter the days of the week and subjects to generate a unique timetable.")
+    gr.Markdown("Generate unique timetables with teacher names and export as PDF, Word, or Text.")
 
-    days_input = gr.Textbox(label="Days (comma-separated, e.g., Monday, Tuesday, Wednesday)", placeholder="Monday, Tuesday, Wednesday, Thursday, Friday")
-    subjects_input = gr.Textbox(label="Subjects (comma-separated, e.g., Math, English, Science)", placeholder="Math, English, Science, History, Art")
+    days_input = gr.Textbox(label="Days (comma-separated)", placeholder="Monday, Tuesday, Wednesday, Thursday, Friday")
+    subjects_input = gr.Textbox(label="Subjects (comma-separated)", placeholder="Math, English, Science")
+    teachers_input = gr.Textbox(label="Teachers (comma-separated, match subjects count or fewer)", placeholder="Mr. Smith, Ms. Johnson, Mr. Lee")
+    export_format = gr.Dropdown(label="Export Format", choices=["None", "PDF", "Word", "Text"], value="None")
 
-    output = gr.Markdown()
+    output_text = gr.Markdown()
+    output_file = gr.File(label="Download File")
 
     generate_btn = gr.Button("Generate Timetable")
-    generate_btn.click(fn=timetable_generator, inputs=[days_input, subjects_input], outputs=output)
+    generate_btn.click(
+        timetable_app, 
+        inputs=[days_input, subjects_input, teachers_input, export_format], 
+        outputs=[output_text, output_file]
+    )
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", share=True)