Update app.py

app.py

@@ -120,7 +120,11 @@ def rag(input_text, history, jezik, file):
         index2 = VectorStoreIndex.from_documents(documents)
         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({"role": "assistant", "content": query_engine.query(input_text)})
+        return history.append(
+            ChatMessage(role="assistant",
+                    content=query_engine.query(input_text)
+        )
 #        collection.add(
 #            documents=documents,
 #        ids=[f"id{last+i}" for i in range(len(documents))],
@@ -147,7 +151,10 @@ def rag(input_text, history, jezik, file):
 #        if (o_jezik!='N/A'):
 #            input_text += " - odgovori " + o_jezik + "."
 #        return query_engine.query(input_text)
-        return history.append({"role": "assistant", "content": chat_engine.chat(input_text)})
+        return history.append(
+            ChatMessage(role="assistant",
+                    content=chat_engine.chat(input_text)}
+        )
 # Interface
 # gr.Textbox(label="Pitanje:", lines=6),
 # outputs=[gr.Textbox(label="Odgovor:", lines=6)],