Update app.py

app.py

@@ -74,7 +74,6 @@ option_leggicalendar = {'Leggi Calendar': {'systemRole': 'Tu sei BONSI AI, il mi
                        'tipo': 'CALENDAR', 
                        'RAG': False}
                         }
-
 # ----------------------------------------------------------- Interfaccia --------------------------------------------------------------------
 st.set_page_config(page_title="Bonsi A.I.", page_icon="🏫")
 
@@ -118,6 +117,12 @@ def init_state() :
     if "numero_siti" not in st.session_state:
         st.session_state.numero_siti = 3
 
+    if "suddividi_siti" not in st.session_state:
+        st.session_state.suddividi_siti = 1
+    
+    if "cerca_dominio" not in st.session_state:
+        st.session_state.cerca_dominio = False
+
     if "numero_generazioni" not in st.session_state:
         st.session_state.numero_generazioni = 1
 
@@ -268,12 +273,14 @@ def sidebar():
         st.session_state.cerca_online = st.toggle("Attivata", value=False)
         with st.popover("Siti Specifici", disabled=not st.session_state.cerca_online,use_container_width=True):
             st.markdown("#### Inserisci Siti Web ")
+            st.session_state.cerca_dominio = st.toggle("Cerca nel dominio", value=False, help="Se disattivo, cerca SOLO all'interno dell'url per esempio se digito www.ansa.it, cerca solo nella HOME PAGE di ansa, altrimenti cerca all'interno di tutto il DOMINIO ansa.it")
             for i in range(5):
-                st.session_state.urls[i] = st.text_input(f"URL Sito {i+1}", placeholder='Sito Web...', help='è possibile specificare anche il link di un video Youtube, in tal caso verrà restituita la trascrizione del video')        
-        st.session_state.selected_tbs = st.selectbox("Periodo:", list(st.session_state.tbs_options.keys()), disabled=(not st.session_state.cerca_online) or (st.session_state.urls[0]!=""))
+                if (i<2) or ((i>1) and (st.session_state.urls[i-2] != "")):
+                    st.session_state.urls[i] = st.text_input(f"URL Sito {i+1}", placeholder='Sito Web...', help='è possibile specificare anche il link di un video Youtube, in tal caso verrà restituita la trascrizione del video')        
+        st.session_state.selected_tbs = st.selectbox("Periodo:", list(st.session_state.tbs_options.keys()), disabled=(not st.session_state.cerca_online) or (st.session_state.urls[0]!="" and not st.session_state.cerca_dominio))
         st.session_state.tbs_value = st.session_state.tbs_options[st.session_state.selected_tbs]
-        st.session_state.numero_siti = st.slider(label="Risultati", min_value = 1, max_value=20, value=3, disabled=(not st.session_state.cerca_online) or (st.session_state.urls[0]!=""))
-        #st.session_state.suddividi_ricerca = st.toggle("Attivata", value=False)
+        st.session_state.numero_siti = st.slider(label="Risultati", min_value = 1, max_value=20, value=3, disabled=(not st.session_state.cerca_online) or (st.session_state.urls[0]!="" and not st.session_state.cerca_dominio))
+        st.session_state.suddividi_siti = st.slider(label="Scomponi Ricerca", min_value = 1, max_value=10, value=1, disabled=(not st.session_state.cerca_online) or (st.session_state.urls[0]!="" and not st.session_state.cerca_dominio))
         st.markdown("---")
     
     def model_settings():
@@ -384,10 +391,14 @@ def get_search_results(query, top_k):
     results = []
     if st.session_state.urls[0] != "":
         for i in range(5):
-            url = st.session_state.urls[i]
-            if url != "": 
-                results.append(get_search_results_int(url))    
-    else:
+            dominio = st.session_state.urls[i]
+            if dominio != "": 
+                if st.session_state.cerca_dominio: 
+                    for url in search(query + ' site:' + dominio, num=top_k, stop=top_k, tbs=st.session_state.tbs_value):
+                        results.append(get_search_results_int(url))
+                else:
+                    results.append(get_search_results_int(dominio))    
+    else:    
         for url in search(query, num=top_k, stop=top_k, tbs=st.session_state.tbs_value):
             results.append(get_search_results_int(url))
     return results
