Update app.py

app.py

@@ -11,33 +11,12 @@ from helpers.foundation_models import *
 import requests
 
 
+# API Keys
 OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
 SERPAPI_API_KEY = os.environ["SERPAPI_API_KEY"]
 openai_client = openai.OpenAI(api_key=OPENAI_API_KEY)
 
 
-API_URL = "https://sks7h7h5qkhoxwxo.us-east-1.aws.endpoints.huggingface.cloud"
-headers = {
-	"Accept" : "application/json",
-	"Content-Type": "application/json" 
-}
-
-
-def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.json()
-
-
-def llama2_7b_ysa(prompt: str) -> str:
-    output = query({
-        "inputs": prompt,
-        "parameters": {}
-    })
-
-    response = output[0]['generated_text']
-
-    return response
-
 # Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -49,6 +28,7 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 
+# Sidebar
 with st.expander("Instructions"):
     st.sidebar.markdown(
         r"""
@@ -68,9 +48,7 @@ option = st.sidebar.selectbox(
 
 
 clear_button = st.sidebar.button("Clear Conversation", key="clear")
-
 st.sidebar.write("---")
-
 st.sidebar.markdown("Yiqiao Yin: [Site](https://www.y-yin.io/) | [LinkedIn](https://www.linkedin.com/in/yiqiaoyin/)")
 
 
@@ -81,11 +59,14 @@ if clear_button:
 
 # React to user input
 if prompt := st.chat_input("What is up?"):
+
     # 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": "user", "content": prompt})
 
+    # Execute options
     with st.spinner("Wait for it..."):
         if option == "Sentiment Analysis":
             pipe_sentiment_analysis = pipeline("sentiment-analysis")