update

app.py

@@ -98,6 +98,8 @@ def create_app():
             # Use create_analysis_screen to get UI components including visualization container 
             analysis_options, analysis_params, run_analysis_btn, analysis_output, ngram_n = create_analysis_screen()
             
+            topic_count = gr.State(value=3)
+
             # Pre-create visualization components (initially hidden)
             visualization_area_visible = gr.Checkbox(value=False, visible=False, label="Visualization Visible")
             analysis_title = gr.Markdown("## Analysis Results", visible=False)
@@ -121,7 +123,7 @@ def create_app():
             status_message = gr.Markdown(visible=False)
             
             # Define a helper function to extract parameter values and run the analysis
-            def run_analysis(dataset, selected_analysis, bow_top, ngram_n, ngram_top):
+            def run_analysis(dataset, selected_analysis, bow_top, ngram_n, ngram_top, topic_count=3):
                 try:
                     if not dataset or "entries" not in dataset or not dataset["entries"]:
                         return (
@@ -145,7 +147,8 @@ def create_app():
                         "bow_top": bow_top,
                         "ngram_n": ngram_n,
                         "ngram_top": ngram_top,
-                        "topic_count": topic_count
+                        "topic_count": topic_count,
+                        "bias_methods": ["partisan"]
                     }
                     print(f"Running analysis with selected type: {selected_analysis}")
                     print("Parameters:", parameters)
@@ -489,7 +492,7 @@ def create_app():
         # Run analysis with proper parameters
         run_analysis_btn.click(
             fn=run_analysis,
-            inputs=[dataset_state, analysis_options, ngram_n],
+            inputs=[dataset_state, analysis_options, ngram_n, topic_count],
             outputs=[
                 analysis_results_state,
                 analysis_output,