Update app.py

app.py

@@ -4,6 +4,7 @@ from google_function import leggi_gmail
 from google_function import scrivi_bozza_gmail
 from google_function import leggi_calendario_google
 from google_function import connetti_google
+from google_function import crea_documento_google
 import time
 import os
 from dotenv import load_dotenv
@@ -22,13 +23,15 @@ from PIL import Image
 from youtube_transcript_api import YouTubeTranscriptApi
 import webbrowser
 from streamlit_javascript import st_javascript
+import datetime
 
 load_dotenv()
 URL_APP_SCRIPT = os.getenv('URL_APP_SCRIPT')
 URL_PROMPT = URL_APP_SCRIPT + '?IdFoglio=1cLw9q70BsPmxMBj9PIzgXtq6sm3X-GVBVnOB5wE8jr8'
 URL_DOCUMENTI = URL_APP_SCRIPT + '?IdSecondoFoglio=1cLw9q70BsPmxMBj9PIzgXtq6sm3X-GVBVnOB5wE8jr8'
 SYSTEM_PROMPT = ["Sei BonsiAI e mi aiuterai nelle mie richieste (Parla in ITALIANO)", "Esatto, sono BonsiAI. Di cosa hai bisogno?"]
-CHAT_BOTS = {"Mixtral 8x7B v0.1" :"mistralai/Mixtral-8x7B-Instruct-v0.1"}
+CHAT_BOTS = {"Mixtral 8x7B v0.1" :"mistralai/Mixtral-8x7B-Instruct-v0.1", 
+             "Mistral 7B v0.2" :"mistralai/Mistral-7B-Instruct-v0.2"}
 option_personalizzata = {'Personalizzata': {'systemRole': 'Tu sei BONSI AI, il mio assistente personale della scuola superiore del Bonsignori. Aiutami in base alle mie esigenze', 
                        'systemStyle': 'Firmati sempre come BONSI AI. (scrivi in italiano)', 
                        'instruction': '', 
@@ -48,11 +51,6 @@ option_leggicalendar = {'Leggi Calendar': {'systemRole': 'Tu sei BONSI AI, il mi
                        '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="🏫")
 
@@ -97,8 +95,14 @@ def init_state() :
         st.session_state.numero_generazioni = 1
 
     if "numero_elementi" not in st.session_state:
-        st.session_state.numero_elementi = 10
+        st.session_state.numero_elementi = 5
+    
+    if "data_inizio" not in st.session_state:
+        st.session_state.data_inizio = None
     
+    if "data_inizio" not in st.session_state:
+        st.session_state.data_inizio = None
+
     if "testo_documenti" not in st.session_state:
         st.session_state.testo_documenti = ''
 
@@ -113,6 +117,12 @@ def init_state() :
     
     if "login_effettuato" not in st.session_state:
         st.session_state.login_effettuato = False
+        
+    if "ultimo_messaggio" not in st.session_state:
+        st.session_state.ultimo_messaggio = ""
+        
+    if "tutti_messaggi" not in st.session_state:
+        st.session_state.tutti_messaggi = ""
     
     if "tbs_options" not in st.session_state:
         st.session_state.tbs_options = {
@@ -131,9 +141,14 @@ def init_state() :
                 st.write("Inizializzazione Ambiente")
                 time.sleep(1)
                 st.write("Inizializzazione Prompt")
-                options = requests.get(URL_PROMPT).json()
+                URL_REDIRECT = os.getenv('URL_REDIRECT')
+                options = {}
+                if URL_REDIRECT != "http://localhost:8501/":
+                    options = requests.get(URL_PROMPT).json()
                 st.write("Inizializzazione Documenti")
-                documenti = requests.get(URL_DOCUMENTI).json()
+                documenti = {}
+                if URL_REDIRECT != "http://localhost:8501/":
+                    documenti = requests.get(URL_DOCUMENTI).json()
                 st.session_state.options = {**option_personalizzata, **option_leggiemail, **option_leggicalendar, **options}
                 st.session_state.documenti = documenti
                 st.session_state.loaded_data = True
@@ -157,6 +172,24 @@ def read_text_from_file(file):
             st.write(f"Non è possibile leggere il testo dal file '{file.name}'.")
     return text
 
+def open_new_tab(url):
+    placeholder = st.empty()
+    with placeholder:
+        placeholder.empty()
+        new_tab_js = f'''<script type="text/javascript">window.open("{url}", "_blank");</script>'''
+        st.components.v1.html(new_tab_js, height=1)
+        time.sleep(0.3)
+    placeholder.empty()
+    
+def esporta_testo(tipo, ultimo_messaggio):
+    testo = st.session_state.ultimo_messaggio if ultimo_messaggio else st.session_state.tutti_messaggi
+    if tipo == 'Bozza Email': 
+        url = scrivi_bozza_gmail(testo)
+        open_new_tab(url)
+    if tipo == 'Google Documenti': 
+        url = crea_documento_google(testo)
+        open_new_tab(url)
+
 def sidebar():
     def retrieval_settings() :
         st.markdown("# Impostazioni Prompt")
@@ -170,15 +203,18 @@ def sidebar():
             st.session_state.split = st.slider(label="Pagine Suddivisione", min_value=1, max_value=30, value=30, help='Se il documento ha 100 pagine e suddivido per 20 pagine elaborerà la risposta 5 volte. Più alto è il numero e meno volte elaborerà ma la risposta sarà più imprecisa')            
         else:
             st.session_state.instruction = st.session_state.selected_option.get('instruction', '')
-        
         st.session_state.systemRole = st.session_state.selected_option.get('systemRole', '')
         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)
+            st.session_state.data_inizio = st.date_input("Email dal", value=datetime.date.today()-datetime.timedelta(days=7), format='DD/MM/YYYY')
+            st.session_state.data_fine = st.date_input("Email al", value=datetime.date.today(), format='DD/MM/YYYY')
         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.data_inizio = st.date_input("Eventi dal", value=datetime.date.today(), format='DD/MM/YYYY')
+            st.session_state.data_fine = st.date_input("Eventi al", value=datetime.date.today()+datetime.timedelta(days=7), format='DD/MM/YYYY')
         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)
@@ -206,16 +242,26 @@ def sidebar():
         st.markdown("---")
     
     def model_settings():
-        st.markdown("# Impostazioni Modello")
+        st.markdown("# Modello")
         st.session_state.chat_bot = st.sidebar.radio('Modello:', [key for key, value in CHAT_BOTS.items() ])
         st.session_state.numero_generazioni = st.slider(label="Generazioni", min_value = 1, max_value=10, value=1)
         st.session_state.enable_history = st.toggle("Storico Messaggi", value=True)
         st.session_state.temp = st.slider(label="Creatività", min_value=0.0, max_value=1.0, step=0.1, value=0.9)
         st.session_state.max_tokens = st.slider(label="Lunghezza Output", min_value = 2, max_value=2048, step= 32, value=1024)
+        st.markdown("---")
+
+    def export_settings():
+        st.markdown("# Esportazione")
+        st.session_state.export_type =  st.selectbox('Tipologia', ('Non Esportare', 'Google Documenti', 'Bozza Email'), help='Seleziona la tipologia di esportazione del testo generato')
+        st.session_state.export_all  = st.toggle("Considera tutte le chat", value=False)
+        if st.button("Esporta", type="primary", use_container_width=True): 
+            esporta_testo(st.session_state.export_type, st.session_state.export_all)
+        st.markdown("---")
 
     with st.sidebar:
         retrieval_settings()
         model_settings()
+        export_settings()
         st.markdown("""> **Creato da Matteo Bergamelli **""")
 
 def audioRec():
@@ -362,11 +408,11 @@ def inserisci_istruzioni(prompt_originale):
     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)
+            st.session_state.instruction, links = leggi_gmail(max_results=st.session_state.numero_elementi, data_inizio = st.session_state.data_inizio, data_fine = st.session_state.data_fine)
     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)
+            st.session_state.instruction, links = leggi_calendario_google(max_results=st.session_state.numero_elementi, data_inizio = st.session_state.data_inizio, data_fine = st.session_state.data_fine)
     with st.spinner("Generazione in corso...") :
         time.sleep(1)
     #st.session_state.instruction = instruction_originale + '\n----------------------------------------------\n' + st.session_state.instruction
@@ -441,6 +487,8 @@ if st.session_state.login_effettuato == True:
                     st.session_state.messages.append({"role": "assistant", "content": full_response})
                     risposta_completa = risposta_completa + '\n' + full_response
     
+        st.session_state.ultimo_messaggio = full_response
+        st.session_state.tutti_messaggi += '\n\n' + full_response
         if st.session_state.enable_history:
             st.session_state.history.append([prompt_originale, full_response])
         else: