Update app.py

app.py

@@ -101,22 +101,28 @@ def text2sql(dataset_name, query_input):
 
 
 with gr.Blocks() as demo:
+    with gr.Row():
     gr.Markdown("# Generate SQL queries based on a given text for your dataset")
     gr.Markdown("This space showcase how to generate a SQL query from a text and get the result.")
     gr.Markdown("Tech stack: duckdb and DuckDB-NSQL-7B model")
-    dataset_name = gr.Textbox("jamescalam/world-cities-geo", label="Dataset Name")
-    query_input = gr.Textbox("Cities from Albania country", label="Ask something about your data")
-    examples = [
-        ["Cities from Albania country"],
-        ["The continent with the most number of countries"],
-        ["Cities that start with 'A'"],
-        ["Cities by region"],
-    ]
-    gr.Examples(examples=examples, inputs=[query_input],outputs=[])
-    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")
+    with gr.Row():
+        with gr.Column(scale=1, min_width=600):
+            dataset_name = gr.Textbox("jamescalam/world-cities-geo", label="Dataset Name")
+            query_input = gr.Textbox("Cities from Albania country", label="Ask something about your data")
+            examples = [
+                ["Cities from Albania country"],
+                ["The continent with the most number of countries"],
+                ["Cities that start with 'A'"],
+                ["Cities by region"],
+            ]
+            gr.Examples(examples=examples, inputs=[query_input],outputs=[])
+            btn = gr.Button("Generate SQL")
+    with gr.Row():
+        with gr.Column(scale=1, min_width=600):
+            schema_output = gr.Textbox(label="Parquet Schema as CREATE DDL", interactive= False)
+            prompt_output = gr.Textbox(label="Generated prompt", interactive= False)
+        with gr.Column(scale=1, min_width=600):
+            query_output = gr.Textbox(label="Output SQL", interactive= False)    
+            df = gr.DataFrame(datatype="markdown")
     btn.click(text2sql, inputs=[dataset_name, query_input], outputs=[schema_output, prompt_output, query_output,df])
 demo.launch(debug=True)