Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import pandas as pd
 from fuzzywuzzy import process, fuzz
 import tempfile
+import os
 
 def auto_correct_names(series, threshold=90):
     """Auto-correct typos in chatter names using fuzzy matching."""
@@ -23,23 +24,30 @@ def auto_correct_names(series, threshold=90):
 def process_file(input_file):
     """Process uploaded Excel file and return output"""
     try:
+        # Read input file
         input_df = pd.read_excel(input_file, header=1)
+        
+        # Store original date order
         date_columns = input_df.columns[1:].tolist()
         
+        # Melt to long format
         df_long = input_df.melt(
             id_vars=[input_df.columns[0]], 
             var_name='DATE', 
             value_name='CHATTER'
         )
         
+        # Force date order
         df_long['DATE'] = pd.Categorical(
             df_long['DATE'], 
             categories=date_columns,
             ordered=True
         )
         
+        # Clean names
         df_long['CHATTER'] = auto_correct_names(df_long['CHATTER'])
         
+        # Group and pivot
         grouped = df_long.groupby(['CHATTER', 'DATE'], observed=True)[input_df.columns[0]] \
             .apply(lambda x: ', '.join(sorted(x))).reset_index()
         
@@ -47,14 +55,19 @@ def process_file(input_file):
         chatter_order = grouped['CHATTER'].value_counts().index.tolist()
         final_df = pivoted.reindex(chatter_order)[date_columns].fillna("OFF")
         
+        # Reset index to show chatter names in preview
+        final_df = final_df.reset_index()
+        
+        # Create temp file for output
         with tempfile.NamedTemporaryFile(suffix=".xlsx", delete=False) as tmp:
-            final_df.to_excel(tmp.name, sheet_name='Schedule')
+            final_df.to_excel(tmp.name, index=False, sheet_name='Schedule')
             return final_df, tmp.name
 
     except Exception as e:
         error_df = pd.DataFrame({"Error": [f"⚠️ {str(e)}"]})
         return error_df, None
 
+# Gradio interface
 with gr.Blocks(title="Schedule Processor") as demo:
     gr.Markdown("# 📅 Schedule Processor")
     gr.Markdown("Upload your schedule Excel file and download the formatted version")
@@ -63,15 +76,15 @@ with gr.Blocks(title="Schedule Processor") as demo:
         input_file = gr.File(label="Upload Schedule File", type="filepath")
         output_file = gr.File(label="Download Processed File", visible=False)
     
-    output_table = gr.Dataframe(label="Preview")
+    output_table = gr.Dataframe(label="Preview", wrap=True)
     
     btn = gr.Button("Process File")
     
     def process_and_show(file):
         df, out_path = process_file(file)
         if out_path:
-            return df, out_path  # Fixed: Directly return the file path
-        return df, None
+            return df, gr.File.update(value=out_path, visible=True)
+        return df, gr.File.update(visible=False)
     
     btn.click(
         process_and_show,
@@ -80,4 +93,4 @@ with gr.Blocks(title="Schedule Processor") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)  # Added share=True for public link
+    demo.launch(share=True)