Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 import fitz  # PyMuPDF
 import requests
 from graphviz import Digraph
-import io
 
 # Funzione per estrarre testo da un PDF
 def extract_text_from_pdf(pdf_file):
@@ -27,12 +26,23 @@ def search_images(query):
 
 # Funzione per creare una mappa mentale
 def create_mind_map(text, images):
-    dot = Digraph()
-    dot.node('A', 'Concept Central', style='filled', fillcolor='lightblue')
-    for i, img in enumerate(images):
-        dot.node(f'B{i}', text[i] if i < len(text) else 'Other', shape='rect', style='filled', fillcolor='lightyellow')
-        dot.edge('A', f'B{i}', label=f'Image {i+1}')
-        dot.node(f'B{i}', img, image=img)
+    dot = Digraph(format='png', graph_attr={'rankdir': 'LR'})
+    dot.node('A', 'Concept Central', shape='ellipse', style='filled', fillcolor='lightblue', fontsize='24')
+
+    # Divide il testo in righe per i nodi
+    lines = text.split('\n')
+    
+    for i, line in enumerate(lines):
+        if line.strip():  # Solo linee non vuote
+            # Aggiungi un nodo per ogni linea di testo
+            dot.node(f'B{i}', line.strip(), shape='rect', style='filled', fillcolor='lightyellow', fontsize='18')
+            dot.edge('A', f'B{i}', label=f'Subject {i + 1}')
+            
+            # Aggiungi le immagini associate
+            if i < len(images):
+                dot.node(f'Image{i}', '', image=images[i], shape='rect', width='2', height='2')
+                dot.edge(f'B{i}', f'Image{i}')
+
     return dot
 
 # Inizializzazione della app Streamlit
@@ -49,15 +59,16 @@ if input_option == 'Carica PDF':
 else:
     text = st.text_area("Inserisci il testo:", height=300)
 
-# Menu a tendina per le opzioni di mappa
-map_type = st.selectbox("Scegli il tipo di mappa", ["Mappa Concettuale", "Timeline", "Diagramma di Flusso", "Diagramma a Ragnatela"])
-
 # Pulsante per generare la mappa
 if st.button("Genera Mappa"):
     if text:
-        # Esegui una ricerca di immagini basate sul testo
-        images = search_images(text.split()[:3])  # Usa le prime tre parole per la ricerca
-        mind_map = create_mind_map(text.split('\n'), images)
+        # Esegui una ricerca di immagini basate sulle prime tre parole del testo
+        keywords = text.split()[:3]  # Usa le prime tre parole per la ricerca
+        images = []
+        for keyword in keywords:
+            images += search_images(keyword)  # Aggiungi immagini per ciascuna parola
+        
+        mind_map = create_mind_map(text, images)
         st.graphviz_chart(mind_map)
     else:
         st.warning("Per favore, inserisci del testo o carica un PDF.")