Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import os
-import requests
 from tavily import TavilyClient
 from groq import Groq
 
@@ -8,41 +7,41 @@ from groq import Groq
 TAVILY_API_KEY = st.secrets["TAVILY_API_KEY"]
 GROQ_API_KEY = st.secrets["GROQ_API_KEY"]
 
+# Step 2: Initialize Clients for Tavily and Groq
 if TAVILY_API_KEY is None or GROQ_API_KEY is None:
     st.error("API key(s) are missing. Please add the API keys in Hugging Face Secrets.")
 else:
-    # Step 2: Initialize Clients for Tavily and Groq
     tavily_client = TavilyClient(api_key=TAVILY_API_KEY)
     groq_client = Groq(api_key=GROQ_API_KEY)
 
     # Step 3: Fetch Suggestions Using Tavily
-    def fetch_tavily_suggestions(location, need):
+    def fetch_tavily_suggestions(query):
         """
-        Use Tavily API to fetch real-time suggestions based on location and need.
+        Use Tavily API to fetch real-time suggestions based on the query.
         """
-        query = f"Provide internet connectivity options for rural areas near {location}, especially affordable providers."
-        
         try:
-            # Execute the query using TavilyClient
             response = tavily_client.search(query)
-            return response.get("text", "No suggestions available.")
+            return response.get("text", None)  # Return None if no suggestions are available
         except Exception as e:
-            return f"Error: {str(e)}"
+            return None  # Return None in case of error
 
     # Step 4: Enhance Suggestions Using Groq
-    def enhance_with_groq(suggestions):
+    def enhance_with_groq(query, suggestions=None):
         """
-        Use the Groq model to enhance and organize the suggestions.
+        If suggestions are provided by Tavily, enhance them using Groq.
+        Otherwise, generate a response using Groq based on the original query.
         """
-        if suggestions == "No suggestions available.":
-            return "It seems like we couldn't find any suggestions. Try asking in a different way or checking with local ISPs."
-        
-        groq_client = Groq(api_key=GROQ_API_KEY)
-        
+        if suggestions is None:
+            # If no suggestions from Tavily, generate from Groq directly
+            response_content = query
+        else:
+            # If suggestions from Tavily exist, use them as context
+            response_content = suggestions
+
         try:
             # Perform chat completion using Groq for better formatting and response enhancement
             chat_completion = groq_client.chat.completions.create(
-                messages=[{"role": "user", "content": suggestions}],
+                messages=[{"role": "user", "content": response_content}],
                 model="llama-3.3-70b-versatile",  # Specify model for Groq
             )
             # Extract and return the enhanced message
@@ -68,18 +67,24 @@ else:
     # Fetch and Enhance Suggestions
     if st.sidebar.button("Get Suggestions"):
         if location:
+            query = f"Provide internet connectivity options for rural areas near {location}, especially affordable providers."
             st.subheader(f"Suggestions for {needs} in {location}")
-            tavily_suggestions = fetch_tavily_suggestions(location, needs)
-            st.write("Raw Suggestions from Tavily:")
-            st.write(tavily_suggestions)
+            
+            # Get suggestions from Tavily
+            tavily_suggestions = fetch_tavily_suggestions(query)
 
-            # Check if the response is empty before passing to Groq
-            if tavily_suggestions != "No suggestions available.":
+            if tavily_suggestions:
+                st.write("Raw Suggestions from Tavily:")
+                st.write(tavily_suggestions)
                 # Enhance the response using Groq
-                enhanced_suggestions = enhance_with_groq(tavily_suggestions)
-                st.subheader("Enhanced Suggestions:")
+                enhanced_suggestions = enhance_with_groq(query, tavily_suggestions)
+                st.subheader("Enhanced Suggestions from Groq:")
                 st.write(enhanced_suggestions)
             else:
-                st.warning("No suggestions available. Try refining your query.")
+                st.write("No suggestions available from Tavily. Generating response from Groq...")
+                # If Tavily doesn't provide any suggestions, generate a response from Groq
+                groq_response = enhance_with_groq(query)
+                st.subheader("Generated Response from Groq:")
+                st.write(groq_response)
         else:
             st.warning("Please enter your location to get recommendations.")