Update app.py

app.py

@@ -3,14 +3,13 @@ import os
 import openai
 import streamlit as st
 
-# Load OpenAI API key from Hugging Face Secrets
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Load API key from environment variable
+client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-# Set Streamlit page config
+# Page setup
 st.set_page_config(page_title="OpenAI Chatbot", layout="centered")
-
-st.title("💬 OpenAI Chatbot")
-st.markdown("Talk with GPT-4 using OpenAI API.")
+st.title("🤖 OpenAI Chatbot")
+st.markdown("Talk to GPT-4 using OpenAI's latest API.")
 
 # Initialize chat history
 if "messages" not in st.session_state:
@@ -18,28 +17,26 @@ if "messages" not in st.session_state:
         {"role": "system", "content": "You are a helpful assistant."}
     ]
 
-# Display previous messages
+# Show chat history
 for msg in st.session_state.messages[1:]:
-    if msg["role"] == "user":
-        st.chat_message("user").markdown(msg["content"])
-    elif msg["role"] == "assistant":
-        st.chat_message("assistant").markdown(msg["content"])
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"])
 
-# Chat input
-if prompt := st.chat_input("Say something..."):
+# User input
+if prompt := st.chat_input("Type your message..."):
+    # Show user message
     st.chat_message("user").markdown(prompt)
     st.session_state.messages.append({"role": "user", "content": prompt})
 
     try:
-        # Get response from OpenAI
-        response = openai.ChatCompletion.create(
+        # Call OpenAI's chat model using v1.x API
+        response = client.chat.completions.create(
             model="gpt-4",  # or "gpt-3.5-turbo"
             messages=st.session_state.messages,
-            temperature=0.7,
+            temperature=0.7
         )
-        reply = response.choices[0].message["content"]
+        reply = response.choices[0].message.content
         st.chat_message("assistant").markdown(reply)
         st.session_state.messages.append({"role": "assistant", "content": reply})
     except Exception as e:
-        error_msg = f"❌ Error: {str(e)}"
-        st.chat_message("assistant").markdown(error_msg)
+        st.error(f"Error: {e}")