Update app.py

app.py

@@ -55,29 +55,26 @@ def run_sql(sql_text: str):
         cf, gap = pd.DataFrame(), pd.DataFrame()
     return df, "OK", cf, gap
 
-with gr.Blocks(title="ALM LLM") as demo:
-    gr.Markdown(INTRO)
+with gr.Tab("Ask in Natural Language"):
+    nl = gr.Textbox(label="Ask a question", placeholder="e.g., show me the top 10 fds by portfolio value")
+    btn = gr.Button("Run")
+    sql_out = gr.Textbox(label="Generated SQL", interactive=False)
+    why_out = gr.Textbox(label="Reasoning", interactive=False)
+    df_out  = gr.Dataframe(label="Query Result", interactive=True)
+    cf_out  = gr.Dataframe(label="Projected Cash-Flows (if applicable)", interactive=True)
+    gap_out = gr.Dataframe(label="Liquidity Gap (monthly)", interactive=True)
 
-    with gr.Tab("Ask in Natural Language"):
-        nl = gr.Textbox(label="Ask a question", placeholder="e.g., show me the top 10 fds by portfolio value")
-        btn = gr.Button("Run")
-        sql_out = gr.Textbox(label="Generated SQL", interactive=False)
-        why_out = gr.Textbox(label="Reasoning", interactive=False)
-        df_out  = gr.Dataframe(label="Query Result", wrap=True)
-        cf_out  = gr.Dataframe(label="Projected Cash-Flows (if applicable)", wrap=True, height=250)
-        gap_out = gr.Dataframe(label="Liquidity Gap (monthly)", wrap=True, height=200)
-
-        btn.click(fn=run_nl, inputs=[nl], outputs=[df_out, sql_out, why_out, cf_out, gap_out])
+    btn.click(fn=run_nl, inputs=[nl], outputs=[df_out, sql_out, why_out, cf_out, gap_out])
 
     with gr.Tab("Run Raw SQL"):
         sql_in = gr.Code(label="SQL", language="sql", value=f"SELECT * FROM {DEFAULT_SCHEMA}.{DEFAULT_TABLE} LIMIT 20;")
         btn2 = gr.Button("Execute")
-        df2  = gr.Dataframe(label="Result", wrap=True)
+        df2  = gr.Dataframe(label="Result", interactive=True)
         status = gr.Textbox(label="Status", interactive=False)
-        cf2   = gr.Dataframe(label="Projected Cash-Flows (if applicable)", wrap=True, height=250)
-        gap2  = gr.Dataframe(label="Liquidity Gap (monthly)", wrap=True, height=200)
+        cf2   = gr.Dataframe(label="Projected Cash-Flows (if applicable)", interactive=True)
+        gap2  = gr.Dataframe(label="Liquidity Gap (monthly)", interactive=True)
 
-        btn2.click(fn=run_sql, inputs=[sql_in], outputs=[df2, status, cf2, gap2])
+    btn2.click(fn=run_sql, inputs=[sql_in], outputs=[df2, status, cf2, gap2])
 
 if __name__ == "__main__":
     # Spaces set PORT automatically; otherwise, Gradio defaults are fine.