Update app.py

app.py

@@ -126,14 +126,7 @@ def rag(input_text, history, jezik):
         query_engine = index2.as_query_engine()
 #        return query_engine.query(input_text)
 #        return history.append({"role": "assistant", "content": query_engine.query(input_text)})
-        return [history.append(
-            ChatMessage(role="user",
-                        content=input_text
-            ).append(
-                        role="assistant",
-                        content=query_engine.query(input_text)
-            )
-        )]
+        return [input_text query_engine.query(input_text)]
 #        collection.add(
 #            documents=documents,
 #        ids=[f"id{last+i}" for i in range(len(documents))],
@@ -160,11 +153,7 @@ def rag(input_text, history, jezik):
 #        if (o_jezik!='N/A'):
 #            input_text += " - odgovori " + o_jezik + "."
 #        return query_engine.query(input_text)
-        return [history.append(
-            ChatMessage(role="assistant",
-                    content=chat_engine.chat(input_text)
-            )
-        )]
+        return [input_text chat_engine.chat(input_text)]
 # Interface
 # gr.Textbox(label="Pitanje:", lines=6),
 # outputs=[gr.Textbox(label="Odgovor:", lines=6)], 
@@ -172,7 +161,7 @@ def rag(input_text, history, jezik):
 iface = gr.ChatInterface(rag,        
         title="UChat",
         description="Postavite pitanje ili opišite problem koji imate",
-        chatbot=gr.Chatbot(placeholder="Kako Vam mogu pomoći?", type="messages", label="UChat", height=300),
+        chatbot=gr.Chatbot(placeholder="Kako Vam mogu pomoći?", type="tuples", label="UChat", height=300),
         textbox=gr.Textbox(placeholder="Pitanje ili opis problema", container=False, scale=7),
         theme="soft",
 #        examples=["Ne radi mi internet", "Koje usluge imam na raspologanju?", "Ne radi mi daljinski upravljač, šta da radim?"],