Update app.py

app.py

@@ -17,51 +17,6 @@ if 'API_Key' not in st.session_state:
 
 # Step 3: Build the Streamlit UI
 st.set_page_config(page_title="Chat GPT Clone", page_icon=":robot_face:")
-
-# Custom CSS to center the conversation and responses
-st.markdown("""
-    <style>
-        .main-container {
-            display: flex;
-            justify-content: center;
-            align-items: center;
-            text-align: center;
-            flex-direction: column;
-            width: 100%;
-        }
-        .chat-container {
-            max-width: 800px;
-            width: 100%;
-            padding: 20px;
-            border-radius: 8px;
-            background-color: #2f2f2f;
-            color: white;
-            box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.3);
-            margin-top: 20px;
-        }
-        .message-container {
-            display: flex;
-            flex-direction: column;
-            gap: 10px;
-        }
-        .message {
-            padding: 10px;
-            border-radius: 5px;
-            margin: 5px;
-            max-width: 70%;
-        }
-        .user-message {
-            background-color: #4a90e2;
-            align-self: flex-start;
-        }
-        .ai-message {
-            background-color: #444;
-            align-self: flex-start;
-            margin-left: 15%;
-        }
-    </style>
-""", unsafe_allow_html=True)
-
 st.markdown("<h1 style='text-align: center;'>How can I assist you? </h1>", unsafe_allow_html=True)
 
 # Sidebar for API key input
@@ -106,15 +61,8 @@ with container:
             st.session_state['messages'].append(model_response)
 
 with response_container:
-    # Wrapping the chat in a container to center it
-    st.markdown('<div class="main-container">', unsafe_allow_html=True)
-    st.markdown('<div class="chat-container">', unsafe_allow_html=True)
-    
-    # Display the conversation
     for i in range(len(st.session_state['messages'])):
         if (i % 2) == 0:
-            message(st.session_state['messages'][i], is_user=True, key=str(i) + '_user', message_class="user-message")
+            message(st.session_state['messages'][i], is_user=True, key=str(i) + '_user')
         else:
-            message(st.session_state['messages'][i], key=str(i) + '_AI', message_class="ai-message")
-    
-    st.markdown('</div></div>', unsafe_allow_html=True)
+            message(st.session_state['messages'][i], key=str(i) + '_AI')