Update app.py

app.py

@@ -8,10 +8,6 @@ from collections import namedtuple
 import numpy as np
 import streamlit as st
 from groq import Groq
-from dotenv import load_dotenv
-import os
-load_dotenv()
-API_GROQ = os.getenv('API_GROQ')
 
 def clean_text(text):
     return re.sub(r'\s+', ' ', text).strip()
@@ -83,7 +79,7 @@ def extract_info(provincia, comune, prezzo_medio_mq, listing):
 def scrape_immobiliare(provincia, comune, prezzo_medio_mq, prezzo_minimo, prezzo_massimo, locali_minimo, locali_massimo):
     print(provincia + " " + comune + " " + prezzo_medio_mq)
     comune_url = comune.replace(" ", "-")
-    if tipologia_case == "Asta Immobiliare":
+    if st.tipologia_case == "Asta Immobiliare":
         tipologia_url = "aste-immobiliari"
     else: 
         tipologia_url = "vendita-case"
@@ -131,52 +127,14 @@ def get_elenco_comuni(provincia):
             })
     return results 
 
-st.set_page_config(layout="wide")
-st.title('🏠 Immobiliare A.I. ')
-st.write("##### Il tuo assistente di intelligenza artificiale per la ricerca di occasioni immobiliari")
-with st.expander("Informazioni"):
-    st.write("Immobiliare A.I. è la webapp che semplifica la ricerca di immobili, grazie a algoritmi avanzati che calcolano il vantaggio di ogni offerta. Trova le migliori occasioni sul mercato con analisi precise e personalizzate. Scopri l’immobile giusto per te con facilità e sicurezza!")
-    
 cerca_premuto = False
 comuni_provincia = {}
 
-with st.sidebar: 
-    if "numero_immobili_validi" not in st.session_state:
-        st.numero_immobili_validi = 0
-    comuni_provincia_Brescia = get_elenco_comuni('Brescia')
-    comuni_provincia_Bergamo = get_elenco_comuni('Bergamo')
-    comuni_provincia = comuni_provincia_Brescia + comuni_provincia_Bergamo
-    st.title("Filtri")
-    
-    tipologia_case = st.selectbox("Tipologia", ("Acquisto Immobile", "Asta Immobiliare"))    
-    elenco = [d['comune'] for d in comuni_provincia]
-    comune_input = st.multiselect(
-        "Comuni",
-        elenco
-    )
-    prezzo_minimo = st.sidebar.slider("Prezzo Minimo (k)", min_value=0, max_value=1000, value=200)
-    prezzo_massimo = st.sidebar.slider("Prezzo Massimo (k)", min_value=0, max_value=1000, value=230)
-    locali = list(range(1, 21))  # Intervallo da 1 a 10
-    locali_range = st.sidebar.select_slider(
-        "Locali",
-        options=locali,
-        value=(locali[2], locali[4])  # Valore iniziale, da 1 a 5 locali
-    )
-    mostra_grafici = st.toggle("Mostra grafici", value = True)
-    
-    modello_ai = st.selectbox("Modello A.I.", ("Llama 3.1", "Mixtral 7x8b"))
-    analisi_ai = st.toggle("Analizza i dati tramite l'A.I.", value = True)
-
-    locali_minimo, locali_massimo = locali_range
-    prezzo_minimo = prezzo_minimo*1000
-    prezzo_massimo = prezzo_massimo*1000
-    cerca_premuto = st.button("Cerca", use_container_width=True, type='primary')
-
-def scrivi_dataframe(output, riepilogo):
+def scrivi_dataframe(output, riepilogo, comune):
     if len(output) > 0: 
         st.numero_immobili_validi = st.numero_immobili_validi + 1
         if not riepilogo:
-            st.write(f"### {comune_provincia['comune']}")
+            st.write(f"### {comune}")
         df = pd.DataFrame(output)
         df_originale = df.sort_values(by=["Vantaggio", "PrezzoMq"], ascending=[False, True])
         if not riepilogo:
@@ -189,7 +147,7 @@ def scrivi_dataframe(output, riepilogo):
         vantaggioso_count = df_originale["Vantaggioso"].sum()
         total_rows = df_originale.shape[0]
 
