Update app.py

app.py

@@ -2,41 +2,68 @@ import streamlit as st
 import openai
 import os
 
-# Set the OpenAI API key from Hugging Face Secrets
-openai.api_key = os.getenv('API_KEY')
+# Function to get the API key from Streamlit secrets
+def get_api_key():
+    try:
+        return st.secrets["API_KEY"]
+    except KeyError:
+        st.error("API_KEY not found in Streamlit secrets. Please add it.")
+        return None
 
-st.title("ChatGPT-powered Chatbot")
+# Function to interact with the OpenAI API
+def generate_response(prompt, model_name, api_key):
+    openai.api_key = api_key
+    try:
+        completion = openai.ChatCompletion.create(
+            model=model_name,
+            messages=[{"role": "user", "content": prompt}]
+        )
+        return completion.choices[0].message.content
+    except openai.error.OpenAIError as e:
+        st.error(f"Error with {model_name}: {e}")
+        return None
 
-# Initialize session state to store the conversation
-if 'messages' not in st.session_state:
-    st.session_state.messages = []
+# Main Streamlit app
+def main():
+    st.title("Chatbot with Model Switching")
 
-# Display previous messages
-for message in st.session_state.messages:
-    st.write(f"**{message['role'].capitalize()}:** {message['content']}")
+    # Initialize conversation history in session state
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
 
-# Input for user message
-user_input = st.text_input("You:", "")
+    # Display previous messages
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
 
-if user_input:
-    # Add user message to session state
-    st.session_state.messages.append({"role": "user", "content": user_input})
+    # Get user input
+    prompt = st.chat_input("Say something")
 
-    # Prepare messages for API call
-    messages = [{"role": msg["role"], "content": msg["content"]} for msg in st.session_state.messages]
+    if prompt:
+        # Add user message to the state
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
 
-    # Call OpenAI API
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=messages
-    )
+        # Define model priority
+        models = ["gpt-4", "gpt-3.5-turbo"]  # Add more models as needed
 
-    # Get assistant's reply
-    assistant_reply = response['choices'][0]['message']['content']
+        # Get API key
+        api_key = get_api_key()
+        if not api_key:
+            return
 
-    # Add assistant's reply to session state
-    st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
+        response = None
+        for model in models:
+            response = generate_response(prompt, model, api_key)
+            if response:
+                break # If a response is generated, break out of loop.
+        
+        if response:
+            # Add bot message to state
+            st.session_state.messages.append({"role": "assistant", "content": response})
+            with st.chat_message("assistant"):
+                st.markdown(response)
 
-    # Display assistant's reply
-    st.write(f"**Assistant:** {assistant_reply}")
-    
+if __name__ == "__main__":
+    main()