Update app.py

app.py

@@ -1,34 +1,46 @@
 import openai
 import streamlit as st
 
+# Load API key from secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
 # Fine-tuned model name
 MODEL_NAME = "ft:gpt-3.5-turbo-0125:brenin::AjNcIvnw"
 
+# Streamlit App UI
 st.title("Chat with Fine-Tuned GPT-3.5 Turbo")
 st.markdown("This chatbot uses a fine-tuned GPT-3.5 Turbo model.")
 
+# Initialize session state for messages
 if "messages" not in st.session_state:
     st.session_state.messages = [
         {"role": "system", "content": "You are a German chatbot. You should help the user by answering their questions."}
     ]
 
+# User input field
 user_input = st.text_input("Enter your question:")
 
+# If user submits a question
 if user_input:
+    # Add user message to conversation history
     st.session_state.messages.append({"role": "user", "content": user_input})
 
+    # Call the OpenAI API using client.chat.completions.create()
     try:
         response = openai.ChatCompletion.create(
             model=MODEL_NAME,
             messages=st.session_state.messages
         )
-        assistant_reply = response.choices[0].message.content
+
+        # Extract the assistant's reply
+        assistant_reply = response['choices'][0]['message']['content']
         st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
+
     except Exception as e:
         assistant_reply = f"Error: {str(e)}"
+        st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
 
+    # Display the conversation
     for message in st.session_state.messages:
         if message["role"] == "user":
             st.markdown(f"**User:** {message['content']}")