Layout fixes and removes checkboxes

app.py

@@ -3,7 +3,8 @@ import pandas as pd
 import duckdb
 
 PARQUET_FILES = {
-    "Animagine XL 3.1": 'data.parquet', 
+    "File 1": 'data1.parquet',  # Ganti dengan path file Anda
+    "File 2": 'data2.parquet'   # Ganti dengan path file Anda
 }
 
 preset_values = [
@@ -13,23 +14,16 @@ preset_values = [
     "1girl, elaina (majo no tabitabi), majo no tabitabi"
 ]
 
-def search_data(search_term, selected_file, use_duckdb=False):
+def search_data(search_term, selected_file):
     try:
-        if use_duckdb:
-            con = duckdb.connect(database=':memory:', read_only=False)
-            con.execute(f"CREATE TABLE my_table AS SELECT * FROM read_parquet('{PARQUET_FILES[selected_file]}')")
-            if search_term.strip():
-                results = con.execute(f"SELECT * FROM my_table WHERE lower(teks) LIKE '%{search_term.lower()}%'").fetchdf()
-            else:
-                search_terms_sql = " OR ".join([f"lower(teks) LIKE '%{term.lower()}%'" for term in preset_values])
-                results = con.execute(f"SELECT * FROM my_table WHERE {search_terms_sql}").fetchdf()
-            con.close()
+        con = duckdb.connect(database=':memory:', read_only=False)
+        con.execute(f"CREATE TABLE my_table AS SELECT * FROM read_parquet('{PARQUET_FILES[selected_file]}')")
+        if search_term.strip():
+            results = con.execute(f"SELECT * FROM my_table WHERE lower(teks) LIKE '%{search_term.lower()}%'").fetchdf()
         else:
-            df = pd.read_parquet(PARQUET_FILES[selected_file])
-            if search_term.strip():
-                results = df[df['teks'].str.lower().str.contains(search_term.lower(), na=False)]
-            else:
-                results = df[df['teks'].str.lower().str.contains("|".join(preset_values).lower(), na=False)]
+            search_terms_sql = " OR ".join([f"lower(teks) LIKE '%{term.lower()}%'" for term in preset_values])
+            results = con.execute(f"SELECT * FROM my_table WHERE {search_terms_sql}").fetchdf()
+        con.close()
 
         if len(results.columns) > 12:
             results = results.iloc[:, :12]
@@ -42,40 +36,34 @@ def search_data(search_term, selected_file, use_duckdb=False):
 if __name__ == "__main__":
     with gr.Blocks() as app:
         gr.Markdown("##  Text Search for Animagine tag characters")
-        with gr.Row(): #Menambahkan row untuk dropdown dan checkbox
-          file_dropdown = gr.Dropdown(
-              choices=list(PARQUET_FILES.keys()),
-              label="Select Parquet File",
-              value=list(PARQUET_FILES.keys())[0], # Default value
-          )
-          use_duckdb_checkbox = gr.Checkbox(label="Use DuckDB", value=True) # Checkbox untuk DuckDB, default true
-        with gr.Column():
-            search_input = gr.Textbox(
-                label="Search for characters or series:",
-                placeholder="sousou no frieren",
-            )
-            search_output = gr.Dataframe(
-                label="Search Results",
-                value=pd.DataFrame({'Characters tag': preset_values}),
-                headers="auto",
-            )
+        with gr.Row(): # Menggunakan Row untuk layout horizontal
+            with gr.Column(): # Kolom untuk input pencarian
+                search_input = gr.Textbox(
+                    label="Search for characters or series:",
+                    placeholder="sousou no frieren",
+                )
+            with gr.Column(): # Kolom untuk dropdown
+                file_dropdown = gr.Dropdown(
+                    choices=list(PARQUET_FILES.keys()),
+                    label="Select Parquet File",
+                    value=list(PARQUET_FILES.keys())[0], # Default value
+                )
+        search_output = gr.Dataframe( # Output di luar row, di bawahnya
+            label="Search Results",
+            value=pd.DataFrame({'Characters tag': preset_values}),
+            headers="auto",
+        )
 
-        inputs = [search_input, file_dropdown, use_duckdb_checkbox] # Menambahkan input dropdown dan checkbox
+        inputs = [search_input, file_dropdown]
         search_input.change(
             fn=search_data,
             inputs=inputs,
             outputs=search_output,
         )
-        file_dropdown.change( #Menambahkan event change untuk dropdown
-            fn=search_data,
-            inputs=inputs,
-            outputs=search_output,
-        )
-        use_duckdb_checkbox.change( #Menambahkan event change untuk checkbox
+        file_dropdown.change(
             fn=search_data,
             inputs=inputs,
             outputs=search_output,
         )
 
-
     app.launch()