Upload folder using huggingface_hub

Dockerfile

@@ -20,4 +20,4 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-CMD ["lite", "demo.yaml", "launch", "--server_name", "0.0.0.0", "--server_port", "7860"]
+RUN python gsql_app.py

gsql_app.py

@@ -1,18 +1,16 @@
-sql_query='''SELECT * FROM input_table 
-WHERE text LIKE '%curls%';''';
+
 import duckdb
 import pandas as pd
 import gradio as gr
 from datasets import load_dataset
 import tempfile
-import re
-
+from query import sql_query
 
 max_rows = 20
+max_cols = None
+
 df_display_kwargs = dict(
     wrap = True,
-    max_rows = max_rows,
-    type = "pandas",
     row_count = 3,
     col_count = 4,
 )
@@ -24,13 +22,6 @@ dataset_choices = [
 
 def apply_sql(input_table, sql_query):
 
-    # Use regex to extract the table name from the SQL query
-    match = re.search(r"\bFROM\s+(\w+)", sql_query, re.IGNORECASE)
-    if match:
-        table_name = match.group(1)
-
-    sql_query = sql_query.replace(table_name, "input_table")
-
     output_df = duckdb.query(sql_query).to_df()
 
     return output_df
@@ -39,18 +30,18 @@ def display_dataset(dataset_id):
 
     dataset = load_dataset(dataset_id, split="train")
     df = dataset.to_pandas()
-    return df, df
+    display_df = df.iloc[:max_rows, :max_cols]
+    return display_df, df
 
 def upload_dataset(dataset_file):
 
     if dataset_file is None:
         return None, None
+    
+    df = pd.read_csv(dataset_file.name).iloc[:max_rows, :max_cols]
+    display_df = df.iloc[:max_rows, :max_cols]
 
-    print(dataset_file.name)
-
-    df = pd.read_csv(dataset_file.name)
-
-    return df, df
+    return display_df, df
 
 
 def process_dataset(full_dataset, sql_query):
@@ -69,8 +60,7 @@ theme = gr.themes.Soft(
     neutral_hue="slate",
 )
 
-
-with gr.Blocks(analytics_enabled=False, theme=theme) as demo:
+with gr.Blocks(theme=theme) as demo:
     full_dataset = gr.State()
 
     with gr.Column():
@@ -123,3 +113,6 @@ with gr.Blocks(analytics_enabled=False, theme=theme) as demo:
     )
     demo.load(**toggle_dark_mode_args)
     dark_mode_btn.click(**toggle_dark_mode_args)
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)

query.py

@@ -0,0 +1,3 @@
+sql_query='''
+SELECT * FROM input_table WHERE text LIKE '%the rock%'
+'''

requirements.txt

@@ -1,2 +1,4 @@
-gradio==3.32
-liteobj==0.0.7
+gradio==3.33.1
+pandas>=2.0
+duckdb>=0.8.0
+datasets