update

app.py

@@ -589,7 +589,7 @@ def create_app():
         # RoBERTa Sentiment Analysis Tab (FIXED)
         with gr.Tab("RoBERTa Sentiment"):
             # Create the RoBERTa analysis UI components
-            run_roberta_btn, roberta_output, visualization_style, visualization_container, roberta_status = create_roberta_screen()
+            run_roberta_btn, roberta_output, visualization_container, roberta_status = create_roberta_screen()
             
             # Create a container for visualization results
             with gr.Column() as roberta_viz_container:
@@ -598,7 +598,7 @@ def create_app():
                 roberta_viz_content = gr.HTML("", visible=False)
             
             # Function to run RoBERTa sentiment analysis (FIXED)
-            def run_roberta_analysis(dataset, visualization_style):
+            def run_roberta_analysis(dataset):
                 try:
                     print("Starting run_roberta_analysis function")
                     if not dataset or "entries" not in dataset or not dataset["entries"]:
@@ -610,10 +610,10 @@ def create_app():
                             gr.update(visible=False)   # roberta_viz_content
                         )
                     
-                    print(f"Running RoBERTa sentiment analysis with sentence-level, style={visualization_style}")
+                    print(f"Running RoBERTa sentiment analysis with sentence-level, style=")
                     
                     # Process the analysis request
-                    roberta_results = process_roberta_request(dataset, visualization_style)
+                    roberta_results = process_roberta_request(dataset)
 
                     print(f"RoBERTa results obtained. Size: {len(str(roberta_results))} characters")
                     
@@ -693,7 +693,7 @@ def create_app():
             # Connect the run button to the analysis function (FIXED)
             run_roberta_btn.click(
                 fn=run_roberta_analysis,
-                inputs=[dataset_state, visualization_style],
+                inputs=[dataset_state],
                 outputs=[
                     roberta_results_state,
                     roberta_status,

ui/roberta_screen.py

@@ -34,9 +34,9 @@ def create_roberta_screen():
         # but is included for backward compatibility
         visualization_container = gr.Markdown(visible=False)
         
-    return run_roberta_btn, roberta_output, visualization_style, visualization_container, roberta_status
+    return run_roberta_btn, roberta_output, visualization_container, roberta_status
 
-def process_roberta_request(dataset, visualization_style="Standard"):
+def process_roberta_request(dataset):
     """
     Process the RoBERTa sentiment analysis request
     
@@ -84,7 +84,7 @@ def process_roberta_request(dataset, visualization_style="Standard"):
         # Add metadata about the analysis
         results["analyses"][prompt_text]["roberta_sentiment"]["metadata"] = {
             "sentence_level": "",
-            "visualization_style": visualization_style
+            "visualization_style": ""
         }
         
         return results