Update app.py

app.py

@@ -81,11 +81,21 @@ 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
-openai_api_key = os.getenv("OPENAI_API_KEY")
+openai_api_key = os.getenv("OPENAI_API_KEY_NEW_3")
+# For debugging in Hugging Face Spaces, you can temporarily uncomment this:
+# openai_api_key = "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"
 
 # Configure OpenAI client
 if openai_api_key:
-    openai.api_key = openai_api_key
+    client = openai.OpenAI(api_key=openai_api_key)
+    # Optional: Check available models
+    try:
+        available_models = client.models.list()
+        model_ids = [model.id for model in available_models.data]
+        if 'o3-mini-2025-01-31' not in model_ids:
+            st.sidebar.warning("The 'o3-mini-2025-01-31' model is not available with this API key.")
+    except Exception as e:
+        st.sidebar.error(f"Could not fetch available models: {str(e)}")
 
 # Available models with descriptions and token limits
 AVAILABLE_MODELS = {
@@ -113,101 +123,250 @@ AVAILABLE_MODELS = {
         "output_tokens": 1200,
         "temperature": 0.7
     },
-    "o3-mini": {
-        "description": "Latest cost-efficient reasoning model",
-        "max_tokens": 200000,
+    "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
     }
 }
 
 # Function to call OpenAI API
 def get_ai_response(prompt, history):
-    # Use demo mode if no API key is provided
     if not openai_api_key:
         return get_demo_response(prompt)
     
     try:
-        # Format messages for API
-        messages = []
-        
-        # Add system message
-        system_message = "You are a helpful assistant that provides clear, concise, and accurate information."
-        messages.append({
-            "role": "system", 
-            "content": system_message
-        })
-        
-        # Add conversation history
+        messages = [
+            {"role": "system", "content": "You are a helpful assistant that provides clear, concise, and accurate information."}
+        ]
         for msg in history:
-            messages.append({
-                "role": msg["role"],
-                "content": msg["content"]
-            })
+            messages.append({"role": msg["role"], "content": msg["content"]})
+        messages.append({"role": "user", "content": prompt})
         
-        # Add the current prompt
-        messages.append({
-            "role": "user",
-            "content": prompt
-        })
-        
-        # Get model configuration
         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"]
         
-        # Set reasoning effort for o3-mini model
-        reasoning_effort = "medium"  # Default reasoning effort
-        if model == "o3-mini":
-            reasoning_effort = st.sidebar.selectbox(
-                "Select reasoning effort:",
-                ["low", "medium", "high"],
-                index=1
-            )
-        
-        # Call OpenAI API
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model=model,
             messages=messages,
             temperature=temperature,
             max_tokens=output_tokens,
-            reasoning_effort=reasoning_effort if model == "o3-mini" else None,
             stream=False
         )
+        return response.choices[0].message.content
         
-        # Extract the response
-        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:
+        return f"Invalid request: {str(inv_err)}. The model '{model}' might not be available with your API key."
     except Exception as e:
-        st.error(f"An error occurred: {str(e)}")
-        return "I'm sorry, I encountered an error while processing your request. Please check your OpenAI API key or try again later."
+        return f"An error occurred: {str(e)}. Please try a different model or check your setup."
 
 # Demo mode responses for when no API key is available
 def get_demo_response(prompt):
     prompt_lower = prompt.lower()
-    
-    # Simple response templates
     greetings = [
         "Hello! How can I assist you today?",
         "Hi there! I'm a demo AI assistant. What can I help you with?",
         "Greetings! I'm running in demo mode. Feel free to ask simple questions."
     ]
-    
     farewells = [
         "Goodbye! Have a great day!",
         "Farewell! Come back soon!",
         "Take care! It was nice chatting with you."
     ]
-    
     info_responses = [
         "I'm a simple AI assistant running in demo mode. To use the full features, please provide an OpenAI API key.",
         "This is a demo version with limited capabilities. For a better experience, add your OpenAI API key.",
         "I'm just demonstrating basic functionality. Get a free API key from OpenAI to unlock my full potential!"
     ]
