Update app.py

app.py

@@ -4,11 +4,25 @@ import plotly.express as px
 import os
 
 def process_file(file):
-    # Load CSV and convert Unix timestamp to datetime
-    df = pd.read_csv(file.name)
-    df['date'] = pd.to_datetime(df['timestamp'], unit='s')  # convert Unix timestamp
+    # Determine file extension
+    file_ext = os.path.splitext(file.name)[1].lower()
 
-    # --- Plot 1: Messages over time ---
+    # Load file accordingly
+    if file_ext == '.csv':
+        df = pd.read_csv(file.name)
+    elif file_ext in ['.xls', '.xlsx']:
+        df = pd.read_excel(file.name)
+    else:
+        return "Unsupported file format", None, None
+
+    # Ensure 'timestamp' column exists
+    if 'timestamp' not in df.columns or 'author' not in df.columns:
+        return "The file must contain 'timestamp' and 'author' columns.", None, None
+
+    # Convert Unix timestamp to datetime
+    df['date'] = pd.to_datetime(df['timestamp'], unit='s')
+
+    # --- Plot 1: Messages per Day ---
     df['date_only'] = df['date'].dt.date
     messages_per_day = df.groupby("date_only").size().reset_index(name="count")
     fig1 = px.line(messages_per_day, x="date_only", y="count", title="Messages per Day")
@@ -19,21 +33,22 @@ def process_file(file):
     fig2 = px.bar(top_authors, x='author', y='count', title="Top 20 Authors", text='count')
     fig2.update_layout(xaxis_tickangle=-45)
 
-    # (Optional) delete the uploaded file
+    # Optional: delete file after processing
     os.remove(file.name)
 
-    return fig1, fig2
+    return "Success", fig1, fig2
 
 # Gradio interface
 interface = gr.Interface(
     fn=process_file,
-    inputs=gr.File(label="Upload your CSV file", file_types=[".csv"]),
+    inputs=gr.File(label="Upload CSV or XLSX", file_types=[".csv", ".xls", ".xlsx"]),
     outputs=[
-        gr.Plot(label="Messages by Day"),
-        gr.Plot(label="Messages by Author"),
+        gr.Textbox(label="Status"),
+        gr.Plot(label="Messages per Day"),
+        gr.Plot(label="Top Authors"),
     ],
     title="Message Analyzer",
-    description="Upload a CSV file with at least 'author' and 'timestamp' (Unix) columns."
+    description="Upload a CSV or XLSX file with 'timestamp' (Unix) and 'author' columns."
 )
 
 if __name__ == "__main__":