Update app.py

app.py

@@ -9,8 +9,9 @@ from groq import Groq
 # Initialize Groq client
 client = Groq(api_key=os.environ.get('GroqApi'))
 
-# Global variable to store scraped data
-SCRAPED_DATA = []
+# Initialize session state for scraped data
+if "scraped_data" not in st.session_state:
+    st.session_state.scraped_data = []
 
 def scrape_web_data(url, scrape_option):
     """Scrape data from the given URL based on the scrape option."""
@@ -71,12 +72,13 @@ website_url = st.text_input("Enter the URL to scrape:")
 scrape_option = st.selectbox("Select what to scrape:", ['data', 'links'])
 
 if st.button("Scrape Data"):
-    SCRAPED_DATA = scrape_web_data(website_url, scrape_option)
-    if SCRAPED_DATA:
-        st.success(f"Scraping completed. {len(SCRAPED_DATA)} items found.")
+    scraped_data = scrape_web_data(website_url, scrape_option)
+    if scraped_data:
+        st.session_state.scraped_data = scraped_data
+        st.success(f"Scraping completed. {len(scraped_data)} items found.")
 
         # Save data to a temporary CSV file
-        df = pd.DataFrame(SCRAPED_DATA)
+        df = pd.DataFrame(scraped_data)
         csv_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
         df.to_csv(csv_file.name, index=False)
 
@@ -97,6 +99,6 @@ if st.button("Get Answer"):
     if user_query.strip() == "":
         st.warning("Please enter a valid query.")
     else:
-        answer = process_query_with_groq(user_query, SCRAPED_DATA)
+        answer = process_query_with_groq(user_query, st.session_state.scraped_data)
         st.write("**Answer:**")
         st.write(answer)