@@ -395,10 +406,28 @@ def get_search_results(query, top_k):
 def gen_online_prompt(prompt, top_k) :   
     links = []
     context = ''
-    results = get_search_results(prompt, top_k)
-    for i, result in enumerate(results, start=1):
-        context += result['title'] + '\n' +  result['description'] + '\n' + '\n\n' + result['body'].replace('\n','.') + '\n\n------------------------------------------------------------'
-        links.append((str(i) + '. ' + result['title'], result['description'] + '\n\n' + result['url']))
+    if st.session_state.suddividi_siti > 1: 
+        prompt_suddivisione = f'''Genera un elenco PUNTATO di {st.session_state.suddividi_siti} PAROLE CHIAVE in ITALIANO ottimizzate per il WEB e l'indicizzazione SEO. 
+        Attenzione, restituisci solo le KEYWORD e non scrivere nulla prima o dopo! 
+        Genera le PAROLE CHIAVE a partire da questo argomento, mi raccomando devono essere un elenco PUNTATO in ITALIANO: "{prompt}" '''
+        chat_stream = generate_chat_stream(prompt_suddivisione)
+        risposta = stream_handler(chat_stream, None)
+        sottoricerche = risposta.strip().split('\n')
+        for sottoricerca in sottoricerche:
+            with st.spinner('Sottoricerca: ' + sottoricerca):
+                results = get_search_results(sottoricerca, top_k)
+                for i, result in enumerate(results, start=1):
+                    st.toast(result['title'], icon = '🗞️'  )
+                    context += result['title'] + '\n' +  result['description'] + '\n' + '\n\n' + result['body'].replace('\n','.') + '\n\n------------------------------------------------------------'
+                    links.append((str(i) + '. ' + result['title'], result['description'] + '\n\n' + result['url']))
+    else:            
+        results = get_search_results(prompt, top_k)
+        for i, result in enumerate(results, start=1):
+            st.toast(result['title'])
+            context += result['title'] + '\n' +  result['description'] + '\n' + '\n\n' + result['body'].replace('\n','.') + '\n\n------------------------------------------------------------'
+            links.append((str(i) + '. ' + result['title'], result['description'] + '\n\n' + result['url']))
+    if context == '':
+        context = "Non sono state trovate informazioni sul Web in merito all'argomento specificato. Riprova con un altra ricerca o modifica i parametri"
     return context, links
 
 def generate_chat_stream(prompt):
@@ -429,7 +458,7 @@ def inserisci_istruzioni(prompt_originale):
     #st.session_state.instruction = instruction_originale + '\n----------------------------------------------\n' + st.session_state.instruction
     return links
 
-def stream_handler(chat_stream, placeholder) :
+def stream_handler(chat_stream, placeholder):
     full_response = ''
     for chunk in chat_stream :
         if CHAT_BOTS[st.session_state.chat_bot]["model"][:3] == 'gpt':
@@ -438,8 +467,10 @@ def stream_handler(chat_stream, placeholder) :
         else:
             if chunk.token.text!='</s>' :
                 full_response += chunk.token.text
-        placeholder.markdown(full_response + "▌")
-    placeholder.markdown(full_response)
+        if placeholder:
+            placeholder.markdown(full_response + "▌")
+    if placeholder:
+        placeholder.markdown(full_response)
     return full_response
 
 def show_source(links) :
@@ -510,7 +541,7 @@ if st.session_state.login_effettuato or not EFFETTUA_LOGIN_GOOGLE:
                         if CHAT_BOTS[st.session_state.chat_bot]["model"][:3] == 'gpt':
                             st.error('Inserisci una CHIAVE API valida e controlla il CREDITO residuo: https://platform.openai.com/api-keys')  
                         else:
-                            st.error('Modello AI attualmente non disponibile')    
+                            st.error('Errore nella generazione riprova o utilizza un altro modello AI')    
 
         if errore_generazione == False: 
             st.session_state.ultimo_messaggio = full_response