shoom013 commited on
Commit
13939ef
1 Parent(s): 0c3d0b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -20
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
- 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)],
 
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)],