Update app.py

app.py

@@ -1,30 +1,61 @@
 import gradio as gr
 import pandas as pd
+import matplotlib.pyplot as plt
+import io
 from transformers import pipeline
+import plotly.express as px
 
-# Load a small hosted model (no download needed)
+# Load small hosted model (no key or local model needed)
 generator = pipeline("text-generation", model="mrm8488/t5-base-finetuned-e2m")
 
-def analyze_xlsx(file, prompt):
-    # Read the first few rows from the XLSX
+# Global variable to store uploaded dataframe
+uploaded_df = {}
+
+def read_xlsx(file):
     df = pd.read_excel(file.name)
-    preview = df.head(5).to_string(index=False)
+    uploaded_df['df'] = df
+    return df.columns.tolist(), "Preview:\n" + df.head(5).to_string(index=False)
 
-    # Combine prompt and table data
+def generate_ai_summary(prompt):
+    df = uploaded_df.get('df')
+    if df is None:
+        return "No data loaded."
+    preview = df.head(5).to_string(index=False)
     full_prompt = prompt + "\n\n" + preview
-
-    # Generate AI response
     result = generator(full_prompt, max_new_tokens=100, do_sample=True, temperature=0.7)
     return result[0]['generated_text']
 
-# Gradio interface
-gr.Interface(
-    fn=analyze_xlsx,
-    inputs=[
-        gr.File(file_types=[".xlsx"], label="Upload XLSX file"),
-        gr.Textbox(lines=2, placeholder="Enter your prompt (e.g. Summarize this data)", label="Your Prompt")
-    ],
-    outputs=gr.Textbox(label="AI Response"),
-    title="📊 XLSX AI Analyzer with Transformers",
-    description="Upload an Excel file and ask questions or request a summary. No API key or local model required.",
-).launch()
+def plot_columns(x_col, y_col):
+    df = uploaded_df.get('df')
+    if df is None or x_col not in df.columns or y_col not in df.columns:
+        return None
+    fig = px.bar(df.head(20), x=x_col, y=y_col, title=f"{y_col} by {x_col}")
+    return fig
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# 📈 XLSX Analyzer + AI Summary + Plot")
+
+    with gr.Row():
+        file_input = gr.File(file_types=[".xlsx"], label="Upload XLSX File")
+        col_output = gr.Textbox(label="Columns", interactive=False)
+
+    with gr.Row():
+        x_dropdown = gr.Dropdown(label="X Axis", choices=[])
+        y_dropdown = gr.Dropdown(label="Y Axis", choices=[])
+
+    plot_output = gr.Plot(label="Bar Plot (Top 20)")
+    
+    with gr.Row():
+        prompt = gr.Textbox(label="Your Prompt", placeholder="Summarize this table or ask a question", lines=2)
+        ai_output = gr.Textbox(label="AI Summary")
+
+    file_input.change(read_xlsx, inputs=file_input, outputs=[x_dropdown, col_output])
+    file_input.change(read_xlsx, inputs=file_input, outputs=[y_dropdown, gr.Textbox(visible=False)])
+
+    x_dropdown.change(plot_columns, inputs=[x_dropdown, y_dropdown], outputs=plot_output)
+    y_dropdown.change(plot_columns, inputs=[x_dropdown, y_dropdown], outputs=plot_output)
+
+    prompt.submit(generate_ai_summary, inputs=prompt, outputs=ai_output)
+
+demo.launch()