Update app.py

app.py

@@ -1,30 +1,49 @@
 import os
 from dotenv import load_dotenv
 import streamlit as st
-from langchain_openai import ChatOpenAI
-from langchain.schema import HumanMessage, AIMessage
 import openai
+
 # Load environment variables
 load_dotenv()
 
+# Set OpenAI API key
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-
-openai_api_key =  os.getenv("OPENAI_API_KEY")
+# Streamlit app title
 st.title("💬 Chatbot")
 
+# Initialize chat messages in session state
 if "messages" not in st.session_state:
     st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
 
-for msg in st.session_state.messages:
+# Display existing chat messages
+for msg in st.session_state["messages"]:
     st.chat_message(msg["role"]).write(msg["content"])
 
+# Input from user
 if prompt := st.chat_input():
+    # Append user message to session state
+    st.session_state["messages"].append({"role": "user", "content": prompt})
+    st.chat_message("user").write(prompt)
     
+    try:
+        # OpenAI ChatCompletion API call
+        response = openai.chat.completions.create(
+            model="gpt-4o",
+            messages=[
+                {"role": "system", "content": "You are an helpful assistant "},
+                *st.session_state["messages"]
+            ],
+            max_tokens=150,  # Adjust as needed
+            temperature=0.7  # Adjust as needed
+        )
+        
+        # Extract response message content
+        msg = response.choices[0].message["content"]
+        
+        # Append assistant message to session state
+        st.session_state["messages"].append({"role": "assistant", "content": msg})
+        st.chat_message("assistant").write(msg)
 
-    client = openai_api_key
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    st.chat_message("user").write(prompt)
-    response = client.chat.completions.create(model="gpt-4o", messages=st.session_state.messages)
-    msg = response.choices[0].message.content
-    st.session_state.messages.append({"role": "assistant", "content": msg})
-    st.chat_message("assistant").write(msg)
+    except Exception as e:
+        st.error(f"An error occurred: {e}")