Update app.py

app.py

@@ -3,16 +3,16 @@ import fitz  # PyMuPDF
 import requests
 from graphviz import Digraph
 
-# Funzione per estrarre testo da un PDF
+# Function to extract text from a PDF
 def extract_text_from_pdf(pdf_file):
-    pdf_document = fitz.open(stream=pdf_file.read(), filetype="pdf")
+    pdf_document = fitz.open(pdf_file)
     text = ""
     for page in pdf_document:
         text += page.get_text()
     pdf_document.close()
     return text
 
-# Funzione per cercare immagini tramite Wikipedia API
+# Function to search images via Wikipedia API
 def search_images(query):
     url = f"https://en.wikipedia.org/w/api.php?action=query&list=search&srsearch={query}&format=json"
     response = requests.get(url)
@@ -24,32 +24,23 @@ def search_images(query):
         images.append(img_url)
     return images
 
-# Funzione per creare una mappa mentale
+# Function to create a mind map
 def create_mind_map(text, images):
-    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}')
-
+    dot = Digraph()
+    dot.node('A', 'Concept Central', shape='box', style='filled', fillcolor='lightblue')
+    for i, img in enumerate(images):
+        node_label = text[i] if i < len(text) else 'Other'
+        dot.node(f'B{i}', node_label, shape='ellipse', style='filled', fillcolor='lightyellow')
+        dot.edge('A', f'B{i}', label=f'Image {i+1}')
+        dot.node(f'C{i}', img, image=img, shape='rect')
+        dot.edge(f'B{i}', f'C{i}')
     return dot
 
-# Inizializzazione della app Streamlit
+# Streamlit app initialization
 st.title("Generatore di Mappe Mentali")
 st.write("Carica un PDF o inserisci del testo per generare una mappa mentale.")
 
-# Opzioni di input
+# Input options
 input_option = st.radio("Scegli un'opzione:", ('Carica PDF', 'Scrivi Testo'))
 if input_option == 'Carica PDF':
     uploaded_file = st.file_uploader("Scegli un file PDF...", type="pdf")
@@ -59,16 +50,14 @@ if input_option == 'Carica PDF':
 else:
     text = st.text_area("Inserisci il testo:", height=300)
 
-# Pulsante per generare la mappa
+# Dropdown for map options
+map_type = st.selectbox("Scegli il tipo di mappa", ["Mappa Concettuale", "Timeline", "Diagramma di Flusso", "Diagramma a Ragnatela"])
+
+# Button to generate the map
 if st.button("Genera Mappa"):
     if text:
-        # 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)
+        images = search_images(text.split()[:3])  # Use the first three words for the search
+        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.")