Update app.py

app.py

@@ -125,6 +125,7 @@ def rag(input_text, history, jezik, file):
         return history.append(
             ChatMessage(role="assistant",
                     content=query_engine.query(input_text)
+            )
         )
 #        collection.add(
 #            documents=documents,
@@ -155,6 +156,7 @@ def rag(input_text, history, jezik, file):
         return history.append(
             ChatMessage(role="assistant",
                     content=chat_engine.chat(input_text)}
+            )
         )
 # Interface
 # gr.Textbox(label="Pitanje:", lines=6),