Update app.py

app.py

@@ -2,10 +2,11 @@ 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):
-    pdf_document = fitz.open(pdf_file)
+    pdf_document = fitz.open(stream=pdf_file.read(), filetype="pdf")
     text = ""
     for page in pdf_document:
         text += page.get_text()
@@ -24,31 +25,18 @@ def search_images(query):
         images.append(img_url)
     return images
 
-# Funzione per creare una mappa mentale dettagliata con più immagini
-def create_detailed_mind_map(text_lines, images):
+# Funzione per creare una mappa mentale
+def create_mind_map(text, images):
     dot = Digraph()
-
-    # Nodo principale
-    dot.node('A', 'Concetti Principali', shape='ellipse', style='filled', color='lightblue', fontcolor='black', fontsize='20')
-
-    for i, line in enumerate(text_lines):
-        # Aggiungi nodi per ogni riga di testo
-        node_color = f"#{(i*40)%255:02x}{(i*80)%255:02x}{(i*120)%255:02x}"  # Colore variabile per ogni nodo
-        dot.node(f'B{i}', line, shape='rect', style='filled', color=node_color, fontcolor='white', fontsize='14', width='2.5')
-
-        # Collega il nodo principale con i nodi secondari
-        dot.edge('A', f'B{i}', label='Dettaglio', color='black', style='dashed')
-
-        # Aggiungi immagini se disponibili
-        for j in range(min(3, len(images))):  # Aggiungi fino a 3 immagini per nodo
-            img_node = f'Img{i}_{j}'
-            dot.node(img_node, '', image=images[j], shape='rect', width='0.5', height='0.5')
-            dot.edge(f'B{i}', img_node, label='Immagine', color='black')
-
+    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)
     return dot
 
 # Inizializzazione della app Streamlit
-st.title("Generatore di Mappe Mentali Iperdettagliate")
+st.title("Generatore di Mappe Mentali")
 st.write("Carica un PDF o inserisci del testo per generare una mappa mentale.")
 
 # Opzioni di input
@@ -67,14 +55,9 @@ map_type = st.selectbox("Scegli il tipo di mappa", ["Mappa Concettuale", "Timeli
 # Pulsante per generare la mappa
 if st.button("Genera Mappa"):
     if text:
-        text_lines = text.split('\n')
-        images = []
-        
-        # Esegui una ricerca di immagini per ogni riga di testo
-        for line in text_lines[:3]:  # Limita la ricerca alle prime tre righe
-            images.extend(search_images(line))
-
-        mind_map = create_detailed_mind_map(text_lines, images)
+        # 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)
         st.graphviz_chart(mind_map)
     else:
         st.warning("Per favore, inserisci del testo o carica un PDF.")