Update app.py

app.py

@@ -60,12 +60,14 @@ def text2sql(dataset_name, query_input):
         first_parquet = get_first_parquet(dataset_name)
     except Exception as error:
         return {
+            schema_output: "",
+            prompt_output: "",
             query_output: "",
             df:pd.DataFrame([{"error": f"❌ Could not get dataset schema. {error=}"}])
         }
 
     first_parquet_url = first_parquet["url"]
-    print(first_parquet_url)
+    print(f"getting schema from {first_parquet_url}")
     con = duckdb.connect()
     con.execute("INSTALL 'httpfs'; LOAD httpfs;")
     # could get from Parquet instead?
@@ -89,6 +91,8 @@ def text2sql(dataset_name, query_input):
         sql_output =  query_remote_model(text)
     except Exception as error:
         return {
+            schema_output: ddl_create,
+            prompt_output: text,
             query_output: "",
             df:pd.DataFrame([{"error": f"❌ Unable to get the SQL query based on the text. {error=}"}])
         }
@@ -100,6 +104,8 @@ def text2sql(dataset_name, query_input):
     finally:
         con.close()
     return {
+        schema_output: ddl_create,
+        prompt_output: text,
         query_output:sql_output,
         df:query_result
     }
@@ -112,7 +118,9 @@ with gr.Blocks() as demo:
     dataset_name = gr.Textbox("sksayril/medicine-info", label="Dataset Name")
     query_input = gr.Textbox("How many rows there are?", label="Ask something about your data")
     btn = gr.Button("Generate SQL")
+    schema_output = gr.Textbox(label="Parquet Schema as CREATE DDL", interactive= False)
+    prompt_output = gr.Textbox(label="Generated prompt", interactive= False)
     query_output = gr.Textbox(label="Output SQL", interactive= False)    
     df = gr.DataFrame(datatype="markdown")
-    btn.click(text2sql, inputs=[dataset_name, query_input], outputs=[query_output,df])
+    btn.click(text2sql, inputs=[dataset_name, query_input], outputs=[schema_output, prompt_output, query_output,df])
 demo.launch(debug=True)