Commit
·
8f8b860
1
Parent(s):
2377601
feat: warning when no dataset is selected
Browse files
app.py
CHANGED
|
@@ -16,6 +16,8 @@ client = InferenceClient(
|
|
| 16 |
|
| 17 |
|
| 18 |
def get_iframe(hub_repo_id, sql_query=None):
|
|
|
|
|
|
|
| 19 |
if sql_query:
|
| 20 |
sql_query = urllib.parse.quote(sql_query)
|
| 21 |
url = f"https://huggingface.co/datasets/{hub_repo_id}/embed/viewer?sql_console=true&sql={sql_query}"
|
|
@@ -97,7 +99,7 @@ with gr.Blocks() as demo:
|
|
| 97 |
placeholder="Enter a natural language query to generate SQL",
|
| 98 |
)
|
| 99 |
with gr.Row():
|
| 100 |
-
sql_out = gr.Code(label="SQL Query")
|
| 101 |
with gr.Row():
|
| 102 |
btn2 = gr.Button("Query Dataset")
|
| 103 |
|
|
@@ -116,7 +118,7 @@ with gr.Blocks() as demo:
|
|
| 116 |
fn=query_dataset,
|
| 117 |
inputs=[search_in, features, query],
|
| 118 |
outputs=[sql_out, search_out],
|
| 119 |
-
)
|
| 120 |
|
| 121 |
if __name__ == "__main__":
|
| 122 |
demo.launch()
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
def get_iframe(hub_repo_id, sql_query=None):
|
| 19 |
+
if not hub_repo_id:
|
| 20 |
+
raise ValueError("Hub repo id is required")
|
| 21 |
if sql_query:
|
| 22 |
sql_query = urllib.parse.quote(sql_query)
|
| 23 |
url = f"https://huggingface.co/datasets/{hub_repo_id}/embed/viewer?sql_console=true&sql={sql_query}"
|
|
|
|
| 99 |
placeholder="Enter a natural language query to generate SQL",
|
| 100 |
)
|
| 101 |
with gr.Row():
|
| 102 |
+
sql_out = gr.Code(label="SQL Query", interactive=True, language="sql", lines=1)
|
| 103 |
with gr.Row():
|
| 104 |
btn2 = gr.Button("Query Dataset")
|
| 105 |
|
|
|
|
| 118 |
fn=query_dataset,
|
| 119 |
inputs=[search_in, features, query],
|
| 120 |
outputs=[sql_out, search_out],
|
| 121 |
+
)z
|
| 122 |
|
| 123 |
if __name__ == "__main__":
|
| 124 |
demo.launch()
|