Update app.py

app.py

@@ -11,7 +11,7 @@ import random
 # Load environment variables
 load_dotenv()
 
-# Set page config with a wider layout
+# Set page config
 st.set_page_config(
     page_title="GPT-Style Chat Assistant",
     page_icon="🤖",
@@ -21,7 +21,7 @@ st.set_page_config(
 # Initialize session state
 if "conversations" not in st.session_state:
     st.session_state.conversations = {}
-    
+
 if "current_conversation_id" not in st.session_state:
     new_id = str(uuid.uuid4())
     st.session_state.current_conversation_id = new_id
@@ -30,14 +30,11 @@ if "current_conversation_id" not in st.session_state:
         "messages": []
     }
 
-# Initialize selected model
 if "selected_model" not in st.session_state:
     st.session_state.selected_model = "gpt-3.5-turbo"
 
-# Get OpenAI API key from environment or let user enter it
+# Get OpenAI API key from environment or user input
 openai_api_key = os.getenv("OPENAI_API_KEY_NEW_3")
-
-# Configure OpenAI client
 if openai_api_key:
     openai.api_key = openai_api_key
 
@@ -49,11 +46,65 @@ AVAILABLE_MODELS = {
         "output_tokens": 500,
         "temperature": 0.7
     },
+    "gpt-3.5-turbo-16k": {
+        "description": "Longer context window",
+        "max_tokens": 16384,
+        "output_tokens": 1000, 
+        "temperature": 0.7
+    },
     "gpt-4": {
         "description": "More capable but slower",
         "max_tokens": 8192,
         "output_tokens": 800,
         "temperature": 0.7
+    },
+    "gpt-4-turbo": {
+        "description": "Most powerful model (if available)",
+        "max_tokens": 128000,
+        "output_tokens": 1200,
+        "temperature": 0.7
+    },
+    "gpt-4o": {
+        "description": "Latest GPT-4 Omni model",
+        "max_tokens": 128000,
+        "output_tokens": 1200,
+        "temperature": 0.7
+    },
+    "gpt-4o-mini": {
+        "description": "Efficient version of GPT-4o",
+        "max_tokens": 128000,
+        "output_tokens": 1000,
+        "temperature": 0.7
+    },
+    "o1-mini": {
+       "description": "OpenAI Reasoning Model - Mini",
+       "max_tokens": 180000,
+       "output_tokens": 1000,
+       "temperature": 0.7
+    },
+    "o1": {
+       "description": "OpenAI Reasoning Model - Standard",
+       "max_tokens": 200000,
+       "output_tokens": 1200,
+       "temperature": 0.7
+    },
+    "o1-pro": {
+       "description": "OpenAI Reasoning Model - Professional",
+       "max_tokens": 200000,
+       "output_tokens": 1500,
+       "temperature": 0.7
+    },
+    "o3-mini": {
+       "description": "OpenAI Advanced Reasoning - Mini",
+       "max_tokens": 200000,
+       "output_tokens": 1000,
+       "temperature": 0.7
+    },
+    "o3-mini-2025-01-31": {
+       "description": "OpenAI Advanced Reasoning - Enhanced",
+       "max_tokens": 200000,
+       "output_tokens": 1200,
+       "temperature": 0.7
     }
 }
 
@@ -64,7 +115,7 @@ def get_ai_response(prompt, history):
     
     try:
         messages = [
-            {"role": "system", "content": "You are a helpful assistant that provides clear, concise, and accurate information."}
+            {"role": "system", "content": "You are a helpful assistant providing clear, concise, and accurate information."}
         ]
         for msg in history:
             messages.append({"role": msg["role"], "content": msg["content"]})
@@ -72,18 +123,16 @@ def get_ai_response(prompt, history):
         
         model = st.session_state.selected_model
         model_config = AVAILABLE_MODELS.get(model, AVAILABLE_MODELS["gpt-3.5-turbo"])
