Code cleanup

app.py

@@ -13,6 +13,9 @@ if 'fixer_conversation' not in st.session_state:
     st.session_state['fixer_conversation'] = fixer_conversation
 
 def message_writer(input_text, response):
+    '''
+        Writes the messages to the chat window.
+    '''
     messages = []
     current_message = ""
     current_is_user = True
@@ -70,16 +73,16 @@ def main():
         st.markdown("---")
 
     elif page == "Job Description Generator":
-        c1 = st.container()
-        c1.title("A Job Description Generating Chatbot")
-        c1.markdown(
+        container_one = st.container()
+        container_one.title("A Job Description Generating Chatbot")
+        container_one.markdown(
             "JobGPT is a chatbot that generates job descriptions. \
                 This is built just for demo purpose."
         )
-        input_text = c1.text_input(
+        input_text = container_one.text_input(
             "Prompt",
             "Hi, can you please help me generate an unbiased job description?")
-        button = c1.button("Send")
+        button = container_one.button("Send")
 
         st.sidebar.markdown("---")
         st.sidebar.markdown("Click on `new chat` to start a new chat. \
@@ -89,7 +92,7 @@ def main():
 
         if clear_session:
             st.session_state['generator_conversation'] = conversation
-        c1.markdown("---")
+        container_one.markdown("---")
 
         initial_message = "Hello, how can I help you?"
         message(initial_message)
@@ -99,15 +102,15 @@ def main():
                                                st.session_state['generator_conversation'])
             message_writer(input_text, response)
     elif page == "Job Description Fixer":
-        c2 = st.container()
-        c2.title("JobGPT: A Job Description Fixing Chatbot")
-        c2.markdown(
+        container_two = st.container()
+        container_two.title("JobGPT: A Job Description Fixing Chatbot")
+        container_two.markdown(
             "JobGPT is a chatbot that fixes job descriptions. This is built just for demo purpose."
         )
-        input_text = c2.text_area(
+        input_text = container_two.text_area(
             "Prompt",
             "Hi, can you please help me fix my job description? It's biased.")
-        button = c2.button("Send")
+        button = container_two.button("Send")
 
         st.sidebar.markdown("---")
         st.sidebar.markdown("Click on `new chat` to start a new chat. \
@@ -117,7 +120,7 @@ def main():
 
         if clear_session:
             st.session_state['fixer_conversation'] = fixer_conversation
-        c2.markdown("---")
+        container_two.markdown("---")
 
         initial_message = "Hello, how can I help you?"
         message(initial_message)