code cleaned up

app.py

@@ -1,8 +1,7 @@
+import os
 from datetime import datetime
 
 import streamlit as st
-import os
-from typing import List, Dict, Any
 
 from helper import current_year, invoke_search_api
 
@@ -16,7 +15,8 @@ st.title("SearchBot 🤖")
 # Front-end: Sidebar
 with st.sidebar:
     with st.expander("Instruction Manual"):
-        st.markdown("""
+        st.markdown(
+            """
             ## SearchBot 🤖
             This Streamlit app allows you to search anything.
 
@@ -31,13 +31,16 @@ with st.sidebar:
             - **Developer**: [Yiqiao Yin](https://www.y-yin.io/) | [App URL](https://huggingface.co/spaces/eagle0504/serpapi-demo) | [LinkedIn](https://www.linkedin.com/in/yiqiaoyin/) | [YouTube](https://youtube.com/YiqiaoYin/)
 
             Enjoy chatting with SearchBot!
-        """)
+        """
+        )
 
     # Example:
     st.success("Example: Yiqiao Yin.")
     st.success("Example: Weather in Westchester, NY today")
     st.success("Example: Latest news about Tesla")
-    st.success("Example: Taking medicine during pregnancy (you can select WebMD to filter only results from WebMD)")
+    st.success(
+        "Example: Taking medicine during pregnancy (you can select WebMD to filter only results from WebMD)"
+    )
 
     # Input
     src_key_word = st.selectbox(
@@ -52,18 +55,20 @@ with st.sidebar:
             "NYTimes",
             "Economist",
             "WebMD",
-            "BetterHelp", 
+            "BetterHelp",
             "Zocdoc",
             "Clevelandclinic",
             "Drugs.com",
             "EMedicineHealth.com",
             "Reddit",
-        )
+        ),
     )
     src_key_word = "" if src_key_word == "Google" else src_key_word
-    num = st.number_input("Insert a number", value=20, step=1, placeholder="Type a number...")
+    num = st.number_input(
+        "Insert a number", value=20, step=1, placeholder="Type a number..."
+    )
     location = st.selectbox(
-       "Where would you like to search from?",
+        "Where would you like to search from?",
         (
             "New York, New York, United States",
             "Los Angeles, California, United States",
@@ -104,8 +109,8 @@ with st.sidebar:
             "Kansas City, Missouri, United States",
             "Atlanta, Georgia, United States",
             "Omaha, Nebraska, United States",
-            "Colorado Springs, Colorado, United States"
-        )
+            "Colorado Springs, Colorado, United States",
+        ),
     )
 
     # Add a button to clear the session state
@@ -139,19 +144,27 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 # Front-end: Chat Panel - React to user input
-if prompt := st.chat_input("😉 Ask any question or feel free to use the examples provided in the left sidebar."):
-
+if prompt := st.chat_input(
+    "😉 Ask any question or feel free to use the examples provided in the left sidebar."
+):
     # Display user message in chat message container
     st.chat_message("user").markdown(prompt)
 
     # Add user message to chat history
-    st.session_state.messages.append({"role": "system", "content": f"You are a helpful assistant. Year now is {current_year}"})
+    st.session_state.messages.append(
+        {
+            "role": "system",
+            "content": f"You are a helpful assistant. Year now is {current_year}",
+        }
+    )
     st.session_state.messages.append({"role": "user", "content": prompt})
 
     # API Call
-    query = src_key_word + ' ' + prompt
+    query = src_key_word + " " + prompt
     try:
-        md_data = invoke_search_api(api_key=API_KEY, query=query, location=location, num=num)
+        md_data = invoke_search_api(
+            api_key=API_KEY, query=query, location=location, num=num
+        )
         md_data = md_data["data"]
         response = f"""
         Please see search results below: \n{md_data}
@@ -165,4 +178,4 @@ if prompt := st.chat_input("😉 Ask any question or feel free to use the exampl
         st.markdown(response, unsafe_allow_html=True)
 
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response})

helper.py

@@ -1,19 +1,19 @@
+import json
+import os
 from datetime import datetime
+from typing import Any, Dict, List
 
-import streamlit as st
-import os
-from typing import List, Dict, Any
 import requests
-import json
+import streamlit as st
 
 api_key = os.environ["API_KEY"]
 
 
 def invoke_search_api(
-        api_key: str,
-        query: str,
-        location: str = "New York, New York, United States",
-        num: int = 10
+    api_key: str,
+    query: str,
+    location: str = "New York, New York, United States",
+    num: int = 10,
 ) -> Dict[str, any]:
     """
     Invoke the Vitanexus SearchBot API to retrieve search results based on a query and location.
@@ -25,30 +25,26 @@ def invoke_search_api(
         num (int, optional): The number of search results to retrieve. Default is 10.
 
     Returns:
-        Dict[str, any]: A dictionary containing the JSON response from the API. 
+        Dict[str, any]: A dictionary containing the JSON response from the API.
                         If an error occurs, returns a dictionary with error details.
     """
-    
+
     # Construct the API endpoint URL with the provided API key
     url = f"https://vitanexus-searchbot-v2.azurewebsites.net/api/http_trigger?code={api_key}"
-    
+
     # Create the JSON payload for the POST request
-    payload = {
-        "query": query,
-        "location": location,
-        "num": num
-    }
-    
+    payload = {"query": query, "location": location, "num": num}
+
     try:
         # Send a POST request to the API endpoint with the JSON payload
         response = requests.post(url, json=payload)
-        
+
         # Raise an exception for any HTTP error responses
         response.raise_for_status()
-        
+
         # Return the parsed JSON response if the request is successful
         return response.json()
-    
+
     except requests.exceptions.RequestException as e:
         # Handle any request exceptions and return an error message
         print(f"An error occurred: {e}")
@@ -60,7 +56,7 @@ def current_year() -> int:
     """
     Returns the current year as an integer.
 
-    This function fetches the current date and time, then extracts and returns 
+    This function fetches the current date and time, then extracts and returns
     the year from the current date.
 
     Returns:
@@ -68,6 +64,6 @@ def current_year() -> int:
     """
     # Get the current date and time
     now: datetime = datetime.now()
-    
+
     # Extract and return the current year
     return now.year