-    
     reasoning_examples = [
         "This is a demonstration of how I would process a reasoning task. In a real scenario with the full model, I would analyze the problem step by step, consider multiple angles, and provide a detailed explanation.",
-        "When solving problems, I would typically break them down into smaller parts, examine each component, and build
-::contentReference[oaicite:0]{index=0}
- 
+        "When solving problems, I would typically break them down into smaller parts, examine each component, and build towards a comprehensive solution. This demo just simulates that process.",
+        "Reasoning typically involves identifying key facts, applying logical rules, and drawing conclusions based on available information. With a proper API key, I could demonstrate this more effectively."
+    ]
+    
+    if any(word in prompt_lower for word in ["hello", "hi", "hey", "greetings"]):
+        return random.choice(greetings)
+    elif any(word in prompt_lower for word in ["bye", "goodbye", "farewell", "see you"]):
+        return random.choice(farewells)
+    elif any(phrase in prompt_lower for phrase in ["who are you", "what are you", "tell me about yourself", "what can you do"]):
+        return random.choice(info_responses)
+    elif any(word in prompt_lower for word in ["think", "reason", "analyze", "solve", "explain", "why", "how"]):
+        return random.choice(reasoning_examples)
+    elif "weather" in prompt_lower:
+        return "I'm sorry, I don't have access to real-time weather data in demo mode."
+    elif any(word in prompt_lower for word in ["help", "assist", "support"]):
+        return "To get better assistance, please add your OpenAI API key. You can get one for free at https://platform.openai.com/account/api-keys."
+    else:
+        return "I'm running in demo mode with limited responses. For a full conversation experience, please add your OpenAI API key above."
+
+# Function to create a new conversation
+def create_new_chat():
+    new_id = str(uuid.uuid4())
+    st.session_state.current_conversation_id = new_id
+    st.session_state.conversations[new_id] = {
+        "title": f"New chat {datetime.now().strftime('%H:%M')}",
+        "messages": []
+    }
+
+# Function to update conversation title based on first message
+def update_conversation_title(conv_id, user_message):
+    current_title = st.session_state.conversations[conv_id]["title"]
+    if current_title.startswith("New chat"):
+        new_title = user_message[:30] + "..." if len(user_message) > 30 else user_message
+        st.session_state.conversations[conv_id]["title"] = new_title
+
+# Function to delete a conversation
+def delete_conversation(conv_id):
+    if conv_id in st.session_state.conversations:
+        del st.session_state.conversations[conv_id]
+        if conv_id == st.session_state.current_conversation_id:
+            if st.session_state.conversations:
+                st.session_state.current_conversation_id = next(iter(st.session_state.conversations))
+            else:
+                create_new_chat()
+
+# 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):
+        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
+            client = openai.OpenAI(api_key=openai_api_key)
+    
+    st.sidebar.subheader("Model Selection")
+    model_options = list(AVAILABLE_MODELS.keys())
+    model_descriptions = [f"{model} - {AVAILABLE_MODELS[model]['description']}" for model in model_options]
+    selected_model_index = model_options.index(st.session_state.selected_model) if st.session_state.selected_model in model_options else 0
+    
+    selected_description = st.sidebar.selectbox(
+        "Choose a model:",
+        model_descriptions,
+        index=selected_model_index
+    )
+    
+    selected_model = model_options[model_descriptions.index(selected_description)]
+    if selected_model != st.session_state.selected_model:
+        st.session_state.selected_model = selected_model
+        st.sidebar.info(f"Model set to {selected_model}")
+    
+    with st.sidebar.expander("Model Details"):
+        model_info = AVAILABLE_MODELS[selected_model]
+        st.write(f"**Description:** {model_info['description']}")
+        st.write(f"**Max tokens:** {model_info['max_tokens']}")
+        st.write(f"**Default temperature:** {model_info['temperature']}")
+        st.write("""
+        **Note:** Some models may not be available with your current API key.
+        If you encounter an error, try selecting a different model.
+        """)
+    
+    st.sidebar.markdown("---")
+    
+    for conv_id, conv_data in st.session_state.conversations.items():
+        col1, col2 = st.sidebar.columns([4, 1])
+        is_active = conv_id == st.session_state.current_conversation_id
+        
+        with col1:
+            if st.button(
+                conv_data["title"], 
+                key=f"conv_{conv_id}", 
+                use_container_width=True,
+                type="secondary" if is_active else "tertiary"
+            ):
+                st.session_state.current_conversation_id = conv_id
+                st.rerun()
+                
+        with col2:
+            if st.button("🗑️", key=f"del_{conv_id}"):
+                delete_conversation(conv_id)
+                st.rerun()
+
+# Main content area
+with main_content:
+    st.write("")  # Add some space at the top
+    
+    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 i, message in enumerate(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})
+        
+        if len(messages) == 1:
+            update_conversation_title(current_id, 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)
+            messages.append({"role": "assistant", "content": full_response})
+            st.rerun()