Update app.py

app.py

@@ -11,6 +11,10 @@ from llama_index.llms.huggingface import (
     HuggingFaceLLM,
 )
 from huggingface_hub import login
+import chromadb as chromadb
+from chromadb.utils import embedding_functions
+#
+
 CHROMA_DATA_PATH = "chroma_data/"
 EMBED_MODEL = "BAAI/bge-m3"
 # all-MiniLM-L6-v2
@@ -46,11 +50,7 @@ HF_TOKEN = "wncSKewozDfuZCXCyFbYbAMHgUrfcrumkc"
 #
 login(token=("hf_" + HF_TOKEN))
 
-#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_propmpt = "You are a friendly Chatbot."
+system_propmpt = system_sr
 
 # "facebook/blenderbot-400M-distill",  facebook/blenderbot-400M-distill, stabilityai/stablelm-zephyr-3b, BAAI/bge-small-en-v1.5
 Settings.llm = HuggingFaceInferenceAPI(model_name="mistralai/Mistral-Nemo-Instruct-2407",
@@ -79,9 +79,9 @@ def rag(input_text, file):
         input_text
     )
 
-iface = gr.Interface(fn=rag, inputs=[gr.Textbox(label="Question", lines=6), gr.File()], 
-                     outputs=[gr.Textbox(label="Result", lines=6)], 
-                     title="Answer my question",
+iface = gr.Interface(fn=rag, inputs=[gr.Textbox(label="Pitanje:", lines=6), gr.File()], 
+                     outputs=[gr.Textbox(label="Odgovor:", lines=6)], 
+                     title="Kako Vam mogu pomoći?",
                      description= "UChat"
                     )
 iface.launch()