Update app.py

app.py

@@ -1,12 +1,11 @@
-#
 import streamlit as st
 import PyPDF2
 import openai
 import faiss
 import os
 import numpy as np
-import pyautogui
-import time
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.metrics.pairwise import cosine_similarity
 
 # Function to extract text from a PDF file
 def extract_text_from_pdf(pdf_file):
@@ -82,7 +81,7 @@ if openai_api_key:
 
             # Get the GPT-4 response
             response = openai.ChatCompletion.create(
-                model="gpt-4",  # Update to GPT-4 (or your desired model)
+                model="gpt-4o-mini",  # Update to GPT-4 (or your desired model)
                 messages=[{"role": "user", "content": modified_prompt}]
             )
 
@@ -91,16 +90,4 @@ if openai_api_key:
 
             # Display the response in Streamlit (Intelligent Reply)
             st.write("### Intelligent Reply:")
-            st.write(response_content)
-
-            # Capture a screenshot of the content displayed in Streamlit
-            # Wait for the UI to render
-            time.sleep(2)  # Allow time for Streamlit to render the response
-
-            # Capture screenshot of the whole screen
-            screenshot = pyautogui.screenshot()
-
-            # Save the screenshot
-            screenshot.save("response_screenshot.png")
-
-            st.write("Screenshot has been saved as 'response_screenshot.png'.")
+            st.write(response_content)