shoom013 commited on
Commit
fbdfa3d
·
verified ·
1 Parent(s): 4202dae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -126,7 +126,7 @@ 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 [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))],
@@ -153,7 +153,7 @@ def rag(input_text, history, jezik):
153
  # if (o_jezik!='N/A'):
154
  # input_text += " - odgovori " + o_jezik + "."
155
  # return query_engine.query(input_text)
156
- return [input_text chat_engine.chat(input_text)]
157
  # Interface
158
  # gr.Textbox(label="Pitanje:", lines=6),
159
  # 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))],
 
153
  # if (o_jezik!='N/A'):
154
  # input_text += " - odgovori " + o_jezik + "."
155
  # return query_engine.query(input_text)
156
+ return history + [input_text, chat_engine.chat(input_text)]
157
  # Interface
158
  # gr.Textbox(label="Pitanje:", lines=6),
159
  # outputs=[gr.Textbox(label="Odgovor:", lines=6)],