Update app.py

app.py

@@ -1,66 +1,34 @@
 import gradio as gr
 import pandas as pd
 import plotly.express as px
-import io
 
-# Global cache for dataframe
-df_cache = {}
+def load_table(file):
+    df = pd.read_excel(file.name) if file.name.endswith(".xlsx") else pd.read_csv(file.name)
+    table_md = df.to_markdown(index=False)
+    return table_md, df
 
-# Convert DataFrame to markdown
-def df_to_markdown(df):
-    return df.to_markdown(index=False)
-
-# File upload handler
-def handle_file(file):
-    ext = file.name.split('.')[-1]
-    if ext == "csv":
-        df = pd.read_csv(file.name)
-    elif ext in ["xlsx", "xls"]:
-        df = pd.read_excel(file.name)
-    else:
-        return "Unsupported file format", None
-
-    # Save dataframe globally
-    df_cache["df"] = df
-    markdown_table = df_to_markdown(df)
-    return markdown_table, f"File loaded: {file.name}"
-
-# Chatbot response (simple echo for now)
-def chat_response(message, history):
-    return history + [[message, "Terima kasih! Fitur analisis lanjut sedang dikembangkan."]]
-
-# Plot page: visualize table
-def show_plot():
-    df = df_cache.get("df")
-    if df is None:
-        return "Please upload a file on Page 1 first."
-
-    fig = px.bar(df, x=df.columns[0], y=df.columns[1], title="Visualisasi Kolom 1 vs Kolom 2")
+def visualize(df):
+    if df is None or df.empty:
+        return "No data to plot"
+    fig = px.bar(df, x=df.columns[0], y=df.columns[1])
     return fig
 
-# Interface: Page 1 - Upload + Table + Chat
-with gr.Blocks() as page1:
-    gr.Markdown("# Selamat Datang di Tertata 🎉")
-
-    file_input = gr.File(label="Upload CSV atau Excel", file_types=[".csv", ".xlsx", ".xls"])
-    markdown_output = gr.Markdown()
-    file_status = gr.Textbox(label="Status", interactive=False)
-
-    file_input.change(fn=handle_file, inputs=file_input, outputs=[markdown_output, file_status])
-
-    gr.Markdown("## Tanya Jawab 📩")
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Tanyakan sesuatu tentang tabel...", lines=1)
-    msg.submit(chat_response, inputs=[msg, chatbot], outputs=chatbot)
-
-# Interface: Page 2 - Visualization
-with gr.Blocks() as page2:
-    gr.Markdown("# 📊 Visualisasi Data")
-    gr.Markdown("Menampilkan grafik dari dua kolom pertama.")
-    plot_output = gr.Plot()
-    btn = gr.Button("Tampilkan Visualisasi")
-    btn.click(show_plot, outputs=plot_output)
-
-# Launch with tabs
-app = gr.TabbedInterface([page1, page2], ["Data & Tanya", "Visualisasi"])
-app.launch()
+with gr.Blocks() as demo:
+    gr.Markdown("## Selamat Datang di Tertata")
+
+    with gr.Tab("1. Upload & Chat"):
+        with gr.Row():
+            with gr.Column(scale=1, min_width=300):
+                file_input = gr.File(label="Upload file CSV/XLSX")
+                chatbox = gr.Textbox(label="Masukkan pertanyaan Anda", placeholder="Tanyakan sesuatu...", lines=1)
+            with gr.Column(scale=2, min_width=500, scrollable=True):
+                table_output = gr.Textbox(label="Tabel (Markdown)", lines=20, interactive=False, show_copy_button=True)
+        df_state = gr.State()
+        file_input.change(fn=load_table, inputs=file_input, outputs=[table_output, df_state])
+
+    with gr.Tab("2. Visualisasi"):
+        vis_output = gr.Plot()
+        vis_btn = gr.Button("Tampilkan Visualisasi")
+        vis_btn.click(fn=visualize, inputs=df_state, outputs=vis_output)
+
+demo.launch()