Update app.py

app.py

@@ -18,6 +18,11 @@ import spaces
 import threading
 import sys
 import torch
+from llama_index.core.vector_stores import (
+    MetadataFilter,
+    MetadataFilters,
+    FilterOperator,
+)
 
 
 head = """
@@ -119,6 +124,7 @@ file_path=""
 num_responses=0
 current_chat_mode="STANDARD"
 retriever=None
+index = None
 token_count_bandi=0
 token_count_bandi_sistema_puglia=0
 chat_engine_bandi=None
@@ -141,7 +147,8 @@ def main():
     def select_initial_collection():
         global current_collection
         global retriever
-
+        global index
+        
         pc = Pinecone(api_key="7e412663-a2dc-44a6-ab57-25dd0bdce226")
         # connect to index
         pinecone_index = pc.Index("indexbandisistemapuglia")
@@ -168,6 +175,7 @@ def main():
     def select_collection(evt: gr.SelectData):
         global current_collection
         global retriever
+        global index
         global chat_engine_bandi
         global chat_engine_bandi_sistema_puglia
         global token_count_bandi
@@ -473,6 +481,66 @@ def main():
             else:
                 print("MODALITA STANDARD")
                 if(str(current_response_mode)=="tree_summarize"):
+                    nome_bando = ""
+                    userMessage = userMessage.lower()
+                    if("diploma professionale" in userMessage):
+                        nome_bando += "Scheda Avviso Pubblico Diploma Professionale 2022.pdf,"
+                    if("red" in userMessage):
+                        nome_bando += "Scheda RED 2020.pdf,"
+                    if("ifts" in userMessage):
+                        nome_bando += "Scheda Avviso Pubblico IFTS_2023.pdf,"
+                    if(("impianti" in userMessage) OR ("idrogeno" in userMessage)):
+                        nome_bando += "Scheda Avviso PNRR - Impianti idrogeno rinnovabile.pdf,"
+                    if("laureati" in userMessage):
+                        nome_bando += "Scheda Pass Laureati 2023.pdf,"
+                    if("nidi" in userMessage):
+                        nome_bando += "Scheda NIDI - Nuove iniziative d'impresa_ Strumento di ingegneria finanziaria.pdf,"
+                    if("microprestito" in userMessage):
+                        nome_bando += "Scheda MicroPrestito della Regione Puglia - edizione 2021.pdf,"
+                    if("gol" in userMessage):
+                        nome_bando += "Scheda Garanzia di occupabilità dei lavoratori - GOL.pdf,
+                    if("edifici pubblici" in userMessage):
+                        nome_bando += "Scheda Efficientamento Energetico Edifici Pubblici.pdf,"
+                    if("innoaid" in userMessage):
+                        nome_bando += "Scheda di sintesi Avviso _INNOAID - RIAPERTURA_.pdf,"
+                    if("tecnonidi" in userMessage):
+                        nome_bando += "Scheda Avviso Tecnonidi - Aiuti alle piccole imprese innovative.pdf,"
+                    if("of" in userMessage):
+                        nome_bando += "Scheda Avviso Pubblico OF a_f_ 2023_2024.pdf,"
+                    if("giardiniere" in userMessage):
+                        nome_bando += "Scheda Avviso Pubblico _Giardiniere d'arte per giardini e parchi storici_.pdf,"
+                    if("punti cardinali" in userMessage):
+                        nome_bando += "Scheda Avviso _Punti Cardinali_ punti di orientamento per la formazione e il lavoro.pdf,"
+                    if("accreditamento servizi" in userMessage):
+                        nome_bando += "Scheda Accreditamento Servizi per il Lavoro.pdf,"
+                    if("accreditamento organismi" in userMessage):
+                        nome_bando += "Scheda Accreditamento Organismi Formativi.pdf,"
+                    if("multimisura POC" in userMessage):
+                        nome_bando += "Avviso Multimisura POC.pdf,"
+                    if("garanzia giovani" in userMessage):
+                        nome_bando += "Avviso Multimisura - Garanzia Giovani II Fase.pdf,"
+                    if("apprendistato professionalizzante" in userMessage):
+                        nome_bando += "Apprendistato Professionalizzante.pdf,"                  
+
+                    # Rimuovi l'ultima virgola
+                    if nome_bando.endswith(","):
+                        nome_bando = nome_bando[:-1]                  
+                    # Crea una lista di bandi separati dalla virgola
+                    lista_bandi = nome_bando.split(",")               
+                    # Crea una lista di oggetti MetadataFilter
+                    filter_list = []
+                    for bando in lista_bandi:
+                        filter_list.append(MetadataFilter(key="nome_bando", value=bando))
+                    
+                    #crea una lista di MetadataFilter
+                    filters = MetadataFilters(
+                        filters=filter_list,
+                        condition=FilterCondition.OR,
+                    )
+
+                    retriever = VectorIndexRetriever(index=index, similarity_top_k=3, vector_store_query_mode="hybrid", embed_model=embed_model, alpha=0.5, filters=filters)
+
+
                     # define response synthesizer
                     response_synthesizer = get_response_synthesizer(streaming=True,response_mode="tree_summarize",text_qa_template=text_qa_template)
                     query_engine = None