Update app.py

app.py

@@ -44,7 +44,7 @@ def chat_mode(query: str):
     messages.append(response.content)
     if len(messages) >= 6:
         messages = messages[-6:]
-    return "IT-Assistant: " + response.content
+    return f"You: {query}\nIT-Assistant: {response.content}"
 
 def web_search(query: str):
     similar_docs = db.similarity_search(query, k=3)
@@ -67,7 +67,7 @@ def web_search(query: str):
     messages.append(response.content)
     if len(messages) >= 6:
         messages = messages[-6:]
-    return "IT-Assistant: " + response.content
+    return f"You: {query}\nIT-Assistant: {response.content}"
 
 demo = gr.Interface(
     fn=handle_query,