Update app.py

app.py

@@ -6,27 +6,23 @@ from chatbot import (
     save_chat_history
 )
 
-# Centered Title
 st.markdown("""
     <h1 style='text-align: center;'>Welcome, ChatGPT Clone</h1>
 """, unsafe_allow_html=True)
 
 client = get_googlegenai_client()
 
-# Initialize session state
 if "googlegenai_model" not in st.session_state:
     st.session_state["googlegenai_model"] = get_default_model()
 
 if "messages" not in st.session_state:
     st.session_state.messages = load_chat_history()
 
-# Sidebar: Clear history
 with st.sidebar:
     if st.button("Delete Chat History"):
         st.session_state.messages = []
         save_chat_history([])
 
-# Show chat history
 for msg in st.session_state.messages:
     role = msg["role"]
     content = msg["content"]
@@ -45,9 +41,7 @@ for msg in st.session_state.messages:
         """, unsafe_allow_html=True)
     st.markdown("<div style='clear: both'></div>", unsafe_allow_html=True)
 
-# Input field
 if prompt := st.chat_input("Ask anything"):
-    # Append user message
     st.session_state.messages.append({"role": "user", "content": prompt})
     st.markdown(f"""
     <div style='text-align: right; background-color: #f0f0f5; padding: 10px 15px; border-radius: 20px; margin: 10px 0; display: inline-block; max-width: 80%; float: right; clear: both;'>
@@ -55,7 +49,6 @@ if prompt := st.chat_input("Ask anything"):
     </div>
     """, unsafe_allow_html=True)
 
-    # Stream assistant response
     full_response = ""
     response_container = st.empty()
     response = client.stream(st.session_state.messages)
@@ -68,7 +61,6 @@ if prompt := st.chat_input("Ask anything"):
         </div>
         """, unsafe_allow_html=True)
 
-    # Final response (cleaned up)
     response_container.markdown(f"""
     <div style='text-align: left; margin: 10px 0; max-width: 80%; float: left; clear: both;'>
         {full_response}
@@ -76,6 +68,5 @@ if prompt := st.chat_input("Ask anything"):
     """, unsafe_allow_html=True)
     st.markdown("<div style='clear: both'></div>", unsafe_allow_html=True)
 
-    # Save assistant response
     st.session_state.messages.append({"role": "assistant", "content": full_response})
     save_chat_history(st.session_state.messages)