Update app.py

app.py

@@ -6,52 +6,27 @@ from langchain.schema import HumanMessage, AIMessage
 
 # Load environment variables
 load_dotenv()
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-
-# Initialize OpenAI chat model using LangChain
-chat = ChatOpenAI(
-    openai_api_key=OPENAI_API_KEY,
-    model_name="gpt-4o",
-    temperature=0.7,
-    max_retries=3  # Automatically retry failed requests
-)
-
-# Streamlit UI setup
-st.set_page_config(
-    page_title="ChatGPT",
-    page_icon="",
-    layout="centered",
-)
-
-# Chat history initialization
+
+
+
+openai_api_key =  os.getenv("OPENAI_API_KEY")
+st.title("💬 Chatbot")
+
 if "messages" not in st.session_state:
-    st.session_state.messages = []
+    st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
 
-# Display the chat history
-st.markdown(
-        "<h1 style='text-align: center;'><b> Chatgpt <b/>.</h1>",
-        unsafe_allow_html=True
-    )
 for msg in st.session_state.messages:
-    if isinstance(msg, HumanMessage):
-        st.write(f"**You:** {msg.content}")
-    elif isinstance(msg, AIMessage):
-        st.write(f"**Assistant:** {msg.content}")
-
-# User input box
-user_input = st.text_input("Ask me anything:")
-
-if user_input:
-    # Display user message
-    st.write(f"**You:** {user_input}")
-    st.session_state.messages.append(HumanMessage(content=user_input))
-    
-    # Generate assistant response
-    with st.spinner("Thinking..."):
-        try:
-            # Use .invoke() instead of calling chat directly
-            response = chat.invoke(st.session_state.messages)
-            st.write(f"**Assistant:** {response.content}")
-            st.session_state.messages.append(AIMessage(content=response.content))
-        except Exception as e:
-            st.error(f"Error: {e}")
+    st.chat_message(msg["role"]).write(msg["content"])
+
+if prompt := st.chat_input():
+    if not openai_api_key:
+        st.info("Please add your OpenAI API key to continue.")
+        st.stop()
+
+    client = OpenAI(api_key=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-3.5-turbo", 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)