Update app.py

app.py

@@ -20,6 +20,7 @@ import pytesseract
 from PIL import Image
 from youtube_transcript_api import YouTubeTranscriptApi
 import webbrowser
+from streamlit_javascript import st_javascript
 
 load_dotenv()
 URL_APP_SCRIPT = os.getenv('URL_APP_SCRIPT')
@@ -45,6 +46,12 @@ option_leggicalendar = {'Leggi Calendar': {'systemRole': 'Tu sei BONSI AI, il mi
                        'tipo': 'CALENDAR', 
                        'RAG': False}
                         }
+
+def local_storage_get(key):
+    return st_javascript(f"localStorage.getItem('{key}');")
+
+def local_storage_set(key, value):
+    return st_javascript(f"localStorage.setItem('{key}', '{value}');")
 # ----------------------------------------------------------- Interfaccia --------------------------------------------------------------------
 st.set_page_config(page_title="Bonsi A.I.", page_icon="🏫")
 
@@ -100,6 +107,9 @@ def init_state() :
     if "urls" not in st.session_state:
         st.session_state.urls = [""] * 5
     
+    if "login_effettuato" not in st.session_state:
+        st.session_state.login_effettuato = False
+    
     if "tbs_options" not in st.session_state:
         st.session_state.tbs_options = {
             "Sempre": "0",
@@ -109,7 +119,8 @@ def init_state() :
             "Ultimo giorno": "qdr:d"
         }
 
-    if not st.session_state.loaded_data:
+    if not st.session_state.loaded_data and st.session_state.login_effettuato == True:
+        place=st.empty()
         place=st.empty()
         with place:
             with st.status("Caricamento in corso...", expanded=True) as status:
@@ -372,11 +383,6 @@ def show_source(links) :
             reference, testo = link
             st.info('##### ' + reference.replace('_', ' ') + '\n\n'+ testo)
 
-init_state()
-sidebar()
-header()
-chat_box()
-
 def split_text(text, chunk_size):
     testo_suddiviso = []
     if text == '': 
@@ -387,44 +393,53 @@ def split_text(text, chunk_size):
         testo_suddiviso.append(text[i:i+chunk_size])
     return testo_suddiviso
 
-if prompt := st.chat_input("Chatta con BonsiAI..."):
-    prompt_originale = prompt
-    links = inserisci_istruzioni(prompt_originale)
-    st.session_state.instruction+= ' \n\n' + st.session_state.testo_documenti
-    instruction_suddivise = split_text(st.session_state.instruction, st.session_state.split*2000)
-    ruolo_originale = st.session_state.systemRole
-    ruoli_divisi = ruolo_originale.split("&&")
-    parte=1           
-    i=1 
-    risposta_completa = ''
-    for ruolo_singolo in ruoli_divisi:
-        for instruction_singola in instruction_suddivise:
-            for numgen in range(1, st.session_state.numero_generazioni+1):
-                if i==1:
-                    st.chat_message("user").markdown(prompt_originale + (': Parte ' + str(parte) if i > 1 else ''))
-                i+=1   
-                prompt = formattaPrompt(prompt_originale, ruolo_singolo, st.session_state.systemStyle, instruction_singola)
-                print('------------------------------------------------------------------------------------')
-                print(prompt)
-                st.session_state.messages.append({"role": "user", "content": prompt_originale})
-                chat_stream = generate_chat_stream(prompt)
-                with st.chat_message("assistant"):
-                    placeholder = st.empty()
-                    full_response = stream_handler(chat_stream, placeholder)
-                    if st.session_state.rag_enabled or st.session_state.cerca_online or st.session_state.selected_option["tipo"]=='EMAIL' or st.session_state.selected_option["tipo"]=='CALENDAR':
-                        show_source(links)  
-                    if st.session_state.options.get(st.session_state.selected_option_key, {})["tipo"]=='DOCUMENTO':
-                        with st.expander("Mostra Documento") :
-                            st.info('##### ' + st.session_state.selected_documento_key + ' (Parte ' + str(parte) +')'+ '\n\n\n' + instruction_singola)
-                            parte+=1
-                st.session_state.messages.append({"role": "assistant", "content": full_response})
-                risposta_completa = risposta_completa + '\n' + full_response
-
-    if st.session_state.enable_history:
-        st.session_state.history.append([prompt_originale, full_response])
-    else:
-        st.session_state.history.append(['', ''])
-    st.success('Generazione Completata')
-    payload = {"domanda": prompt_originale, "risposta": risposta_completa}
-    json_payload = json.dumps(payload)
-    response = requests.post(URL_APP_SCRIPT, data=json_payload)
+init_state()
+if st.session_state.login_effettuato == False:
+    connetti_google()
+else:
+    init_state()
+    sidebar()
+    header()
+    chat_box()
+    
+    if prompt := st.chat_input("Chatta con BonsiAI..."):
+        prompt_originale = prompt
+        links = inserisci_istruzioni(prompt_originale)
+        st.session_state.instruction+= ' \n\n' + st.session_state.testo_documenti
+        instruction_suddivise = split_text(st.session_state.instruction, st.session_state.split*2000)
+        ruolo_originale = st.session_state.systemRole
+        ruoli_divisi = ruolo_originale.split("&&")
+        parte=1           
+        i=1 
+        risposta_completa = ''
+        for ruolo_singolo in ruoli_divisi:
+            for instruction_singola in instruction_suddivise:
+                for numgen in range(1, st.session_state.numero_generazioni+1):
+                    if i==1:
+                        st.chat_message("user").markdown(prompt_originale + (': Parte ' + str(parte) if i > 1 else ''))
+                    i+=1   
+                    prompt = formattaPrompt(prompt_originale, ruolo_singolo, st.session_state.systemStyle, instruction_singola)
+                    print('------------------------------------------------------------------------------------')
+                    print(prompt)
+                    st.session_state.messages.append({"role": "user", "content": prompt_originale})
+                    chat_stream = generate_chat_stream(prompt)
+                    with st.chat_message("assistant"):
+                        placeholder = st.empty()
+                        full_response = stream_handler(chat_stream, placeholder)
+                        if st.session_state.rag_enabled or st.session_state.cerca_online or st.session_state.selected_option["tipo"]=='EMAIL' or st.session_state.selected_option["tipo"]=='CALENDAR':
+                            show_source(links)  
+                        if st.session_state.options.get(st.session_state.selected_option_key, {})["tipo"]=='DOCUMENTO':
+                            with st.expander("Mostra Documento") :
+                                st.info('##### ' + st.session_state.selected_documento_key + ' (Parte ' + str(parte) +')'+ '\n\n\n' + instruction_singola)
+                                parte+=1
+                    st.session_state.messages.append({"role": "assistant", "content": full_response})
+                    risposta_completa = risposta_completa + '\n' + full_response
+    
+        if st.session_state.enable_history:
+            st.session_state.history.append([prompt_originale, full_response])
+        else:
+            st.session_state.history.append(['', ''])
+        st.success('Generazione Completata')
+        payload = {"domanda": prompt_originale, "risposta": risposta_completa}
+        json_payload = json.dumps(payload)
+        response = requests.post(URL_APP_SCRIPT, data=json_payload)