Update app.py

app.py

@@ -10,7 +10,7 @@ from groq import Groq
 client = Groq(api_key=os.environ.get('GroqApi'))
 
 # Global variable to store scraped data
-SCRAPED_DATA = None
+SCRAPED_DATA = []
 
 def scrape_web_data(url, scrape_option):
     """Scrape data from the given URL based on the scrape option."""
@@ -41,6 +41,9 @@ def scrape_web_data(url, scrape_option):
 
 def process_query_with_groq(query, data):
     """Process the user's query with Groq based on the scraped data."""
+    if not data:
+        return "No data available to process. Please scrape data first."
+
     try:
         # Combine the scraped data into a single text block
         combined_text = "\n".join([str(item) for sublist in data for item in sublist.values()])
@@ -63,13 +66,14 @@ def process_query_with_groq(query, data):
 st.title("Web Scraping and Query Tool")
 
 # Step 1: Scraping
+st.subheader("Step 1: Scrape Data")
 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.write(f"Scraping completed. {len(SCRAPED_DATA)} items found.")
+        st.success(f"Scraping completed. {len(SCRAPED_DATA)} items found.")
 
         # Save data to a temporary CSV file
         df = pd.DataFrame(SCRAPED_DATA)
@@ -84,12 +88,15 @@ if st.button("Scrape Data"):
             mime="text/csv",
         )
     else:
-        st.write("No data found.")
+        st.warning("No data found. Please check the URL or scrape option.")
 
 # Step 2: Querying
-if SCRAPED_DATA:
-    user_query = st.text_input("Enter your query:")
-    if st.button("Get Answer"):
+st.subheader("Step 2: Ask a Query")
+user_query = st.text_input("Enter your query:")
+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)
         st.write("**Answer:**")
         st.write(answer)