Update app.py

app.py

@@ -1,48 +1,47 @@
 import openai
 import streamlit as st
 
-# Load API key from secrets
-openai.api_key = st.secrets["OPENAI_API_KEY"]
+class FineTunedChatbot:
+    def __init__(self, api_key, model_name):
+        openai.api_key = api_key
+        self.client = openai
+        self.model_name = model_name
+
+    def get_response(self, query_text):
+        try:
+            # Use client.chat.completions.create
+            response = self.client.chat.completions.create(
+                model=self.model_name,
+                messages=[
+                    {"role": "system", "content": "You are a German chatbot. You should help the user by answering their question."},
+                    {"role": "user", "content": query_text}
+                ]
+            )
+            # Extract the response content
+            answer = response.choices[0].message.content
+            return {'answer': answer}
+
+        except Exception as e:
+            return {'error': f'Error processing query: {str(e)}'}
 
 # Fine-tuned model name
 MODEL_NAME = "ft:gpt-3.5-turbo-0125:brenin::AjNcIvnw"
 
-# Streamlit App UI
+# 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."}
-    ]
+# Initialize chatbot
+chatbot = FineTunedChatbot(api_key=st.secrets["OPENAI_API_KEY"], model_name=MODEL_NAME)
 
 # 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
-        )
-
-        # 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']}")
-        elif message["role"] == "assistant":
-            st.markdown(f"**Assistant:** {message['content']}")
+    # Fetch response from chatbot
+    response = chatbot.get_response(user_input)
+
+    if 'answer' in response:
+        st.markdown(f"**Assistant:** {response['answer']}")
+    else:
+        st.markdown(f"**Error:** {response['error']}")