Update app.py

app.py

@@ -1,302 +1,107 @@
-import streamlit as st
-import requests
-from dotenv import load_dotenv
-import os
-import sys
-import time
-import random
-import uuid
-from datetime import datetime
-import openai
-
-# Load environment variables
-load_dotenv()
-
-# Set page config with a wider layout
-st.set_page_config(
-    page_title="GPT-Style Chat Assistant",
-    page_icon="🤖",
-    layout="wide"
-)
-
-# Add custom CSS for better styling
-st.markdown("""
-<style>
-    .main-content {
-        max-width: 800px;
-        margin: 0 auto;
-        padding: 1rem;
-    }
-    .chat-message {
-        padding: 1.5rem;
-        border-radius: 0.5rem;
-        margin-bottom: 1rem;
-        display: flex;
-        flex-direction: column;
-    }
-    .user-message {
-        background-color: #f0f2f6;
-    }
-    .assistant-message {
-        background-color: #e6f3f7;
-    }
-    .chat-input {
-        position: fixed;
-        bottom: 0;
-        width: 100%;
-        padding: 1rem;
-        background-color: white;
-    }
-    .sidebar-conv {
-        padding: 0.5rem 1rem;
-        border-radius: 0.5rem;
-        margin-bottom: 0.5rem;
-        cursor: pointer;
-        transition: background-color 0.3s;
-    }
-    .sidebar-conv:hover {
-        background-color: #f0f2f6;
-    }
-    .active-conv {
-        background-color: #e6f3f7;
-        font-weight: bold;
-    }
-</style>
-""", unsafe_allow_html=True)
-
-# 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
-    st.session_state.conversations[new_id] = {
-        "title": f"New chat {datetime.now().strftime('%H:%M')}",
-        "messages": []
-    }
-
-# Get OpenAI API key from environment or let user enter it
-openai_api_key = os.getenv("OPENAI_API_KEY")
-
-# Configure OpenAI client
-if openai_api_key:
-    client = openai.OpenAI(api_key=openai_api_key)
-
-# 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 OpenAI
-        messages = []
-        
-        # Add system message
-        messages.append({
-            "role": "system", 
-            "content": "You are a helpful assistant that provides clear, concise, and accurate information."
-        })
-        
-        # Add conversation history
-        for msg in history:
-            messages.append({
-                "role": msg["role"],
-                "content": msg["content"]
-            })
-        
-        # Add the current prompt
-        messages.append({
-            "role": "user",
-            "content": prompt
-        })
-        
-        # Call OpenAI API
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
-            messages=messages,
-            temperature=0.7,
-            max_tokens=500,
-            stream=False
-        )
-        
-        # Extract the response
-        return response.choices[0].message.content
-        
-    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."
-
-# 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 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."
-    ]
-    
-    # Simple pattern matching
-    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"):
-        # Limit title length to prevent overflow
-        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 we deleted the current conversation, set a new one
-        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")
-    
-    # Add a new chat button
-    if st.sidebar.button("+ New Chat", use_container_width=True):
-        create_new_chat()
-        st.rerun()
-    
-    st.sidebar.markdown("---")
-    
-    # API token input in sidebar if not available
-    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.markdown("---")
-    
-    # Display conversation history
-    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
-    
-    # Get current conversation
-    current_id = st.session_state.current_conversation_id
-    current_conv = st.session_state.conversations.get(current_id, {"messages": []})
-    messages = current_conv["messages"]
-    
-    # Create a container for the chat area (scrollable)
-    chat_container = st.container()
-    
-    # Display chat messages
-    with chat_container:
-        for i, message in enumerate(messages):
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
-    
-    # Chat input at the bottom
-    prompt = st.chat_input("What's on your mind?")
-    
-    if prompt:
-        # Add user message to the current conversation
-        messages.append({"role": "user", "content": prompt})
-        
-        # Update conversation title if this is the first message
-        if len(messages) == 1:
-            update_conversation_title(current_id, prompt)
-        
-        # Display user message
-        with st.chat_message("user"):
-            st.markdown(prompt)
-        
-        # Display assistant response with typing animation
-        with st.chat_message("assistant"):
-            message_placeholder = st.empty()
-            
-            # Get response from AI
-            full_response = get_ai_response(prompt, messages[:-1])
-            
-            # Simulate typing
-            displayed_response = ""
-            for i in range(len(full_response)):
-                displayed_response += full_response[i]
-                message_placeholder.markdown(displayed_response + "▌")
-                time.sleep(0.005)  # Slightly faster typing
-            
-            # Display final response
-            message_placeholder.markdown(full_response)
-            
-            # Add assistant response to the conversation
-            messages.append({"role": "assistant", "content": full_response})
-            
-            # Force a rerun to update the sidebar with the new conversation title
-            st.rerun() 
+---
+title: GPT-Style Chat Assistant
+emoji: 🤖
+colorFrom: blue
+colorTo: green
+sdk: streamlit
+sdk_version: 1.24.0
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+# GPT-Style Chat Assistant
+
+A sophisticated Streamlit app that mimics the ChatGPT interface, using OpenAI's GPT models for advanced reasoning capabilities.
+
+![Screenshot](https://i.ibb.co/bJfyXXn/gpt-style-assistant.png)
+
+## Features
+
+- **ChatGPT-like interface** with conversation sidebar
+- **Multiple conversation management**:
+  - Create new chats
+  - Switch between conversations
+  - Delete old conversations
+- **Model selection** - choose between different OpenAI models:
+  - GPT-3.5-Turbo (default)
+  - GPT-4
+  - GPT-3.5-Turbo-16k (for longer conversations)
+  - GPT-4-Turbo
+- **Conversation history** preserved between sessions
+- **Demo mode** that works without an API key
+- **Responsive design** that adapts to different screen sizes
+
+## Setup
+
+1. Clone this repository
+2. Install the required packages:
+   ```
+   pip install -r requirements.txt
+   ```
+3. (Required for full functionality) Get an OpenAI API key:
+   - Create an account at [OpenAI](https://platform.openai.com/signup)
+   - Go to [API Keys](https://platform.openai.com/account/api-keys)
+   - Create a new API key
+   - Add it to a `.env` file in the root directory:
+     ```
+     OPENAI_API_KEY=your_api_key_here
+     ```
+
+## Running Locally
+
+Run the app with the following command:
+
+```
+streamlit run app.py
+```
+
+If you're having trouble with the `streamlit` command not being found, you can use the full path to the executable:
+```
+python -m streamlit run app.py
+```
+
+Or use the provided runner script:
+```
+python run_app.py
+```
+
+This will start the app and open it in your default browser.
+
+## Demo Mode
+
+The app includes a demo mode that works without an API key. In this mode:
+- The assistant will provide pre-defined responses to common questions
+- You'll see a text field in the sidebar where you can optionally enter an OpenAI API key
+- For a full chatbot experience with reasoning capabilities, it's recommended to add your OpenAI API key
+
+## Deploying to Hugging Face Spaces
+
+1. Create a new Space on Hugging Face Spaces (https://huggingface.co/spaces)
+2. Choose Streamlit as the SDK
+3. Link your GitHub repository or upload the files directly
+4. Add your OpenAI API key as a secret:
+   - Go to Settings > Repository Secrets
+   - Add a secret named `OPENAI_API_KEY` with your key
+
+The app will automatically deploy and be available at your Hugging Face Spaces URL.
+
+## About the Model
+
+This app lets you select from multiple OpenAI models:
+
+- **GPT-3.5-Turbo**: Fast and cost-effective, suitable for most tasks
+- **GPT-4**: More capable but slower and more expensive
+- **GPT-3.5-Turbo-16k**: Supports longer conversations with extended context window
+- **GPT-4-Turbo**: OpenAI's most advanced model with significantly larger context window
+
+You can select your preferred model in the sidebar. Note that model availability depends on your OpenAI API key permissions and subscription tier.
+
+## Customization
+
+You can easily customize the app by:
+- Changing the model parameters in the `AVAILABLE_MODELS` dictionary
+- Adjusting the styling in the CSS section
+- Adding more features to the sidebar
+- Creating additional response templates for the demo mode