Update app.py

app.py

@@ -2,16 +2,16 @@ import gradio as gr
 import pandas as pd
 from transformers import pipeline
 
-# Load a public model without authentication
+# Load model (no key required)
 generator = pipeline("text2text-generation", model="google/flan-t5-base")
 
+# Store uploaded DataFrame
 uploaded_df = {}
 
 def read_xlsx(file):
     df = pd.read_excel(file.name)
     uploaded_df['df'] = df
-    preview = "📄 Preview of the first 5 rows:\n\n" + df.head(5).to_string(index=False)
-    return preview
+    return gr.Dataframe(value=df.head(10), wrap=True, label="📄 Table Preview")
 
 def generate_ai_summary(prompt):
     df = uploaded_df.get('df')
@@ -19,19 +19,25 @@ def generate_ai_summary(prompt):
         return "⚠️ No data uploaded."
     preview = df.head(5).to_string(index=False)
     full_prompt = prompt + "\n\n" + preview
-    result = generator(full_prompt, max_length=100)
-    return result[0]['generated_text']
+    result = generator(full_prompt, max_length=150)[0]['generated_text']
+    return result
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Smart XLSX Analyzer 📊\nUpload a spreadsheet and ask questions about it.")
+    gr.Markdown("# 📊 Smart XLSX Analyzer + AI Summary")
 
-    file_input = gr.File(file_types=[".xlsx"], label="Upload your XLSX file")
-    file_preview = gr.Textbox(label="Preview", interactive=False)
+    with gr.Row():
+        file_input = gr.File(file_types=[".xlsx"], label="Upload XLSX File")
+    
+    table_preview = gr.Dataframe(headers=[], interactive=False, label="Table Preview")
 
-    prompt = gr.Textbox(label="Ask something about the table", placeholder="Ex: Summarize the data", lines=2)
-    ai_output = gr.Textbox(label="AI Response")
+    with gr.Row():
+        prompt = gr.Textbox(label="🧠 Ask a question about the table", placeholder="E.g. Summarize this data", lines=2)
+        generate_btn = gr.Button("⚡ Generate Summary")
 
-    file_input.change(read_xlsx, inputs=file_input, outputs=file_preview)
-    prompt.submit(generate_ai_summary, inputs=prompt, outputs=ai_output)
+    ai_output = gr.Textbox(label="AI Response", lines=6)
+
+    # Wire components
+    file_input.change(read_xlsx, inputs=file_input, outputs=table_preview)
+    generate_btn.click(generate_ai_summary, inputs=prompt, outputs=ai_output)
 
 demo.launch()