Update app.py

remove unused components

app.py

@@ -41,9 +41,7 @@ from sentence_transformers import SentenceTransformer, CrossEncoder, util
 from torch import tensor as torch_tensor
 from datasets import load_dataset
 
-
 from greg_funcs import get_llm_response
-from wiki_funcs import mysearch, mygreetings
 
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")  
@@ -68,7 +66,7 @@ tmdb_bearer_token = os.environ["TMDB_BEARER_TOKEN"]
 
 TOOLS_LIST = ['serpapi', 'wolfram-alpha', 'pal-math', 'pal-colored-objects', 'news-api', 'tmdb-api',
               'open-meteo-api']  # 'google-search'
-TOOLS_DEFAULT_LIST = ['mysearch']
+TOOLS_DEFAULT_LIST = ['serpapi']
 BUG_FOUND_MSG = "Congratulations, you've found a bug in this application!"
 AUTH_ERR_MSG = "Please paste your OpenAI key from openai.com to use this application. It is not necessary to hit a button or key after pasting it."
 MAX_TOKENS = 512
@@ -123,8 +121,6 @@ EXAMPLES = ["What is the name of the plan described by this summary of benefits?
 AUTHORS = """
         <p>This application, developed by <b>Greg Hayworth, Srikanth Tangelloju, Lincoln Snyder, Michal Piekarczyk, and Xingde Jiang</b>, 
         demonstrates a conversational agent implemented with OpenAI GPT-3.5 and LangChain. 
-        When necessary, it leverages tools for complex math, searching the internet, and accessing news and weather.
-        Uses talking heads from <a href='https://exh.ai/'>Ex-Human</a>.
         For faster inference without waiting in queue, you may duplicate the space.
         </p>"""
 # UNCOMMENT TO USE WHISPER
@@ -268,42 +264,42 @@ def transform_text(desc, express_chain, num_words, formality,
     return generated_text
 
 
-def load_chain(tools_list, llm):
-    chain = None
-    express_chain = None
-    if llm:
-        print("\ntools_list", tools_list)
-        tool_names = tools_list
-        # tools = load_tools(tool_names, llm=llm, news_api_key=news_api_key,
-        #                    tmdb_bearer_token=tmdb_bearer_token)
-        # tools = load_tools(tool_names, llm=llm)
-        tools = [mysearch, mygreetings]
-        memory = ConversationBufferMemory(memory_key="chat_history")
-
-        chain = initialize_agent(
-            tools, llm, agent="conversational-react-description", verbose=True, memory=memory)
-        express_chain = LLMChain(llm=llm, prompt=PROMPT_TEMPLATE, verbose=True)
-
-    return chain, express_chain
-
-
-def set_openai_api_key(api_key):
-    """Set the api key and return chain.
-    If no api_key, then None is returned.
-    """
-    # if api_key and api_key.startswith("sk-") and len(api_key) > 50:
-    #     # os.environ["OPENAI_API_KEY"] = api_key
-    #     os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
-    #     llm = OpenAI(temperature=TEMPERATURE, max_tokens=MAX_TOKENS)
-    #     chain, express_chain = load_chain(TOOLS_DEFAULT_LIST, llm)
-    #     os.environ["OPENAI_API_KEY"] = ""
-    #     return chain, express_chain, llm
-    # return None, None, None
-
-    os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
-    llm = OpenAI(temperature=TEMPERATURE, max_tokens=MAX_TOKENS)
-    chain, express_chain = load_chain(TOOLS_DEFAULT_LIST, llm)
-    return chain, express_chain, llm
+# def load_chain(tools_list, llm):
+#     chain = None
+#     express_chain = None
+#     if llm:
+#         print("\ntools_list", tools_list)
+#         tool_names = tools_list
+#         # tools = load_tools(tool_names, llm=llm, news_api_key=news_api_key,
+#         #                    tmdb_bearer_token=tmdb_bearer_token)
+#         # tools = load_tools(tool_names, llm=llm)
+#         tools = [mysearch, mygreetings]
+#         memory = ConversationBufferMemory(memory_key="chat_history")
+
+#         chain = initialize_agent(
+#             tools, llm, agent="conversational-react-description", verbose=True, memory=memory)
+#         express_chain = LLMChain(llm=llm, prompt=PROMPT_TEMPLATE, verbose=True)
+
+#     return chain, express_chain
+
+
+# def set_openai_api_key(api_key):
+#     """Set the api key and return chain.
+#     If no api_key, then None is returned.
+#     """
+#     # if api_key and api_key.startswith("sk-") and len(api_key) > 50:
+#     #     # os.environ["OPENAI_API_KEY"] = api_key
+#     #     os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
+#     #     llm = OpenAI(temperature=TEMPERATURE, max_tokens=MAX_TOKENS)
+#     #     chain, express_chain = load_chain(TOOLS_DEFAULT_LIST, llm)
+#     #     os.environ["OPENAI_API_KEY"] = ""
+#     #     return chain, express_chain, llm
+#     # return None, None, None
+
+#     os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
+#     llm = OpenAI(temperature=TEMPERATURE, max_tokens=MAX_TOKENS)
+#     chain, express_chain = load_chain(TOOLS_DEFAULT_LIST, llm)
+#     return chain, express_chain, llm
 
 
 def run_chain(chain, inp, capture_hidden_text):