Update app.py

app.py

@@ -119,7 +119,8 @@ def rag(input_text, history, jezik, file):
         documents = SimpleDirectoryReader(file.name).load_data()
         index2 = VectorStoreIndex.from_documents(documents)
         query_engine = index2.as_query_engine()
-        return query_engine.query(input_text)
+#        return query_engine.query(input_text)
+        return history.append({"role": "assistant", "content": query_engine.query(input_text)})
 #        collection.add(
 #            documents=documents,
 #        ids=[f"id{last+i}" for i in range(len(documents))],
@@ -146,14 +147,14 @@ def rag(input_text, history, jezik, file):
 #        if (o_jezik!='N/A'):
 #            input_text += " - odgovori " + o_jezik + "."
 #        return query_engine.query(input_text)
-        return chat_engine.chat(input_text)
+        return history.append({"role": "assistant", "content": chat_engine.chat(input_text)})
 # Interface
 # gr.Textbox(label="Pitanje:", lines=6),
 # outputs=[gr.Textbox(label="Odgovor:", lines=6)], 
 iface = gr.ChatInterface(rag,        
-        title="Kako Vam mogu pomoći?",
+        title="UChat",
         description="Postavite pitanje ili opišite problem koji imate",
-        chatbot=gr.Chatbot(height=300),
+        chatbot=gr.Chatbot("Kako Vam mogu pomoći?", type='messages', 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?"],