Update app.py

app.py

@@ -15,6 +15,7 @@ from PIL import Image
 import openai
 import pandas as pd
 from graphviz import Digraph
+import plotly.express as px
 
 HRA_TOKEN=os.getenv("HRA_TOKEN")
 
@@ -83,7 +84,15 @@ def getmindmap(topic,openapikey):
     name='temp.png'
     img.save(name)
 
-    return img
+    fig = px.treemap(df, path=['name', 'level1', 'level2'], 
+                 values='count', color='level1')
+    fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
+    fig.show()
+    fig.write_image('temp1.png')
+    img1 = Image.open("temp1.png")
+
+
+    return img,img1
 
 with gr.Blocks() as demo:
     gr.Markdown("<h1><center>Mind Map Generator</center></h1>")
@@ -97,11 +106,14 @@ with gr.Blocks() as demo:
             btn = gr.Button("Generate")    
             examples = gr.Examples(examples=['English Premier League','Heavy metal music','Face recognition','Arsenal Football Club',],
                            inputs=[textbox1])
+    with gr.Row() as row:    
+        with gr.Column():
+            output_image1 = gr.components.Image(label="Your Mind Map as Graph")
         with gr.Column():
-            output_image1 = gr.components.Image(label="Your Mind Map")
+            output_image2 = gr.components.Image(label="Your Mind Map as Tree Map")
 
 
-    btn.click(getmindmap,inputs=[textbox1,textbox2], outputs=[output_image1])
+    btn.click(getmindmap,inputs=[textbox1,textbox2], outputs=[output_image1,output_image2])
     
 
 demo.launch()