Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import pandas as pd
-import requests
 from apify_client import ApifyClient
 
 # Function to fetch Google Maps info using the updated actor
@@ -19,9 +18,9 @@ def fetch_weather_info(lat, lon):
     return response.json()
 
 # Function to fetch website content using the updated actor
-def fetch_website_content(website_url):
+def fetch_website_content(google_maps_url):
     apify_client = ApifyClient("apify_api_uz0y556N4IG2aLcESj67kmnGSUpHF12XAkLp")
-    run_input = {}
+    run_input = {"startUrls": [{"url": google_maps_url}]}
     run = apify_client.actor("moJRLRc85AitArpNN").call(run_input=run_input)
     items = list(apify_client.dataset(run["defaultDatasetId"]).iterate_items())
     return items if items else None
@@ -31,7 +30,7 @@ def fetch_customer_reviews(google_maps_url):
     apify_client = ApifyClient("apify_api_uz0y556N4IG2aLcESj67kmnGSUpHF12XAkLp")
     run_input = {
         "startUrls": [{"url": google_maps_url}],
-        "maxReviews": 100,
+        "maxReviews": 20,
         "reviewsSort": "newest",
         "language": "en",
         "personalData": True,
@@ -55,56 +54,21 @@ if website_name:
     progress_bar.progress(50)
 
     if google_maps_data:
-        # Display website link
-        website_link = google_maps_data.get('website')
-        st.text_area("Website Link:", website_link)
-
-        # Display location and fetch weather info
-        lat = google_maps_data["location"]["lat"]
-        lng = google_maps_data["location"]["lng"]
-        st.map(pd.DataFrame({'lat': [lat], 'lon': [lng]}))
-        weather_data = fetch_weather_info(lat, lng)
-        current_weather = weather_data.get("current", {})
-        temp = current_weather.get('temp')
-        temp_in_celsius = temp - 273.15
-        st.write(f"**Location:** {lat}, {lng}")
-        st.write(f"**Temperature:** {temp_in_celsius:.2f}°C")
-        st.write(f"**Weather:** {current_weather.get('weather')[0].get('description')}")
-        
-        # Display Occupancy Data
-        st.subheader("Occupancy Data")
-        occupancy_data = google_maps_data.get('popularTimesHistogram', {})
-        for day, day_data in occupancy_data.items():
-            hours = [entry['hour'] for entry in day_data]
-            occupancy = [entry['occupancyPercent'] for entry in day_data]
-            st.write(day)
-            st.bar_chart(pd.Series(occupancy, index=hours))
-
-        # Display Review Count and Distribution
-        st.subheader("Review Count and Distribution")
-        st.write(f"Total Reviews Count: {google_maps_data['reviewsCount']}")
-        review_distribution = google_maps_data.get('reviewsDistribution', {})
-        st.bar_chart(pd.Series(review_distribution))
-        
-        # Fetch customer reviews from the new API
+        # Fetch and Display Website Content
         google_maps_url = google_maps_data.get('url')
-        reviews_data = fetch_customer_reviews(google_maps_url)
+        website_content_data = fetch_website_content(google_maps_url)
         progress_bar.progress(75)
-
-        # Display the reviews from the new API
-        reviews_df = pd.DataFrame(reviews_data)
-        st.subheader("Customer Reviews from New API")
-        st.table(reviews_df)
-        
-        # Fetch and Display Website Content
-        st.subheader("Website Content")
-        website_content_data = fetch_website_content(website_link)
-        progress_bar.progress(100)
         
         if website_content_data:
             website_df = pd.DataFrame(website_content_data)
             st.table(website_df)
         else:
             st.write("Unable to retrieve website content.")
+        
+        # Fetch customer reviews from the new API and display them
+        reviews_data = fetch_customer_reviews(google_maps_url)
+        reviews_df = pd.DataFrame(reviews_data)[['text']]
+        st.subheader("Customer Reviews from New API")
+        st.table(reviews_df)
     else:
         st.write("No results found for this website / company name on Google Maps.")