advance feature add

app.py

@@ -1,106 +1,3 @@
-'''
-import os
-import keyfile
-import warnings
-import streamlit as st
-from langchain_google_genai import ChatGoogleGenerativeAI
-from langchain.schema import HumanMessage, SystemMessage, AIMessage
-
-# Ignore warnings
-warnings.filterwarnings("ignore")
-
-# Streamlit settings
-st.set_page_config(page_title="🌿 ChitChat 🧙‍♀️", page_icon="🧙‍♀️", layout="wide")
-
-# Header
-st.markdown("<h1 style='text-align: center; color: #4B0082;'>Welcome to ChitChat 🌿✨</h1>", unsafe_allow_html=True)
-st.markdown("<h3 style='color: #003366;'>How can I assist with your ailments or worries today? 🧪💫</h3>", unsafe_allow_html=True)
-
-# Initialize session state for messages
-if "sessionMessages" not in st.session_state:
-    st.session_state.sessionMessages = [
-        SystemMessage(content="You are a medieval magical healer known for your peculiar sarcasm.")
-    ]
-
-# Set Google API key
-os.environ["GOOGLE_API_KEY"] = keyfile.GOOGLEKEY
-
-# Initialize the model
-llm = ChatGoogleGenerativeAI(
-    model="gemini-1.5-pro",
-    temperature=0.7,
-    convert_system_message_to_human=True
-)
-
-# Function to create chat bubbles
-def chat_bubble(message, is_user=True):
-    align = 'right' if is_user else 'left'
-    color = '#E1F5FE' if is_user else '#FFEBEE'
-    bubble_style = f"""
-    <div style="text-align: {align}; padding: 10px;">
-        <span style="display: inline-block; padding: 10px; background-color: {color}; color: black;
-                     border-radius: 15px; max-width: 70%; word-wrap: break-word;">
-            {message}
-        </span>
-    </div>
-    """
-    st.markdown(bubble_style, unsafe_allow_html=True)
-
-# Function to load answer from the model
-def load_answer(question):
-    # Add the user question to sessionMessages
-    st.session_state.sessionMessages.append(HumanMessage(content=question))
-    assistant_answer = llm.invoke(st.session_state.sessionMessages)
-    
-    # Ensure the answer is clean and formatted correctly
-    if isinstance(assistant_answer, AIMessage):
-        st.session_state.sessionMessages.append(assistant_answer)
-        return assistant_answer.content.strip()  # Strip any unnecessary whitespace or newlines
-    else:
-        st.session_state.sessionMessages.append(AIMessage(content=assistant_answer))
-        return assistant_answer.strip()  # Also strip here for consistency
-
-# Display the chat history
-for message in st.session_state.sessionMessages:
-    if isinstance(message, HumanMessage):
-        chat_bubble(message.content, is_user=True)
-    elif isinstance(message, AIMessage):
-        chat_bubble(message.content, is_user=False)
-
-# Input area for new prompt
-user_input = st.text_input("You: ", key="input", placeholder="Type your question here...")
-
-# Button for submission
-if st.button("🌟 Get a Magical Answer 🌟"):
-    if user_input:
-        chat_bubble(user_input, is_user=True)  # Display user input
-        response = load_answer(user_input)      # Get response
-        chat_bubble(response, is_user=False)    # Display AI response
-
-# Background styling
-st.markdown("""
-<style>
-    .stApp {
-        background: linear-gradient(to right, #FFEFBA, #FFFFFF);
-        color: #4B0082;
-        font-family: Arial, sans-serif;
-    }
-    input[type="text"] {
-        padding: 10px;
-        border: 2px solid #4B0082;
-        border-radius: 15px;
-        outline: none;
-        width: 100%;
-    }
-    button {
-        background-color: #4B0082;
-        color: white;
-        border-radius: 15px;
-        margin-top: 10px;
-    }
-</style>
-""", unsafe_allow_html=True)
-'''
 import os
 import keyfile
 import warnings
@@ -171,7 +68,16 @@ for message in st.session_state.sessionMessages:
 
 # Sidebar for additional options
 st.sidebar.header("Additional Options")
-mood = st.sidebar.selectbox("Select your mood:", ["Neutral", "Happy", "Concerned", "Curious"])
+
+# Mood selection with emojis
+mood_options = {
+    "😊": "Happy",
+    "😐": "Neutral",
+    "😟": "Concerned",
+    "🤔": "Curious"
+}
+mood = st.sidebar.selectbox("Select your mood:", list(mood_options.keys()))
+
 predefined_responses = st.sidebar.radio("Choose a predefined response:", 
     ["Tell me a joke", "Give me advice on health", "Suggest a potion", "General inquiry"])