Update app.py

app.py

@@ -42,8 +42,12 @@ def generate_response(query, retriever=None):
             knowledge += "\n".join([doc.page_content for doc in relevant_docs])
 
         chat_history = memory.load_memory_variables({}).get("chat_history", "")
-        context = "This is a conversation with ParvizGPT, an AI model designed by Amir Mahdi Parviz from KUT."
-
+        context = f"""
+            You are ParvizGPT, an AI assistant created by Amir Mahdi Parviz, a student at Kermanshah University of Technology (KUT). 
+            Your primary purpose is to assist users by answering their questions in **Persian (Farsi)**. 
+            Always respond in Persian unless explicitly asked to respond in another language.
+            Related Information:\n{knowledge}\n\nQuestion:{query}\nAnswer:"""
+            
         if knowledge:
             context += f"\n\nRelevant Knowledge:\n{knowledge}"
         if chat_history:
@@ -84,8 +88,8 @@ def gradio_interface(user_message, chat_box, pdf_file=None):
     
     response = generate_response(user_message, retriever=retriever)
     
-    chat_box[-1] = ("ParvizGPT", response) 
-    chat_box.append(("You", user_message))
+    chat_box[-1] = ("You", user_message) 
+    chat_box.append(("ParvizGPT", response))
     
     return chat_box
 
@@ -97,8 +101,8 @@ retriever = None
 
 with gr.Blocks() as interface:
     gr.Markdown("## ParvizGPT")
-    user_message = gr.Textbox(label="Your Message", placeholder="Type your message here and press Enter...", lines=1, interactive=True)
     chat_box = gr.Chatbot(label="Chat History", value=[])
+    user_message = gr.Textbox(label="Your Message", placeholder="Type your message here and press Enter...", lines=1, interactive=True)
     clear_memory_btn = gr.Button("Clear Memory", interactive=True)
     pdf_file = gr.File(label="Upload PDF for Context (Optional)", type="filepath", interactive=True, scale=1)
     submit_btn = gr.Button("Submit")