Co

app.py

@@ -140,12 +140,13 @@ def generate_hamiltonians(num_hamiltonians, selected_qubits, selected_order, wri
         store_in_duckdb(results_table)
 
 # Function to load results from either DuckDB or Hugging Face dataset
-def load_results(load_from_hf, load_from_duckdb):
-    if load_from_hf:
-        return load_from_hf_dataset()
-    if load_from_duckdb:
-        return load_from_duckdb()
+def load_results(load_from_hf_checkbox, load_from_duckdb_checkbox):
+    if load_from_hf_checkbox:
+        return load_from_hf_dataset()  # Load from HF dataset
+    if load_from_duckdb_checkbox:
+        return load_from_duckdb()  # Load from DuckDB
 
+# Gradio app
 # Gradio app
 with gr.Blocks() as app:
     gr.Markdown("# Quantum Hamiltonian Generator")
@@ -157,26 +158,40 @@ with gr.Blocks() as app:
         order_choices = [1, 2, 3, 4, 5]
         selected_order = gr.Dropdown(label="Select Trotter order", choices=order_choices, value=1)
 
-        # Checkboxes for writing to HF dataset and DuckDB
-        write_to_hf = gr.Checkbox(label="Write to Hugging Face dataset", value=False)
-        write_to_duckdb = gr.Checkbox(label="Write to DuckDB", value=True)
+        # Radio buttons for selecting either Hugging Face dataset or DuckDB
+        write_option = gr.Radio(label="Where do you want to store the data?", 
+                                choices=["Write to Hugging Face dataset", "Write to DuckDB"], 
+                                value="Write to Hugging Face dataset")
 
         generate_button = gr.Button("Generate Hamiltonians")
         status = gr.Markdown("Click 'Generate Hamiltonians' to start the process.")
 
-        def update_status(num, qubits, order, write_hf, write_duckdb):
-            generate_hamiltonians(num, qubits, order, write_hf, write_duckdb)
+        def update_status(num, qubits, order, write_option):
+            if write_option == "Write to Hugging Face dataset":
+                # Call function to write to Hugging Face dataset
+                generate_hamiltonians(num, qubits, order, write_to_hf=True, write_to_duckdb=False)
+            else:
+                # Call function to write to DuckDB
+                generate_hamiltonians(num, qubits, order, write_to_hf=False, write_to_duckdb=True)
             return "Data stored as per selection."
 
-        generate_button.click(update_status, inputs=[num_hamiltonians, selected_qubits, selected_order, write_to_hf, write_to_duckdb], outputs=status)
+        generate_button.click(update_status, inputs=[num_hamiltonians, selected_qubits, selected_order, write_option], outputs=status)
 
     with gr.Tab("View Results"):
-        load_from_hf = gr.Checkbox(label="Load from Hugging Face dataset", value=False)
-        load_from_duckdb = gr.Checkbox(label="Load from DuckDB", value=True)
+        load_option = gr.Radio(label="Where do you want to load the data from?", 
+                               choices=["Load from Hugging Face dataset", "Load from DuckDB"], 
+                               value="Load from DuckDB")
 
         load_button = gr.Button("Load Results")
         output_display = gr.HTML()
 
-        load_button.click(load_results, inputs=[load_from_hf, load_from_duckdb], outputs=output_display)
+        def load_results(load_option):
+            if load_option == "Load from Hugging Face dataset":
+                return load_from_hf_dataset()
+            else:
+                return load_from_duckdb()
+
+        load_button.click(load_results, inputs=[load_option], outputs=output_display)
+
 
-app.launch()
+app.launch(share=True)