bot added

app.py

@@ -3,7 +3,7 @@ from datetime import datetime
 
 import streamlit as st
 
-from helper import current_year, invoke_search_api
+from helper import ChatBot, current_year, invoke_search_api
 
 # API KEY
 API_KEY = os.environ["API_KEY"]
@@ -169,6 +169,18 @@ if prompt := st.chat_input(
         response = f"""
         Please see search results below: \n{md_data}
         """
+
+        # API Call
+        bot = ChatBot()
+        bot.history = st.session_state.messages.copy()  # Update history from messages
+        response = bot.generate_response(
+            f"""
+            Here's user prompt: {prompt}
+            Here's relevant content: {response}
+
+            Answer user prompt based on the relevant content."""
+        )
+
     except Exception as e:
         st.warning(f"Failed to fetch data: {e}")
         response = "We are fixing the API right now. Please check back later."

helper.py

@@ -5,6 +5,7 @@ from typing import Any, Dict, List
 
 import requests
 import streamlit as st
+from openai import OpenAI
 
 api_key = os.environ["API_KEY"]
 
@@ -67,3 +68,55 @@ def current_year() -> int:
 
     # Extract and return the current year
     return now.year
+
+
+class ChatBot:
+    """
+    A simple chatbot class that interacts with the OpenAI API to generate responses
+    based on user prompts and maintains a conversation history.
+    """
+
+    def __init__(self):
+        """
+        Initialize the ChatBot instance by creating an OpenAI client and setting up
+        an initial conversation history with the role of 'system' as a helpful assistant.
+        """
+        self.client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
+        self.history = [{"role": "system", "content": "You are a helpful assistant."}]
+
+    def generate_response(self, prompt: str) -> str:
+        """
+        Generate a response from the chatbot based on the user's prompt.
+
+        Args:
+            prompt (str): The input message from the user.
+
+        Returns:
+            str: The chatbot's response to the provided prompt.
+        """
+        # Add the user's prompt to the history with the role of 'user'.
+        self.history.append({"role": "user", "content": prompt})
+
+        # Request a completion from the OpenAI API using the conversation history.
+        completion = self.client.chat.completions.create(
+            model="gpt-3.5-turbo",  # NOTE: feel free to change it to gpt-4, or gpt-4o
+            messages=self.history,
+        )
+
+        # Extract the assistant's response from the completion result.
+        response = completion.choices[0].message.content
+
+        # Add the assistant's response to the history with the role of 'assistant'.
+        self.history.append({"role": "assistant", "content": response})
+
+        # Return the generated response.
+        return response
+
+    def get_history(self) -> list:
+        """
+        Retrieve the conversation history between the user and the chatbot.
+
+        Returns:
+            list: A list of messages representing the conversation history.
+        """
+        return self.history