Update app.py

app.py

@@ -4,8 +4,8 @@ import openai
 import faiss
 import os
 import numpy as np
-from sklearn.feature_extraction.text import TfidfVectorizer
-from sklearn.metrics.pairwise import cosine_similarity
+import pyautogui
+import time
 
 # Function to extract text from a PDF file
 def extract_text_from_pdf(pdf_file):
@@ -81,7 +81,7 @@ if openai_api_key:
 
             # Get the GPT-4 response
             response = openai.ChatCompletion.create(
-                model="gpt-4o-mini",  # Update to GPT-4 (or your desired model)
+                model="gpt-4",  # Update to GPT-4 (or your desired model)
                 messages=[{"role": "user", "content": modified_prompt}]
             )
 
@@ -91,3 +91,15 @@ 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'.")