Update app.py

app.py

@@ -1,44 +1,42 @@
-import requests
 import streamlit as st
-# Tavily API Key
-TAVILY_API_KEY= st.secrets["TAVILY_API_KEY"]
+from tavily import TavilyClient
 
+# Step 1: Initialize Tavily Client with API key from Streamlit secrets
+TAVILY_API_KEY = st.secrets["TAVILY_API_KEY"]  # Retrieve API key securely from secrets
+tavily_client = TavilyClient(api_key=TAVILY_API_KEY)
 
-# Function to fetch suggestions from Tavily API
+# Step 2: Fetch Suggestions Using Tavily
 def fetch_tavily_suggestions(location, need):
     """
     Use Tavily API to fetch real-time suggestions based on location and need.
     """
-    headers = {"Authorization": f"Bearer {TAVILY_API_KEY}"}
-    prompt = f"Provide {need.lower()} recommendations for rural areas near {location}. Focus on affordability and accessibility."
-    api_url = "https://api.tavily.ai/v1/generate"
+    query = f"Provide {need.lower()} recommendations for rural areas near {location}. Focus on affordability and accessibility."
+    try:
+        # Execute the query using TavilyClient
+        response = tavily_client.search(query)
+        return response.get("text", "No recommendations available.")  # Extract the 'text' response
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-    payload = {"prompt": prompt, "model": "your-model-choice", "max_tokens": 300}
-
-    response = requests.post(api_url, headers=headers, json=payload)
-    if response.status_code == 200:
-        return response.json().get("text", "No recommendations available.")
-    else:
-        return f"Error: Unable to fetch data from Tavily API. Status code: {response.status_code}"
-
-# Streamlit App
+# Step 3: Build Streamlit UI
 st.title("Rural Connectivity Advisor")
 st.markdown("""
-Welcome to the **Rural Connectivity Advisor**! This app helps rural communities find:
+Welcome to the **Rural Connectivity Advisor**!  
+This app helps rural communities find:
 - Affordable internet solutions.
 - Signal boosting tips.
 - Offline tools for education, healthcare, and business.
 """)
 
-# User Inputs
+# Sidebar User Inputs
 st.sidebar.header("Input Your Details")
 location = st.sidebar.text_input("Enter Your Location (e.g., City, District)")
 needs = st.sidebar.selectbox("What do you need?", ["Internet Provider", "Signal Boosting Tips", "Offline Tools"])
 
-# Get Suggestions Button
+# Fetch and Display Suggestions
 if st.sidebar.button("Get Suggestions"):
     if location:
-        st.subheader(f"Suggestions for {needs}")
+        st.subheader(f"Suggestions for {needs} in {location}")
         suggestions = fetch_tavily_suggestions(location, needs)
         st.write(suggestions)
     else: