Update app.py

app.py

@@ -1,55 +1,25 @@
 import streamlit as st
-import requests
-
-from langchain.utilities import ApifyWrapper
-from langchain.document_loaders.base import Document
-
-# Initialize ApifyWrapper and other environment variables
-os.environ["OPENAI_API_KEY"] = "Your OpenAI API key"
-os.environ["APIFY_API_TOKEN"] = "Your Apify API token"
+from apify_client import ApifyClient
 
+APIFY_API_TOKEN = "apify_api_uz0y556N4IG2aLcESj67kmnGSUpHF12XAkLp"
 APIFY_WEATHER_KEY = "91b23cab82ee530b2052c8757e343b0d"
-apify = ApifyWrapper()
 
-def get_website_summary(website_url):
-    # For now, return a dummy summary. Replace this with the API logic you have
-    return "This is a dummy summary for " + website_url
+client = ApifyClient(APIFY_API_TOKEN)
 
-def get_google_reviews(website_url):
-    # Return dummy Google Maps reviews for now.
-    # Replace this with the logic to get reviews summary using GPT-3.5 and Google Maps API.
-    return {
-        "pros": ["Great ambiance", "Friendly staff", "Tasty food"],
-        "cons": ["A bit pricey", "Long waiting times on weekends", "Limited parking"]
+def fetch_places_from_google_maps(website_name, location):
+    run_input = {
+        "searchStringsArray": [website_name],
+        "locationQuery": location,
+        "maxCrawledPlacesPerSearch": 1,  # Fetching only one record for simplicity
+        # Other input parameters can be added as per your requirements
     }
 
-def get_peak_times(website_url):
-    # Dummy data for peak times.
-    # Replace this with the logic to fetch peak times using Google Maps API.
-    return {
-        "Monday": [10, 20, 30, 40, 30, 20, 10],
-        "Tuesday": [15, 25, 35, 45, 35, 25, 15],
-        # ... add other days
-    }
-
-def get_images(website_url):
-    # Return dummy image URLs.
-    # Replace this with logic to fetch place images using Google Maps API.
-    return [
-        "https://via.placeholder.com/150",
-        "https://via.placeholder.com/150",
-        "https://via.placeholder.com/150",
-        "https://via.placeholder.com/150",
-        "https://via.placeholder.com/150",
-    ]
-
-def get_map_widget(website_url):
-    # Return dummy map coordinates (for New York City).
-    # Replace this with the logic to fetch actual coordinates of the place using its website URL.
-    return {
-        "lat": 40.730610,
-        "lon": -73.935242
-    }
+    run = client.actor("mc9KJTQJg3zfQpANg/nwua9Gu5YrADL7ZDj").call(run_input=run_input)
+    
+    items = list(client.dataset(run["defaultDatasetId"]).iterate_items())
+    if items:
+        return items[0]
+    return None
 
 def get_weather_data(lat, lon):
     BASE_URL = f"https://api.openweathermap.org/data/3.0/onecall"
@@ -66,53 +36,42 @@ def get_weather_data(lat, lon):
     else:
         return None
 
-def process_website(website_url):
-    website_summary = get_website_summary(website_url)
-    reviews = get_google_reviews(website_url)
-    peak_times = get_peak_times(website_url)
-    images = get_images(website_url)
-    map_widget = get_map_widget(website_url)
-    
-    weather_data = get_weather_data(map_widget["lat"], map_widget["lon"])
-    if weather_data:
-        temp = weather_data["current"]["temp"]
-        weather_desc = weather_data["current"]["weather"][0]["description"]
-        weather_widget = f"It's {temp}°C with {weather_desc}."
-    else:
-        weather_widget = "Unable to fetch weather data at the moment."
-    
-    return website_summary, reviews, peak_times, images, map_widget, weather_widget
-
 # Streamlit UI
 st.title("Website Information Extractor")
 
-website_url = st.text_input("Enter a website/company name:")
-
-if website_url:
-    summary, reviews, peak_times, images, map_data, weather_info = process_website(website_url)
-
-    st.subheader("Website Summary")
-    st.write(summary)
-
-    st.subheader("Google Reviews")
-    st.write("Pros:")
-    for pro in reviews["pros"]:
-        st.write(f"- {pro}")
-
-    st.write("Cons:")
-    for con in reviews["cons"]:
-        st.write(f"- {con}")
+website_name = st.text_input("Enter a website/company name:")
+location = st.text_input("Enter location (e.g., New York, USA):")
 
-    st.subheader("Peak Times")
-    for day, times in peak_times.items():
-        st.write(f"{day}: {times}")
-
-    st.subheader("Images")
-    for image in images:
-        st.image(image)
-
-    st.subheader("Location")
-    st.map({"lat": map_data["lat"], "lon": map_data["lon"]})
-
-    st.subheader("Current Weather")
-    st.write(weather_info)
+if website_name and location:
+    place_data = fetch_places_from_google_maps(website_name, location)
+    
+    if place_data:
+        st.subheader("Place Details")
+        st.write(f"Name: {place_data.get('name')}")
+        st.write(f"Address: {place_data.get('address')}")
+        st.write(f"Rating: {place_data.get('rating')}")
+        
+        # Display reviews if available
+        if 'reviews' in place_data and place_data['reviews']:
+            st.subheader("Reviews")
+            for review in place_data['reviews']:
+                st.write(f"Reviewer: {review.get('reviewerName', 'N/A')}")
+                st.write(f"Review: {review.get('text', 'N/A')}")
+        
+        # Display images if available
+        if 'imageUrls' in place_data and place_data['imageUrls']:
+            st.subheader("Images")
+            for image_url in place_data['imageUrls']:
+                st.image(image_url)
+
+        lat, lon = place_data.get("location", {}).get("lat"), place_data.get("location", {}).get("lng")
+        
+        if lat and lon:
+            weather_data = get_weather_data(lat, lon)
+            if weather_data:
+                temp = weather_data["current"]["temp"]
+                weather_desc = weather_data["current"]["weather"][0]["description"]
+                st.subheader("Current Weather")
+                st.write(f"It's {temp}°C with {weather_desc}.")
+    else:
+        st.write("Couldn't fetch details for the provided input. Please try again.")