Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -126,35 +126,35 @@ def rag(input_text, history, jezik):
|
|
126 |
## query_engine = index2.as_query_engine()
|
127 |
# return query_engine.query(input_text)
|
128 |
# return history.append({"role": "assistant", "content": query_engine.query(input_text)})
|
129 |
-
return history + [[input_text, query_engine.query(input_text)]]
|
130 |
# collection.add(
|
131 |
# documents=documents,
|
132 |
# ids=[f"id{last+i}" for i in range(len(documents))],
|
133 |
# metadatas=[{"state": "s0", "next": "s0", "used": False, "source": 'None', "page": -1, "lang": jezik } for i in range(len(documents)) ]
|
134 |
# )
|
135 |
## else:
|
136 |
-
|
137 |
-
|
138 |
-
|
139 |
-
|
140 |
-
|
141 |
-
|
142 |
-
|
143 |
-
|
144 |
-
|
145 |
-
|
146 |
-
|
147 |
-
|
148 |
-
|
149 |
-
|
150 |
-
|
151 |
-
|
152 |
-
|
153 |
# if (o_jezik!='N/A'):
|
154 |
# input_text += " - odgovori " + o_jezik + "."
|
155 |
# return query_engine.query(input_text)
|
156 |
-
|
157 |
-
|
158 |
# Interface
|
159 |
# gr.Textbox(label="Pitanje:", lines=6),
|
160 |
# outputs=[gr.Textbox(label="Odgovor:", lines=6)],
|
|
|
126 |
## query_engine = index2.as_query_engine()
|
127 |
# return query_engine.query(input_text)
|
128 |
# return history.append({"role": "assistant", "content": query_engine.query(input_text)})
|
129 |
+
## return history + [[input_text, query_engine.query(input_text)]]
|
130 |
# collection.add(
|
131 |
# documents=documents,
|
132 |
# ids=[f"id{last+i}" for i in range(len(documents))],
|
133 |
# metadatas=[{"state": "s0", "next": "s0", "used": False, "source": 'None', "page": -1, "lang": jezik } for i in range(len(documents)) ]
|
134 |
# )
|
135 |
## else:
|
136 |
+
o_jezik = "N/A"
|
137 |
+
match jezik:
|
138 |
+
case 'hrvatski':
|
139 |
+
o_jezik = 'na hrvatskom jeziku'
|
140 |
+
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://tele mach.hr/podrska>." + "Odgovaraj " + o_jezik
|
141 |
+
case 'slovenski':
|
142 |
+
o_jezik = 'v slovenščini'
|
143 |
+
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>. " + "Odgovor " + o_jezik
|
144 |
+
case 'srpski':
|
145 |
+
o_jezik = 'na srpskom jeziku'
|
146 |
+
Settings.llm.system_prompt = system_sr + "Call centar telefon je 19900 za sve korisnike. Stranica podrške je <https://sbb.rs/podrska/>. " + "Odgovaraj " + o_jezik
|
147 |
+
case 'makedonski':
|
148 |
+
o_jezik = 'на македонски јазикот'
|
149 |
+
Settings.llm.system_prompt = system_sr + "Stranica podrške je https://mn.nettvplus.com/me/podrska/ za NetTV. " + "Oдговори " + o_jezik
|
150 |
+
case 'Eksperimentalna opcija':
|
151 |
+
o_jezik = 'N/A'
|
152 |
+
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. "
|
153 |
# if (o_jezik!='N/A'):
|
154 |
# input_text += " - odgovori " + o_jezik + "."
|
155 |
# return query_engine.query(input_text)
|
156 |
+
response = chat_engine.chat(input_text)
|
157 |
+
return history + [[input_text, response]]
|
158 |
# Interface
|
159 |
# gr.Textbox(label="Pitanje:", lines=6),
|
160 |
# outputs=[gr.Textbox(label="Odgovor:", lines=6)],
|