Update app.py

app.py

@@ -40,7 +40,7 @@ jezik = "srpski"
 
 system_sr = "Zoveš se U-Chat AI asistent i pomažeš korisniku usluga kompanije United Group. Korisnik postavlja pitanje ili problem, upareno sa dodatnima saznanjima. Na osnovu toga napiši korisniku kratak i ljubazan odgovor koji kompletira njegov zahtev ili mu daje odgovor na pitanje. "
 # " Ako ne znaš odgovor, reci da ne znaš, ne izmišljaj ga."
-system_sr += "Usluge kompanije United Group uključuju i kablovsku mrežu za digitalnu televiziju, pristup internetu, uređaj EON SMART BOX za TV sadržaj, kao i fiksnu telefoniju."
+system_sr += "Usluge kompanije United Group uključuju i kablovsku mrežu za digitalnu televiziju, pristup internetu, uređaj EON SMART BOX za TV sadržaj, kao i fiksnu telefoniju. "
 
 chroma_client = chromadb.PersistentClient(CHROMA_DATA_PATH)
 embedding_func = embedding_functions.SentenceTransformerEmbeddingFunction(
@@ -91,7 +91,7 @@ def upload_file(filepath):
     query_engine = index.as_query_engine()
     return filepath
 
-def rag(input_text, file):
+def rag(input_text, jezik, file):
     if (file):
 #        documents = []
 #        for f in file:
@@ -112,10 +112,31 @@ def rag(input_text, file):
 #        metadatas=[{"state": "s0", "next": "s0", "used": False, "source": 'None', "page": -1, "lang": jezik } for i in range(len(documents)) ]
 #    )
     else:
+        o_jezik = "N/A"
+        match jezik:
+            case 'hrvatski':
+                o_jezik = 'na hrvatskom jeziku'
+                Settings.llm.system_prompt = system_sr + "Call centar telefon je 095 1000 444 za privatne i 095 1000 500 za poslovne korisnike. Stranica podrške je https://telemach.hr/podrska"
+            case 'slovenački':
+                o_jezik = 'v slovenščini'
+                Settings.llm.system_prompt = system_sr + "Call centar i pomoč za fizične uporabnike: 070 700 700.stran za podporo je https://telemach.si/pomoc"
+            case 'srpski':
+                o_jezik = 'na srpskom jeziku'
+                Settings.llm.system_prompt = system_sr + "Call centar telefon je 19900 za sve korisnike. Stranica podrške je https://sbb.rs/podrska/"
+            case 'makedonski':
+                o_jezik = 'на македонски јазикот'
+                Settings.llm.system_prompt = system_sr + "Stranica podrške je https://mn.nettvplus.com/me/podrska/ za NetTV"
+            case 'N/A':
+                o_jezik = 'N/A'
+                Settings.llm.system_prompt = system_sr + "Call centar telefon je 12755 za Crnu Goru, 0800 31111 za BIH, 070 700 700 u Sloveniji, 19900 u Srbiji, 095 1000 444 za hrvatske korisnike."
+        if (jezik!='N/A'):
+            input_text += " - odgovori " + o_jezik + "."
         query_engine = index.as_query_engine()
         return query_engine.query(input_text)
 
-iface = gr.Interface(fn=rag, inputs=[gr.Textbox(label="Pitanje:", lines=6), gr.File()], 
+iface = gr.Interface(fn=rag, inputs=[gr.Textbox(label="Pitanje:", lines=6),
+                                     gr.Dropdown(["slovenski", "hrvatski", "srpski", "makedonski", "Eksperimentalna opcija"], value="srpski", label="Jezik", info="N/A"),
+                                     gr.File()], 
                      outputs=[gr.Textbox(label="Odgovor:", lines=6)], 
                      title="Kako Vam mogu pomoći?",
                      description= "UChat"