-        if mostra_grafici: 
+        if st.mostra_grafici: 
             col1, col2, col3 = st.columns(3)
             with col1: 
                 vantaggioso_count = df_originale["Vantaggioso"].sum()
@@ -238,11 +196,12 @@ def scrivi_dataframe(output, riepilogo):
                         "Locali": "Locali",
                         "Link": st.column_config.LinkColumn("App URL")
                     })    
-        st.write(f"Riepilogando nel comune {comune_provincia['comune']} sono presenti **{vantaggioso_count} Immobili vantaggiosi** rispetto ai {total_rows} totali")
+        st.write(f"Riepilogando nel comune {comune} sono presenti **{vantaggioso_count} Immobili vantaggiosi** rispetto ai {total_rows} totali")
         st.divider()
 
-def  analizza_dati_ai(output, tipologia):
-    client = Groq(api_key=API_GROQ)
+def analizza_dati_ai(output, tipologia):
+    Ak = "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"
+    client = Groq(api_key=Ak)
     if tipologia == 1:
         prompt_messages = [
             {
@@ -304,34 +263,114 @@ def  analizza_dati_ai(output, tipologia):
             response_area.markdown(response_text + "▌")
         response_area.markdown(response_text)
 
-if cerca_premuto:
-    if len(comune_input)>0:
-        comuni_selezionati = comune_input
-        comuni_selezionati = [comune.upper() for comune in comuni_selezionati]
-        output = []
-        output_singolo = []
-        for comune_provincia in comuni_provincia:
-            if comune_provincia['comune'].upper() in comuni_selezionati:
-                with st.spinner(f"Ricerca Immobili Comune: {comune_provincia['comune']}"):
-                    output_singolo = json.loads(scrape_immobiliare(comune_provincia['provincia'], 
-                                                                    comune_provincia['comune'], 
-                                                                    comune_provincia['prezzo'], 
-                                                                    prezzo_minimo, 
-                                                                    prezzo_massimo, 
-                                                                    locali_minimo, 
-                                                                    locali_massimo))
-                    scrivi_dataframe(output_singolo, False)
-                    output += output_singolo
-        if len(comuni_selezionati)>1 and st.numero_immobili_validi>1:
-            st.write(f"### Comuni Selezionati")
-            scrivi_dataframe(output, True)
-        if st.numero_immobili_validi > 0 and analisi_ai:  
-            st.title("✨ Analisi Intelligenza Artificiale")
-            st.write("### Considerazioni")
-            analizza_dati_ai(output, 1)
-            #st.divider()
-            #st.write("### Dettagliata")
-            #analizza_dati_ai(output, 2)
-        st.success("Elaborazione Completata")
-    else:
-        st.error("Per favore, inserisci il nome di un comune.")
+def inizializza():
+    if 'logged_in' not in st.session_state:
+        st.session_state['logged_in'] = False
+    if "numero_immobili_validi" not in st.session_state:
+        st.numero_immobili_validi = 0
+    if 'tipologia_case' not in st.session_state:
+        st.session_state['tipologia_case'] = ''
+    if 'mostra_grafici' not in st.session_state:
+        st.session_state['mostra_grafici'] = False
+    if 'analisi_ai' not in st.session_state:
+        st.session_state['analisi_ai'] = False
+    if 'modello_ai' not in st.session_state:
+        st.session_state['modello_ai'] = ''
+
+def main(): 
+    st.title('🏠 Immobiliare A.I. ')
+    st.write("##### Il tuo assistente di intelligenza artificiale per la ricerca di occasioni immobiliari")
+    with st.expander("Informazioni"):
+        st.write("Immobiliare A.I. è la webapp che semplifica la ricerca di immobili, grazie a algoritmi avanzati che calcolano il vantaggio di ogni offerta. Trova le migliori occasioni sul mercato con analisi precise e personalizzate. Scopri l’immobile giusto per te con facilità e sicurezza!")
+
+    with st.sidebar: 
+        comuni_provincia_Brescia = get_elenco_comuni('Brescia')
+        comuni_provincia_Bergamo = get_elenco_comuni('Bergamo')
+        comuni_provincia = comuni_provincia_Brescia + comuni_provincia_Bergamo
+        st.title("Filtri")
+        
+        st.tipologia_case = st.selectbox("Tipologia", ("Acquisto Immobile", "Asta Immobiliare"))    
+        elenco = [d['comune'] for d in comuni_provincia]
+        comune_input = st.multiselect(
+            "Comuni",
+            elenco
+        )
+        prezzo_minimo = st.sidebar.slider("Prezzo Minimo (k)", min_value=0, max_value=1000, value=200)
+        prezzo_massimo = st.sidebar.slider("Prezzo Massimo (k)", min_value=0, max_value=1000, value=230)
+        locali = list(range(1, 21))  # Intervallo da 1 a 10
+        locali_range = st.sidebar.select_slider(
+            "Locali",
+            options=locali,
+            value=(locali[2], locali[4])  # Valore iniziale, da 1 a 5 locali
+        )
+        st.mostra_grafici = st.toggle("Mostra grafici", value = True)
+        
+        st.modello_ai = st.selectbox("Modello A.I.", ("Llama 3.1", "Mixtral 7x8b"))
+        st.analisi_ai = st.toggle("Analizza i dati tramite l'A.I.", value = True)
+
+        locali_minimo, locali_massimo = locali_range
+        prezzo_minimo = prezzo_minimo*1000
+        prezzo_massimo = prezzo_massimo*1000
+        cerca_premuto = st.button("Cerca", use_container_width=True, type='primary')
+
+    if cerca_premuto:
+        if len(comune_input)>0:
+            comuni_selezionati = comune_input
+            comuni_selezionati = [comune.upper() for comune in comuni_selezionati]
+            output = []
+            output_singolo = []
+            for comune_provincia in comuni_provincia:
+                if comune_provincia['comune'].upper() in comuni_selezionati:
+                    with st.spinner(f"Ricerca Immobili Comune: {comune_provincia['comune']}"):
+                        output_singolo = json.loads(scrape_immobiliare(comune_provincia['provincia'], 
+                                                                        comune_provincia['comune'], 
+                                                                        comune_provincia['prezzo'], 
+                                                                        prezzo_minimo, 
+                                                                        prezzo_massimo, 
+                                                                        locali_minimo, 
+                                                                        locali_massimo))
+                        scrivi_dataframe(output_singolo, False, comune_provincia['comune'])
+                        output += output_singolo
+            if len(comuni_selezionati)>1 and st.numero_immobili_validi>1:
+                st.write(f"### Comuni Selezionati")
+                scrivi_dataframe(output, True, '')
+            if st.numero_immobili_validi > 0 and st.analisi_ai:  
+                st.title("✨ Analisi Intelligenza Artificiale")
+                st.write("### Considerazioni")
+                analizza_dati_ai(output, 1)
+                #st.divider()
+                #st.write("### Dettagliata")
+                #analizza_dati_ai(output, 2)
+            st.success("Elaborazione Completata")
+        else:
+            st.error("Per favore, inserisci il nome di un comune.")
+
+def login_page():
+    login_container = st.empty()
+    with login_container: 
+        log1, log2, log3 = st.columns([1.5, 3, 1.5])
+        with log2:
+            st.title('🏠 Immobiliare A.I. ')
+            st.write("")
+            USERNAME = st.secrets["login"]["username"]
+            PASSWORD = st.secrets["login"]["password"]
+
+            with st.form(key='login_form'):
+                username_input = st.text_input('Username')
+                password_input = st.text_input('Password', type='password')
+                login_button = st.form_submit_button('Login')
+
+            if login_button:
+                if username_input == USERNAME and password_input == PASSWORD:
+                    st.session_state['logged_in'] = True
+                    login_container.empty()
+                else:
+                    st.error("Username o password errati")
+
+if __name__ == '__main__':
+    inizializza()
+    st.set_page_config(layout='wide')
+    if not st.session_state['logged_in']:
+        login_page()
+    if st.session_state['logged_in']:    
+        main()