Update app.py

app.py

@@ -30,7 +30,7 @@ def summarize_pdf(file, openai_api_key):
     vector_store = FAISS.from_documents(documents, embeddings)
 
     # Create a RetrievalQA chain for summarization
-    llm = ChatOpenAI(model="gpt-4o", openai_api_key=openai_api_key)  # Passing API key here
+    llm = ChatOpenAI(model="gpt-4", openai_api_key=openai_api_key)  # Passing API key here
     qa_chain = RetrievalQA.from_chain_type(
         llm=llm,
         chain_type="stuff",
@@ -56,7 +56,7 @@ def query_pdf(file, user_query, openai_api_key):
     vector_store = FAISS.from_documents(documents, embeddings)
 
     # Create a RetrievalQA chain for querying the document
-    llm = ChatOpenAI(model="gpt-4o", openai_api_key=openai_api_key)  # Passing API key here
+    llm = ChatOpenAI(model="gpt-4", openai_api_key=openai_api_key)  # Passing API key here
     qa_chain = RetrievalQA.from_chain_type(
         llm=llm,
         chain_type="stuff",
@@ -81,16 +81,27 @@ def create_gradio_interface():
                 pdf_file = gr.File(label="Upload PDF Document")
                 summarize_btn = gr.Button("Summarize")
                 summary_output = gr.Textbox(label="Summary", interactive=False)
-                
+                clear_btn_summary = gr.Button("Clear Response")
+
+            # Summarize Button Logic
             summarize_btn.click(summarize_pdf, inputs=[pdf_file, openai_api_key_input], outputs=summary_output)
+
+            # Clear Response Button Logic for Summary Tab
+            clear_btn_summary.click(lambda: "", inputs=[], outputs=summary_output)
         
         with gr.Tab("Ask Questions"):
             with gr.Row():
                 pdf_file_q = gr.File(label="Upload PDF Document")
                 user_input = gr.Textbox(label="Enter your question")
                 answer_output = gr.Textbox(label="Answer", interactive=False)
-            
+                clear_btn_answer = gr.Button("Clear Response")
+
+            # Submit Question Logic
             user_input.submit(query_pdf, inputs=[pdf_file_q, user_input, openai_api_key_input], outputs=answer_output)
+
+            # Clear Response Button Logic for Answer Tab
+            clear_btn_answer.click(lambda: "", inputs=[], outputs=answer_output)
+
             user_input.submit(None, None, answer_output)  # Clear answer when typing new query
     
     return demo