-        output_tokens = model_config["output_tokens"]
-        temperature = model_config["temperature"]
         
         response = openai.ChatCompletion.create(
             model=model,
             messages=messages,
-            temperature=temperature,
-            max_tokens=output_tokens,
+            temperature=model_config["temperature"],
+            max_tokens=model_config["output_tokens"],
             stream=False
         )
         return response['choices'][0]['message']['content']
-        
+    
     except openai.error.AuthenticationError as auth_err:
         return f"Authentication error: {str(auth_err)}. Check your API key."
     except openai.error.InvalidRequestError as inv_err:
@@ -97,18 +146,6 @@ def get_ai_response(prompt, history):
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}."
 
-# Demo mode responses for when no API key is available
-def get_demo_response(prompt):
-    greetings = ["Hello! How can I assist you today?", "Hi there! What can I help you with?"]
-    farewells = ["Goodbye! Have a great day!", "Take care! It was nice chatting with you."]
-    
-    if any(word in prompt.lower() for word in ["hello", "hi", "hey"]):
-        return random.choice(greetings)
-    elif any(word in prompt.lower() for word in ["bye", "goodbye", "farewell"]):
-        return random.choice(farewells)
-    else:
-        return "I'm running in demo mode. Provide an OpenAI API key for full functionality."
-
 # Function to create a new conversation
 def create_new_chat():
     new_id = str(uuid.uuid4())
@@ -118,64 +155,43 @@ def create_new_chat():
         "messages": []
     }
 
-# Create a two-column layout
-sidebar, main_content = st.columns([1, 3])
-
-# Sidebar (conversation history)
-with sidebar:
-    st.sidebar.title("Conversations")
-    
-    if st.sidebar.button("+ New Chat", use_container_width=True):
+# Sidebar for model selection and conversation management
+with st.sidebar:
+    st.title("Conversations")
+    if st.button("+ New Chat"):
         create_new_chat()
         st.rerun()
-    
-    st.sidebar.markdown("---")
-    
-    if not openai_api_key:
-        st.sidebar.info("⚠️ No OpenAI API key found.", icon="ℹ️")
-        entered_token = st.sidebar.text_input("Enter OpenAI API Key", type="password")
-        if entered_token:
-            openai_api_key = entered_token
-            openai.api_key = openai_api_key
-    
-    st.sidebar.subheader("Model Selection")
-    model_options = list(AVAILABLE_MODELS.keys())
-    selected_model = st.sidebar.selectbox("Choose a model:", model_options)
+
+    st.markdown("---")
+    st.subheader("Model Selection")
+    selected_model = st.selectbox(
+        "Choose a model:",
+        list(AVAILABLE_MODELS.keys()),
+        index=list(AVAILABLE_MODELS.keys()).index(st.session_state.selected_model)
+    )
     st.session_state.selected_model = selected_model
 
-    st.sidebar.markdown("---")
+    st.markdown("---")
 
-# Main content area
-with main_content:
-    st.write("")  # Add some space at the top
-    
+# Main chat window
+with st.container():
     current_id = st.session_state.current_conversation_id
     current_conv = st.session_state.conversations.get(current_id, {"messages": []})
     messages = current_conv["messages"]
     
-    chat_container = st.container()
-    
-    with chat_container:
-        for message in messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
+    for message in messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
     
     prompt = st.chat_input("What's on your mind?")
     
     if prompt:
         messages.append({"role": "user", "content": prompt})
-        
         with st.chat_message("user"):
             st.markdown(prompt)
         
         with st.chat_message("assistant"):
-            message_placeholder = st.empty()
             full_response = get_ai_response(prompt, messages[:-1])
-            displayed_response = ""
-            for i in range(len(full_response)):
-                displayed_response += full_response[i]
-                message_placeholder.markdown(displayed_response + "▌")
-                time.sleep(0.005)
-            message_placeholder.markdown(full_response)
+            st.markdown(full_response)
             messages.append({"role": "assistant", "content": full_response})
             st.rerun()