Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from processing import process_input
 from visualization import update_visibility_and_charts
-import json
+import plotly.graph_objs as go
 
 def create_interface():
     with gr.Blocks() as iface:
@@ -31,10 +31,10 @@ def create_interface():
             # Create and update charts and explanations
             outputs = update_visibility_and_charts(*results)
             
-            # Convert JSON-encoded Plotly figures back to Plotly objects
+            # Convert Plotly figures to the format expected by gr.Plot
             for i, output in enumerate(outputs):
-                if isinstance(output, str) and output.startswith('{"data":'):
-                    outputs[i] = json.loads(output)
+                if isinstance(output, go.Figure):
+                    outputs[i] = output
             
             return outputs