Update app.py

app.py

@@ -10,6 +10,7 @@ import streamlit as st
 import os
 from groq import Groq
 import time
+from openpyxl import load_workbook
 
 def clean_text(text):
     return re.sub(r'\s+', ' ', text).strip()
@@ -154,12 +155,15 @@ def get_elenco_comuni(provincia):
 cerca_premuto = False
 comuni_provincia = {}
 
-def scrivi_dataframe(output, riepilogo, comune):
-    if len(output) > 0: 
+def scrivi_dataframe(output, riepilogo, comune, da_excel = False, df_excel = None):
+    if len(output) > 0 or (da_excel and df_excel is not None): 
         st.numero_immobili_validi = st.numero_immobili_validi + 1
         if not riepilogo:
             st.write(f"### {comune}")
-        df = pd.DataFrame(output)
+        if da_excel: 
+            df = df_excel
+        else:
+            df = pd.DataFrame(output)
         df_originale = df.sort_values(by=["Vantaggio", "PrezzoMq"], ascending=[False, True])
         if not riepilogo:
             columns_to_display = ["Vantaggioso", "Vantaggio", "Immagine", "Titolo", "PrezzoMq", "Prezzo", "Superficie", "Locali", "PrezzoMedioMq", "Link"]
@@ -386,21 +390,58 @@ with st.sidebar:
 
     cerca_premuto = st.button("Cerca", use_container_width=True, type='primary')
 
+def importa_excel(file):
+    df = None
+    if file.name.endswith('.csv'):
+        df = pd.read_csv(file)
+    elif file.name.endswith('.xlsx') or file.name.endswith('.xls'):
+        df = pd.read_excel(file)
+    else:
+        st.error(f"Formato di file non supportato: {file.name}")
+    if df is not None: 
+        wb = load_workbook(file)
+        ws = wb.active
+        if df.columns[0] != "Comune":
+            df = pd.read_excel(file, header=1)
+        if 'Link asta' in df.columns:
+            link_column_index = df.columns.get_loc("Link asta") + 1  
+            for i in range(2, ws.max_row + 1): 
+                cell = ws.cell(row=i, column=link_column_index)
+                if cell.hyperlink:
+                    df.at[i - 2, "Link asta"] = cell.hyperlink.target
+        if 'Perizia' in df.columns:
+            link_column_index = df.columns.get_loc("Perizia") + 1  
+            for i in range(2, ws.max_row + 1): 
+                cell = ws.cell(row=i, column=link_column_index)
+                if cell.hyperlink:
+                    df.at[i - 2, "Perizia"] = cell.hyperlink.target
+        columns_to_display = ["Comune", "Off. min. [€]", "Locali [nr]", "Mq", "Descrizione", "Link asta", "Perizia"]
+        df = df[columns_to_display]
+        df = df.rename(columns={
+            "Off. min. [€]": 'Prezzo',
+            'Locali [nr]': 'Locali', 
+            'Descrizione':'Titolo', 
+            'Mq':'Superficie', 
+            'Link asta': 'Link'
+        })
+        df['PrezzoMq'] = 0 #df['Prezzo'] / df['Superficie']
+        df['Vantaggioso'] = False
+        df['Vantaggio'] = 50
+        df['Immagine'] = ""
+        df['PrezzoMedioMq'] = 0
+    return df
+
 if cerca_premuto:
     dfs = []
     if uploaded_files:
         for file in uploaded_files:
-            if file.name.endswith('.csv'):
-                df = pd.read_csv(file)
-            elif file.name.endswith('.xlsx') or file.name.endswith('.xls'):
-                df = pd.read_excel(file)
-            else:
-                st.error(f"Formato di file non supportato: {file.name}")
-                continue
+            df = importa_excel(file)
             dfs.append(df)
         if dfs:
             concatenated_df = pd.concat(dfs, ignore_index=True)
-            st.write(concatenated_df)    
+            scrivi_dataframe([], False, 'Excel', True, concatenated_df)
+            #st.write(concatenated_df)
+        st.success("File Excel importati con successo")
     
     if len(comune_input)>0:
         comuni_selezionati = comune_input
@@ -448,8 +489,9 @@ if cerca_premuto:
             #st.write("### Dettagliata")
             #analizza_dati_ai(output, 2)
         st.success("Elaborazione Completata")
-    else:
-        st.error("Per favore, inserisci il nome di un comune.")
+    
+    if not uploaded_files and len(comune_input) == 0:
+        st.error("Per favore, inserisci il nome di un comune o seleziona un file excel")
                     
 #if 'initialized' not in st.session_state:
 #    st.session_state['initialized'] = True