Update app.py

app.py

@@ -1,102 +1,109 @@
 import gradio as gr
 import pandas as pd
-import os
+import sqlite3
 from datetime import datetime
 
-# Attendance Tracker Setup
-try:
-    if not os.path.exists("attendance_records"):
-        os.makedirs("attendance_records")
-except error:
-    print(error)
-     
-    
+# Database Setup
+db_file = "attendance_records.db"
 
-# Helper Functions
-def log_attendance(name, day, date, status):
-    month = datetime.strptime(date, "%Y-%m-%d").strftime("%Y-%m")
-    file_path = f"attendance_records/{month}.csv"
-
-    # Load or create attendance sheet for the month
-    if os.path.exists(file_path):
-        df = pd.read_csv(file_path)
-    else:
-        df = pd.DataFrame(columns=["Name", "Day", "Date", "Status"])
+conn = sqlite3.connect(db_file)
+cursor = conn.cursor()
 
-    # Add new attendance record
-    new_entry = {"Name": name, "Day": day, "Date": date, "Status": status}
-    df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)
+# Create attendance table if not exists
+cursor.execute("""
+CREATE TABLE IF NOT EXISTS attendance (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT,
+    day TEXT,
+    date TEXT,
+    status TEXT
+)
+""")
+conn.commit()
 
-    # Save back to the file
-    df.to_csv(file_path, index=False)
+# Helper Functions
+def log_attendance(name, day, date, status):
+    cursor.execute("""
+    INSERT INTO attendance (name, day, date, status)
+    VALUES (?, ?, ?, ?)
+    """, (name, day, date, status))
+    conn.commit()
     return "Attendance logged successfully!"
 
 def calculate_fees():
-    attendance_summaries = []
-
-    for file in os.listdir("attendance_records"):
-        if file.endswith(".csv"):
-            file_path = os.path.join("attendance_records", file)
-            df = pd.read_csv(file_path)
-
-            # Calculate fees for each candidate
-            fees_summary = df[df["Status"] == "Present"].groupby("Name").size() * (1000 / 12)  # Example: Monthly fees divided by 12
-            fees_summary.name = "Fees"
-
-            # Merge fees into the attendance sheet
-            df = df.merge(fees_summary, on="Name", how="left")
-
-            # Save updated file
-            df.to_csv(file_path, index=False)
-
-            # Summarize for all candidates
-            attendance_summaries.append(df[["Name", "Fees"]].drop_duplicates())
-
-    if attendance_summaries:
-        summary_df = pd.concat(attendance_summaries).drop_duplicates()
-        summary_df.to_csv("attendance_records/fees_summary.csv", index=False)
-        return "Fees calculated and updated successfully!"
-
-    return "No attendance records found for fees calculation."
+    # Calculate attendance fees
+    cursor.execute("""
+    SELECT name, COUNT(*) * (1000 / 12) AS fees
+    FROM attendance
+    WHERE status = 'Present'
+    GROUP BY name
+    """)
+    fees_data = cursor.fetchall()
+    fees_dict = {row[0]: row[1] for row in fees_data}
+    return fees_dict
+
+def create_end_of_month_table():
+    today = datetime.now()
+    if today.day != pd.Period(today.strftime("%Y-%m")).days_in_month:
+        return "It's not the end of the month yet."
+
+    # Create end-of-month table
+    month = today.strftime("%Y-%m")
+    table_name = f"fees_{month.replace('-', '_')}"
+    cursor.execute(f"""
+    CREATE TABLE IF NOT EXISTS {table_name} (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT,
+        email TEXT,
+        fees REAL
+    )
+    """)
+
+    # Load participant data
+    participant_file = "participants_form.xlsx"
+    participants = pd.read_excel(participant_file)
+
+    # Calculate fees
+    fees_dict = calculate_fees()
+
+    # Populate table
+    for _, row in participants.iterrows():
+        name = row["Name"]
+        email = row["Email"]
+        fees = fees_dict.get(name, 0)
+
+        cursor.execute(f"""
+        INSERT INTO {table_name} (name, email, fees)
+        VALUES (?, ?, ?)
+        """, (name, email, fees))
+
+    conn.commit()
+    return f"End-of-month table '{table_name}' created successfully!"
 
-# Gradio Interface
 def submit_attendance(name, day, date, status):
     return log_attendance(name, day, date, status)
 
-def generate_fees():
-    return calculate_fees()
-
 def is_month_end():
     today = datetime.now()
-    return (today.day == (pd.Period(today.strftime("%Y-%m")).days_in_month))
+    return today.day == pd.Period(today.strftime("%Y-%m")).days_in_month
 
-# Function to extract data from Excel
+
+# Gradio Interface
 def get_dropdown_options(file_path, column_name):
-    # Read the Excel file
     df = pd.read_excel(file_path)
-    # Extract the unique values from the specified column
     options = df["Name"].dropna().unique().tolist()
     options.sort()
     return options
 
 
-
 with gr.Blocks() as app:
     gr.Markdown("# Attendance Tracker")
 
     with gr.Row():
-        # name = gr.Textbox(label="Name")
         file_path = "participants_form.xlsx"
         column_name = "Name"
         options = get_dropdown_options(file_path, column_name)
-        name = gr.Dropdown(choices=options, label="Select an Option", show_label=True)
-        # output = gr.Textbox(label="You Selected")
-
-        # def show_selection(choice):
-        #     return f"You selected: {choice}"
-        
-        # name.change(show_selection, inputs=[name], outputs=[output])
-        
+        name = gr.Dropdown(choices=options, label="Select an Option")
         day = gr.Textbox(label="Day")
         date = gr.Textbox(label="Date (YYYY-MM-DD)")
         status = gr.Radio(["Present", "Absent"], label="Status")
@@ -104,15 +111,11 @@ with gr.Blocks() as app:
     submit_button = gr.Button("Submit Attendance")
     submit_message = gr.Textbox(label="Message", interactive=False)
 
-    calculate_button = gr.Button("Calculate Fees", interactive=is_month_end())
-    calculate_message = gr.Textbox(label="Fees Calculation Message", interactive=False)
-
     submit_button.click(submit_attendance, inputs=[name, day, date, status], outputs=[submit_message])
-    calculate_button.click(generate_fees, outputs=[calculate_message])
 
-    def update_calculate_button():
-        return gr.update(interactive=is_month_end())
+    def update_end_of_month():
+        return create_end_of_month_table()
 
-    app.load(update_calculate_button, None, [calculate_button])
+    app.load(update_end_of_month)
 
 app.launch()