revert to matplotlib

app.py

@@ -9,16 +9,20 @@ with gr.Blocks(title="PSYCHOHISTORY") as app:
             btn_search = gr.Button("Look", scale=1)
         with gr.Row():
             mem_results = gr.JSON(label="Results")
-            html_output = gr.HTML(label="Graph Visualization") # Use HTML component
             btn_search.click(
                 gen.generate,
                 inputs=[txt_search],
                 outputs=mem_results
             )
+
+        with gr.Row():
+            img_output = gr.Image(label="Graph Visualization", type="filepath")  # Add an Image component
+
+            # Trigger graph generation after JSON is generated
             mem_results.change(
                 psychohistory.main,
                 inputs=[mem_results],
-                outputs=html_output  # Output to the HTML component
+                outputs=img_output
             )
 
 if __name__ == "__main__":