Update app.py

app.py

@@ -7,18 +7,17 @@ from tools.sql_tool import SQLTool
 from tools.ts_preprocess import build_timeseries
 
 DUCKDB_PATH = os.getenv("DUCKDB_PATH", "alm.duckdb")
-DEFAULT_PATH = "my_db.main.masterdataset_v"
 
 sql_tool = SQLTool(DUCKDB_PATH)
+RESOLVED_PATH = sql_tool.get_full_table_path()  # e.g., my_db.main.masterdataset_v or main.masterdataset_v
 
 INTRO = f"""
 ### ALM LLM — Demo
 
 Connected to **DuckDB** at `{DUCKDB_PATH}`  
-Using table **{DEFAULT_PATH}**.
+Using table **{RESOLVED_PATH}** (auto-resolved).
 """
 
-# ---------------------- Handlers ----------------------
 def run_nl(nl_query: str):
     if not nl_query or not nl_query.strip():
         return pd.DataFrame(), "", "Please enter a query.", pd.DataFrame(), pd.DataFrame()
@@ -32,7 +31,6 @@ def run_nl(nl_query: str):
         cf, gap = pd.DataFrame(), pd.DataFrame()
     return df, sql.strip(), why, cf, gap
 
-
 def run_sql(sql_text: str):
     if not sql_text or not sql_text.strip():
         return pd.DataFrame(), "Please paste a SQL statement.", pd.DataFrame(), pd.DataFrame()
@@ -46,7 +44,6 @@ def run_sql(sql_text: str):
         cf, gap = pd.DataFrame(), pd.DataFrame()
     return df, "OK", cf, gap
 
-# ---------------------- UI ----------------------
 with gr.Blocks(title="ALM LLM") as demo:
     gr.Markdown(INTRO)
 
@@ -61,11 +58,7 @@ with gr.Blocks(title="ALM LLM") as demo:
         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_PATH} LIMIT 20;",
-        )
+        sql_in = gr.Code(label="SQL", language="sql", value=f"SELECT * FROM {RESOLVED_PATH} LIMIT 20;")
         btn2 = gr.Button("Execute")
         df2   = gr.Dataframe(label="Result", interactive=True)
         status = gr.Textbox(label="Status", interactive=False)