Update app.py

app.py

@@ -1,5 +1,8 @@
 import streamlit as st
 from chat_client import chat
+from google_function import leggi_gmail
+from google_function import scrivi_bozza_gmail
+from google_function import leggi_calendario_google
 import time
 import os
 from dotenv import load_dotenv
@@ -16,6 +19,7 @@ import PyPDF2
 import pytesseract
 from PIL import Image
 from youtube_transcript_api import YouTubeTranscriptApi
+import webbrowser
 
 load_dotenv()
 URL_APP_SCRIPT = os.getenv('URL_APP_SCRIPT')
@@ -29,7 +33,18 @@ option_personalizzata = {'Personalizzata': {'systemRole': 'Tu sei BONSI AI, il m
                        'tipo': '', 
                        'RAG': False}
                         }
-
+option_leggiemail = {'Leggi Gmail': {'systemRole': 'Tu sei BONSI AI, il mio assistente personale della scuola superiore del Bonsignori. Effettua l operazione richiesta sulla base delle seguenti email: ', 
+                       'systemStyle': 'Firmati sempre come BONSI AI. (scrivi in italiano)', 
+                       'instruction': '', 
+                       'tipo': 'EMAIL', 
+                       'RAG': False}
+                        }
+option_leggicalendar = {'Leggi Calendar': {'systemRole': 'Tu sei BONSI AI, il mio assistente personale della scuola superiore del Bonsignori. Effettua l operazione richiesta sulla base dei seguenti eventi di calendario: ', 
+                       'systemStyle': 'Firmati sempre come BONSI AI. (scrivi in italiano)', 
+                       'instruction': '', 
+                       'tipo': 'CALENDAR', 
+                       'RAG': False}
+                        }
 # ----------------------------------------------------------- Interfaccia --------------------------------------------------------------------
 st.set_page_config(page_title="Bonsi A.I.", page_icon="🏫")
 
@@ -73,6 +88,9 @@ def init_state() :
     if "numero_generazioni" not in st.session_state:
         st.session_state.numero_generazioni = 1
 
+    if "numero_elementi" not in st.session_state:
+        st.session_state.numero_elementi = 10
+    
     if "testo_documenti" not in st.session_state:
         st.session_state.testo_documenti = ''
 
@@ -101,7 +119,7 @@ def init_state() :
                 options = requests.get(URL_PROMPT).json()
                 st.write("Inizializzazione Documenti")
                 documenti = requests.get(URL_DOCUMENTI).json()
-                st.session_state.options = {**option_personalizzata, **options} 
+                st.session_state.options = {**option_personalizzata, **option_leggiemail, **option_leggicalendar, **options}
                 st.session_state.documenti = documenti
                 st.session_state.loaded_data = True
                 status.update(label="Caricamento Completato", state="complete", expanded=False)
@@ -142,6 +160,10 @@ def sidebar():
         st.session_state.systemRole = st.text_area("Descrizione", st.session_state.systemRole, help='Ruolo del chatbot e descrizione dell\'azione che deve svolgere')
         st.session_state.systemStyle = st.session_state.selected_option.get('systemStyle', '')
         st.session_state.systemStyle = st.text_area("Stile", st.session_state.systemStyle, help='Descrizione dello stile utilizzato per generare il testo')
+        if st.session_state.selected_option["tipo"]=='EMAIL':
+            st.session_state.numero_elementi = st.slider(label="Numero Email", min_value=1, max_value=100, value=10)
+        if st.session_state.selected_option["tipo"]=='CALENDAR':            
+            st.session_state.numero_elementi = st.slider(label="Numero Eventi Calendario", min_value=1, max_value=100, value=10)
         st.session_state.rag_enabled = st.session_state.selected_option.get('tipo', '')=='RAG'
         if st.session_state.selected_option_key == 'Decreti':
             st.session_state.top_k = st.slider(label="Documenti da ricercare", min_value=1, max_value=20, value=4, disabled=not st.session_state.rag_enabled)
@@ -322,6 +344,14 @@ def inserisci_istruzioni(prompt_originale):
         with st.spinner("Ricerca nei Decreti...."):
             time.sleep(1)
             st.session_state.instruction, links = gen_augmented_prompt(prompt=prompt_originale, top_k=st.session_state.top_k)        
+    if st.session_state.selected_option["tipo"]=='EMAIL':
+        with st.spinner("Ricerca nelle Email...."):
+            time.sleep(1)
+            st.session_state.instruction, links = leggi_gmail(max_results=st.session_state.numero_elementi)
+    if st.session_state.selected_option["tipo"]=='CALENDAR':
+        with st.spinner("Ricerca nel Calendario...."):
+            time.sleep(1)
+            st.session_state.instruction, links = leggi_calendario_google(max_results=st.session_state.numero_elementi)
     with st.spinner("Generazione in corso...") :
         time.sleep(1)
     #st.session_state.instruction = instruction_originale + '\n----------------------------------------------\n' + st.session_state.instruction
@@ -381,7 +411,7 @@ if prompt := st.chat_input("Chatta con BonsiAI..."):
                 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:
+                    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") :