adding mindmap

app.py

@@ -8,6 +8,7 @@ import langdetect
 import moviepy.editor as mp
 import yt_dlp
 import whisper
+from graphviz import Digraph
 
 print("Starting the program...")
 
@@ -67,6 +68,20 @@ def generate_summary_stream(transcription):
     bullet_summary = "\n".join(f"• {sentence.strip()}" for sentence in raw_summary.split('.') if sentence.strip())
     return bullet_summary
 
+def generate_mindmap_from_summary(summary_text):
+    dot = Digraph(comment='Mind Map')
+    dot.node('A', 'Summary')
+
+    lines = summary_text.split('\n')
+    for idx, line in enumerate(lines):
+        node_id = f'B{idx}'
+        dot.node(node_id, line.replace("• ", "").strip())
+        dot.edge('A', node_id)
+
+    output_path = generate_unique_filename(".png")
+    dot.render(output_path, format='png', cleanup=True)
+    return output_path + ".png"
+
 def process_youtube(url):
     if not url:
         return "No URL", None
@@ -101,8 +116,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Row():
         transcription_output = gr.Textbox(label="📝 Transcription", lines=10, show_copy_button=True)
         summary_output = gr.Textbox(label="📊 Summary Points", lines=10, show_copy_button=True)
+        mindmap_output = gr.Image(label="🧠 Mind Map")
 
     summary_button = gr.Button("📝 Generate Summary")
+    mindmap_button = gr.Button("🧠 Generate Mind Map")
 
     def process_video_and_update(video):
         if video is None:
@@ -113,5 +130,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     video_button.click(process_video_and_update, inputs=[video_input], outputs=[transcription_output, summary_output])
     url_button.click(process_youtube, inputs=[url_input], outputs=[transcription_output, summary_output])
     summary_button.click(generate_summary_stream, inputs=[transcription_output], outputs=[summary_output])
+    mindmap_button.click(generate_mindmap_from_summary, inputs=[summary_output], outputs=[mindmap_output])
 
 demo.launch(share=True)