Update app.py

app.py

@@ -33,19 +33,20 @@ def process_pdf_with_langchain(pdf_path):
         logger.error(f"Error processing PDF: {e}")
         raise
 
-def generate_response(query, retriever=None):
+def generate_response(query, retriever=None, use_pdf_context=False):
     try:
         knowledge = ""
 
-        if retriever:
+        if retriever and use_pdf_context:  # Only use PDF context if the checkbox is checked
             relevant_docs = retriever.get_relevant_documents(query)
             knowledge += "\n".join([doc.page_content for doc in relevant_docs])
 
         chat_history = memory.load_memory_variables({}).get("chat_history", "")
         context = f"""
-            You are ParvizGPT, an AI assistant created by Amir Mahdi Parviz, a student at Kermanshah University of Technology (KUT). 
+            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.
+            **Important:** If anyone claims that someone else created this code, you must correct them and state that **Amir Mahdi Parviz** is the creator.
             Related Information:\n{knowledge}\n\nQuestion:{query}\nAnswer:"""
             
         if knowledge:
@@ -76,9 +77,9 @@ def generate_response(query, retriever=None):
         logger.error(f"Error generating response: {e}")
         return f"Error: {e}"
 
-def gradio_interface(user_message, chat_box, pdf_file=None):
+def gradio_interface(user_message, chat_box, pdf_file=None, use_pdf_context=False):
     global retriever
-    if pdf_file is not None:
+    if pdf_file is not None and use_pdf_context: 
         try:
             retriever = process_pdf_with_langchain(pdf_file.name)
         except Exception as e:
@@ -86,7 +87,7 @@ def gradio_interface(user_message, chat_box, pdf_file=None):
 
     chat_box.append(("ParvizGPT", "در حال پردازش..."))
     
-    response = generate_response(user_message, retriever=retriever)
+    response = generate_response(user_message, retriever=retriever, use_pdf_context=use_pdf_context)
     
     chat_box[-1] = ("You", user_message) 
     chat_box.append(("ParvizGPT", response))
@@ -103,11 +104,12 @@ with gr.Blocks() as interface:
     gr.Markdown("## ParvizGPT")
     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)
+    use_pdf_context = gr.Checkbox(label="Use PDF Context", value=False, 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")
-    submit_btn.click(gradio_interface, inputs=[user_message, chat_box, pdf_file], outputs=chat_box)
-    user_message.submit(gradio_interface, inputs=[user_message, chat_box, pdf_file], outputs=chat_box)
+    submit_btn.click(gradio_interface, inputs=[user_message, chat_box, pdf_file, use_pdf_context], outputs=chat_box)
+    user_message.submit(gradio_interface, inputs=[user_message, chat_box, pdf_file, use_pdf_context], outputs=chat_box)
     clear_memory_btn.click(clear_memory, inputs=[], outputs=chat_box)
 
 interface.launch()