course schedule

app.py

@@ -4,6 +4,9 @@ import pandas as pd
 # from groq import Groq
 from dotenv import load_dotenv
 
+from openpyxl import load_workbook
+from openpyxl.styles import Alignment
+
 os.environ['REQUESTS_CA_BUNDLE'] = '/usr/local/share/zscaler.crt'
 
 
@@ -16,6 +19,7 @@ def read_excel_and_process(fi):
     new_data = {}
     row_names = ['8:30-9:50', '10-11:20', '11:30-12:50', '1-2:20', '2:30-3:50', '4-5:20', '6-7:20', '7:30-8:50']
     col_names = ['Time', 'Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday/Sunday']
+    column_widths = [15, 40, 40, 40, 40, 40, 40]
 
     for t in row_names:
         new_data[t] = {}
@@ -54,8 +58,24 @@ def read_excel_and_process(fi):
         data_2.append(v)
 
     df = pd.DataFrame.from_dict(data_2)
-    df.to_excel("visualization.xlsx", index=False)
-    return "visualization.xlsx"
+    # df.to_excel("visualization.xlsx", index=False)
+    # return "visualization.xlsx"
+
+
+    with pd.ExcelWriter('visualization.xlsx', engine='openpyxl') as writer:
+        df.to_excel(writer, sheet_name='Sheet1', index=False)
+
+        workbook = writer.book
+        worksheet = writer.sheets['Sheet1']
+
+        for i, width in enumerate(column_widths):
+            column_letter = chr(65 + i)  # change index col to（A, B, C,...）
+            worksheet.column_dimensions[column_letter].width = width
+
+            for cell in worksheet[column_letter]:
+                cell.alignment = Alignment(wrap_text=True)
+
+
 
 
 # def excel_to_csv(fi):