feat: add basic working version

app.py

@@ -35,15 +35,17 @@ def get_iframe(hub_repo_id, sql_query=None):
 def get_column_info(hub_repo_id):
     url: str = f"https://datasets-server.huggingface.co/info?dataset={hub_repo_id}"
     response = requests.get(url)
-    data = response.json()
-    data = data.get("dataset_info")
-    key = list(data.keys())[0]
-    description = data.get(key).get("description")
-    features = json.dumps(data.get(key).get("features"))
-    return description, features
+    try:
+        data = response.json()
+        data = data.get("dataset_info")
+        key = list(data.keys())[0]
+        features: str = json.dumps(data.get(key).get("features"))
+    except Exception as e:
+        gr.Error(f"Error getting column info: {e}")
+    return features
 
 
-def query_dataset(hub_repo_id, description, features, query):
+def query_dataset(hub_repo_id, features, query):
     messages = [
         {
             "role": "system",
@@ -51,9 +53,7 @@ def query_dataset(hub_repo_id, description, features, query):
         },
         {
             "role": "user",
-            "content": f"""# Description
-{description}
-
+            "content": f"""table train
 # Features
 {features}
 
@@ -84,12 +84,9 @@ with gr.Blocks() as demo:
     with gr.Row():
         search_out = gr.HTML(label="Search Results")
     with gr.Row():
-        description = gr.Textbox(
-            label="Description", placeholder="Description from dataset or project page"
-        )
-        features = gr.Code(label="Features", language="json")
+        features = gr.Code(label="Features", language="json", visible=False)
     with gr.Row():
-        query = gr.Textbox(label="Query")
+        query = gr.Textbox(label="Query", placeholder="Enter a query to generate SQL")
     with gr.Row():
         sql_out = gr.Code(label="SQL Query")
     with gr.Row():
@@ -103,13 +100,13 @@ with gr.Blocks() as demo:
     ).then(
         fn=get_column_info,
         inputs=[search_in],
-        outputs=[description, features],
+        outputs=[features],
     )
 
     btn2.click(
         fn=query_dataset,
-        inputs=[search_in, description, features, query],
-        outputs=[sql_out],
+        inputs=[search_in, features, query],
+        outputs=[sql_out, search_out],
     )
 
 if __name__ == "__main__":