Update app.py

app.py

@@ -1,156 +1,163 @@
-import streamlit as st
-import requests
-import json
-import time
-
-# Define API Key and Base URL
-API_KEY = "tauHGktuvULkGzYof5jKdVCNfZZryj32"
-API_BASE_URL = "https://oapi.tasking.ai/v1"
-MODEL_ID = "X5lMcszUYZvPpFnigD8mCfPI"
-
-# Streamlit interface setup
-st.set_page_config(page_title="Custom Assistant", page_icon="🤖", layout="centered")
-st.title("🤖 AI-Buddy Assistant")
-
-# Custom CSS to increase font size and make the sidebar interactive
-st.markdown("""
-    <style>
-        .sidebar .sidebar-content {
-            font-size: 1.3rem;
-            color: #333333;
-            background-color: #f9f9f9;
-            padding: 20px;
-            border-radius: 10px;
-        }
-        .sidebar .sidebar-content h1 {
-            font-size: 1.8rem;
-            font-weight: bold;
-            color: #ff4b4b;
-            margin-bottom: 15px;
-        }
-        .sidebar .sidebar-content select, .sidebar .sidebar-content textarea, .sidebar .sidebar-content input {
-            border: 2px solid #ff4b4b;
-            padding: 12px;
-            font-size: 1.2rem;
-            margin-bottom: 15px;
-            border-radius: 5px;
-        }
-        .sidebar .sidebar-content button {
-            background-color: #ff4b4b;
-            color: white;
-            font-size: 1.2rem;
-            padding: 10px 20px;
-            border-radius: 5px;
-            border: none;
-            cursor: pointer;
-            transition: background-color 0.3s ease;
-        }
-        .sidebar .sidebar-content button:hover {
-            background-color: #ff6565;
-        }
-        .st-chat-message p {
-            font-size: 1.3rem;
-        }
-    </style>
-""", unsafe_allow_html=True)
-
-# Sidebar for AI in Profession
-st.sidebar.title("Want to know how AI helps in your profession and the role of AI-Buddy?")
-st.sidebar.write("Select your details below and discover the potential of AI in your career!")
-
-# Dropdown options for profession with 'Other' option
-professions = ["Software Engineer", "Data Scientist", "Marketing Specialist", "Financial Analyst", "Teacher", "Doctor", "Project Manager", "Consultant", "Business Analyst", "Other"]
-
-# Dropdown options for field with 'Other' option
-fields = ["IT", "Healthcare", "Education", "Finance", "Marketing", "Engineering", "Sales", "Human Resources", "Consulting", "Other"]
-
-# Dropdowns for profession and field
-profession = st.sidebar.selectbox("Choose Your Profession", professions)
-field = st.sidebar.selectbox("Choose Your Field/Domain", fields)
-
-# If 'Other' is selected, show a text input for custom entry
-if profession == "Other":
-    profession = st.sidebar.text_input("Please specify your profession")
-
-if field == "Other":
-    field = st.sidebar.text_input("Please specify your field")
-
-# Text area for additional description
-description = st.sidebar.text_area("About you (a short description)", placeholder="Briefly describe your role")
-
-# Submit button
-if st.sidebar.button("Submit"):
-    # Add input details as a message to chat history
-    st.session_state.messages.append(
-        {"role": "user", "content": f"My profession is {profession} in the {field} field. Here’s a bit about me: {description}. Tell me how AI and AI-Buddy can help me."}
-    )
-
-# Initialize chat history
-if "messages" not in st.session_state:
-    st.session_state.messages = [
-        {"role": "assistant", "content": "Hello! How can I assist you today?"}
-    ]
-
-# Function to make a request to the custom API
-def get_chat_response():
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {API_KEY}",
-    }
-    data = {
-        "model": MODEL_ID,
-        "messages": st.session_state.messages  # Send entire conversation history
-    }
-
-    try:
-        response = requests.post(f"{API_BASE_URL}/chat/completions", headers=headers, json=data)
-        
-        if response.status_code == 200:
-            response_json = response.json()
-            content = response_json["choices"][0]["message"]["content"]
-
-            try:
-                content_parsed = json.loads(content)
-                return content_parsed["choices"][0]["message"]["content"]
-            except json.JSONDecodeError:
-                return content
-        else:
-            raise Exception(f"API Error: {response.status_code} - {response.text}")
-
-    except Exception as e:
-        st.error("Re-enter the prompt")
-        print(f"Error: {e}")
-        return None
-
-# Function to simulate streaming effect for displaying assistant's response
-def display_response_streaming(response_content):
-    response_placeholder = st.empty()  # Create a placeholder for assistant's response
-    streaming_text = ""
-    for char in response_content:
-        streaming_text += char
-        response_placeholder.write(streaming_text)
-        time.sleep(0.05)  # Small delay to simulate typing effect
-
-# Chat interface - User Input
-if prompt := st.chat_input("Type your message"):
-    # Append user's message to the chat history
-    st.session_state.messages.append({"role": "user", "content": prompt})
-
-# Display chat history and generate assistant response
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.write(message["content"])
-
-# Generate assistant response if the last message is from the user
-if st.session_state.messages[-1]["role"] == "user":
-    with st.chat_message("assistant"):
-        with st.spinner("Thinking..."):
-            # Get assistant's response based on the entire conversation history
-            response_content = get_chat_response()
-            if response_content:
-                # Stream the assistant's response chunk by chunk
-                display_response_streaming(response_content)
-                # Append the assistant's response to the chat history
-                st.session_state.messages.append({"role": "assistant", "content": response_content})
-            else:
-                st.write("Sorry, Re-enter the prompt")
+import streamlit as st
+import requests
+import json
+import time
+
+# Define API Key and Base URL
+API_KEY = "tauHGktuvULkGzYof5jKdVCNfZZryj32"
+API_BASE_URL = "https://oapi.tasking.ai/v1"
+MODEL_ID = "X5lMcszUYZvPpFnigD8mCfPI"
+
+# Streamlit interface setup
+st.set_page_config(page_title="Custom Assistant", page_icon="🤖", layout="centered")
+st.markdown('<a href="#sidebar" style="font-size: 1.2rem; color: #ff4b4b; text-decoration: none;">⬅️ Go to Sidebar to know how AI helps in your profession</a>', unsafe_allow_html=True)
+st.title("🤖 AI-Buddy Assistant")
+
+# Custom CSS to increase font size and make sidebar mobile-friendly
+st.markdown("""
+    <style>
+        .sidebar .sidebar-content {
+            font-size: 1.1rem;
+            color: #333333;
+            background-color: #f9f9f9;
+            padding: 15px;
+            border-radius: 8px;
+        }
+        .sidebar .sidebar-content h1 {
+            font-size: 1.5rem;
+            font-weight: bold;
+            color: #ff4b4b;
+            margin-bottom: 10px;
+        }
+        .sidebar .sidebar-content select, .sidebar .sidebar-content textarea, .sidebar .sidebar-content input {
+            border: 2px solid #ff4b4b;
+            padding: 10px;
+            font-size: 1.1rem;
+            margin-bottom: 10px;
+            border-radius: 5px;
+        }
+        .sidebar .sidebar-content button {
+            background-color: #ff4b4b;
+            color: white;
+            font-size: 1.1rem;
+            padding: 8px 15px;
+            border-radius: 5px;
+            border: none;
+            cursor: pointer;
+            transition: background-color 0.3s ease;
+        }
+        .sidebar .sidebar-content button:hover {
+            background-color: #ff6565;
+        }
+        .st-chat-message p {
+            font-size: 1.1rem;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+# Sidebar for AI in Profession
+st.sidebar.title("Want to know how AI helps in your profession and the role of AI-Buddy?")
+st.sidebar.write("Select your details below and discover the potential of AI in your career!")
+
+# Dropdown options for profession with 'Other' option
+professions = ["Software Engineer", "Data Scientist", "Marketing Specialist", "Financial Analyst", "Teacher", "Doctor", "Project Manager", "Consultant", "Business Analyst", "Other"]
+
+# Dropdown options for field with 'Other' option
+fields = ["IT", "Healthcare", "Education", "Finance", "Marketing", "Engineering", "Sales", "Human Resources", "Consulting", "Other"]
+
+# Mandatory Dropdowns for profession and field
+profession = st.sidebar.selectbox("Choose Your Profession", professions, key="profession")
+field = st.sidebar.selectbox("Choose Your Field/Domain", fields, key="field")
+
+# If 'Other' is selected, require text input for custom profession/field
+if profession == "Other":
+    profession = st.sidebar.text_input("Please specify your profession", key="custom_profession")
+if field == "Other":
+    field = st.sidebar.text_input("Please specify your field", key="custom_field")
+
+# Text area for additional description
+description = st.sidebar.text_area("About you (a short description)", placeholder="Briefly describe your role", key="description")
+
+# Submit button with mandatory field checks
+if st.sidebar.button("Submit"):
+    # Check if all required fields are filled
+    if not profession or not field or (profession == "Other" and not st.session_state.custom_profession) or (field == "Other" and not st.session_state.custom_field):
+        st.sidebar.error("Please fill in all required fields.")
+    else:
+        # Add input details as a message to chat history
+        st.session_state.messages.append(
+            {"role": "user", "content": f"My profession is {profession} in the {field} field. Here’s a bit about me: {description}. Tell me how AI and AI-Buddy can help me."}
+        )
+        # Clear the sidebar content to simulate sidebar closure
+        st.sidebar.empty()
+        st.sidebar.write("Details submitted! You can continue chatting below.")
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = [
+        {"role": "assistant", "content": "Hello! How can I assist you today?"}
+    ]
+
+# Function to make a request to the custom API
+def get_chat_response():
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_KEY}",
+    }
+    data = {
+        "model": MODEL_ID,
+        "messages": st.session_state.messages  # Send entire conversation history
+    }
+
+    try:
+        response = requests.post(f"{API_BASE_URL}/chat/completions", headers=headers, json=data)
+        
+        if response.status_code == 200:
+            response_json = response.json()
+            content = response_json["choices"][0]["message"]["content"]
+
+            try:
+                content_parsed = json.loads(content)
+                return content_parsed["choices"][0]["message"]["content"]
+            except json.JSONDecodeError:
+                return content
+        else:
+            raise Exception(f"API Error: {response.status_code} - {response.text}")
+
+    except Exception as e:
+        st.error("Re-enter the prompt")
+        print(f"Error: {e}")
+        return None
+
+# Function to simulate streaming effect for displaying assistant's response
+def display_response_streaming(response_content):
+    response_placeholder = st.empty()  # Create a placeholder for assistant's response
+    streaming_text = ""
+    for char in response_content:
+        streaming_text += char
+        response_placeholder.write(streaming_text)
+        time.sleep(0.05)  # Small delay to simulate typing effect
+
+# Chat interface - User Input
+if prompt := st.chat_input("Type your message"):
+    # Append user's message to the chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+
+# Display chat history and generate assistant response
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.write(message["content"])
+
+# Generate assistant response if the last message is from the user
+if st.session_state.messages[-1]["role"] == "user":
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            # Get assistant's response based on the entire conversation history
+            response_content = get_chat_response()
+            if response_content:
+                # Stream the assistant's response chunk by chunk
+                display_response_streaming(response_content)
+                # Append the assistant's response to the chat history
+                st.session_state.messages.append({"role": "assistant", "content": response_content})
+            else:
+                st.write("Sorry, Re-enter the prompt")