Update app.py

app.py

@@ -10,18 +10,18 @@ st.title("Test OpenAI API in Streamlit")
 # User input
 user_input = st.text_area("Enter prompt to generate questions:")
 
-if st.button("Generate Questions"):
+if st.button("Generate Response"):
     if user_input:
-        # Use the new chat-based API
-        response = openai.ChatCompletion.create(
-            model="gpt-4o-mini",  # You can use "gpt-4" if available
+        # Create a chat completion
+        completion = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # Or use "gpt-4" if available to you
             messages=[
-                {"role": "system", "content": "You are a helpful assistant that generates exam questions."},
-                {"role": "user", "content": f"Generate exam questions from the following material: {user_input}"}
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": user_input}
             ]
         )
-        # Display the generated questions
-        st.write("Generated Questions:")
-        st.write(response['choices'][0]['message']['content'])
+        # Display the assistant's reply
+        st.write("Assistant's response:")
+        st.write(completion.choices[0].message['content'])
     else:
         st.warning("Please enter a prompt.")