Update app.py

app.py

@@ -1,64 +1,30 @@
-
+import os
+import openai
 import streamlit as st
 
-#As Langchain team has been working aggresively on improving the tool, we can see a lot of changes happening every weeek,
-#As a part of it, the below import has been depreciated
-#from langchain.chat_models import ChatOpenAI
-
-#New import from langchain, which replaces the above
-from langchain_openai import ChatOpenAI
-
-#import os
-#os.environ["OPENAI_API_KEY"] = "sk-PLfFw23dd932dfg34446dftyvvdfgdfgmvXr2dL8hVowXdt"
-
-
-from langchain.schema import (
-    AIMessage,
-    HumanMessage,
-    SystemMessage
-)
-
-# From here down is all the StreamLit UI
-st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
-st.header("Hey, I'm your Chat GPT")
-
-
-
-if "sessionMessages" not in st.session_state:
-     st.session_state.sessionMessages = [
-        SystemMessage(content="You are a helpful assistant.")
-    ]
-
-
-
-def load_answer(question):
-
-    st.session_state.sessionMessages.append(HumanMessage(content=question))
-
-    assistant_answer  = chat.invoke(st.session_state.sessionMessages )
-
-    st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
-
-    return assistant_answer.content
-
-
-def get_text():
-    input_text = st.text_input("You: ")
-    return input_text
-
-
-chat = ChatOpenAI(temperature=0, model="GPT-4o" )
-
-
-
-
-user_input=get_text()
-submit = st.button('Generate')  
-
-if submit:
-    
-    response = load_answer(user_input)
-    st.subheader("Answer:")
-
-    st.write(response)
 
+def chat(user_message):
+    response = openai.ChatCompletion.create(
+        model="gpt-4",
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": user_message}
+        ],
+        temperature=0.7,
+        max_tokens=150,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+    )
+    return response.choices[0].message['content']
+
+st.title("GPT-4 Chatbot")
+st.write("Chat with GPT-4 using OpenAI API")
+
+user_message = st.text_input("You:", "")
+if st.button("Send"):
+    if user_message:
+        response = chat(user_message)
+        st.write(f"GPT-4: {response}")
+    else:
+        st.write("Please enter a message.")