Update app.py

app.py

@@ -3,20 +3,21 @@ from openai import OpenAI
 import os
 from dotenv import load_dotenv
 
+# Load environment variables
 load_dotenv()
 
 # Set up OpenAI API key
-api_key = os.getenv("OPENAI_API_KEY")  # Replace with your API key if not using an environment variable
+api_key = os.getenv("OPENAI_API_KEY")  # Make sure to have your OpenAI API key in .env
 client = OpenAI(api_key=api_key)
 
 # Function to query OpenAI
-def query_openai(prompt, model="gpt-4o-mini"):
+def query_openai(question, model="gpt-4o-mini"):
     try:
         response = client.chat.completions.create(
             model=model,
             messages=[
-                {"role": "system", "content": "You are a helpful assistant. Respond in 20 words."},
-                {"role": "user", "content": prompt}
+                {"role": "system", "content": "You are a helpful assistant. Please answer in 30 words"},
+                {"role": "user", "content": question}
             ]
         )
         return response.choices[0].message['content']
@@ -24,34 +25,18 @@ def query_openai(prompt, model="gpt-4o-mini"):
         return f"Error: {e}"
 
 # Streamlit app
-st.title("Simple OpenAI Chatbot")
+st.title("Simple Q&A Application")
 
-# Initialize the chat history
-if "messages" not in st.session_state:
-    st.session_state.messages = [{"role": "system", "content": "You are a helpful assistant. Respond in 20 words"}]
+# User input for asking questions
+question = st.text_input("Ask your question:")
 
-# Display chat history
-for message in st.session_state.messages:
-    if message["role"] == "user":
-        st.markdown(f"**You:** {message['content']}")
-    elif message["role"] == "assistant":
-        st.markdown(f"**Bot:** {message['content']}")
+# Handle user input and display the answer
+if st.button("Get Answer"):
+    if question.strip():
+        # Get answer from OpenAI API
+        answer = query_openai(question)
 
-# User input
-user_input = st.text_input("Type your message:")
-
-# Handle user input
-if st.button("Send"):
-    if user_input.strip():
-        # Add user input to chat history
-        st.session_state.messages.append({"role": "user", "content": user_input})
-
-        # Get bot response
-        bot_response = query_openai(user_input)
-
-        # Add bot response to chat history
-        st.session_state.messages.append({"role": "assistant", "content": bot_response})
-
-# Clear chat
-if st.button("Clear Chat"):
-    st.session_state.messages = [{"role": "system", "content": "You are a helpful assistant. Respond in 20 words"}]
+        # Display the answer
+        st.write(f"**Answer:** {answer}")
+    else:
+        st.warning("Please enter a question.")