Added explainer home page

app.py

@@ -5,13 +5,36 @@ from job_description_generator import predict_job_description, get_job_descripti
 from job_description_fixer import fix_job_description, get_job_description_fixer_conversation
 
 conversation = get_job_description_conversation()
-if 'conversation' not in st.session_state:
-    st.session_state['conversation'] = conversation
+if 'generator_conversation' not in st.session_state:
+    st.session_state['generator_conversation'] = conversation
 
 fixer_conversation = get_job_description_fixer_conversation()
 if 'fixer_conversation' not in st.session_state:
     st.session_state['fixer_conversation'] = fixer_conversation
 
+def message_writer(input_text, response):
+    messages = []
+    current_message = ""
+    current_is_user = True
+    for historical_message in response['history']:
+        if "human" in historical_message.lower():
+            messages.append([current_message, current_is_user])
+            current_message = historical_message.replace("Human:", "")
+            current_is_user = True
+        elif "JobGPT" in historical_message:
+            messages.append([current_message, current_is_user])
+            current_message = historical_message.replace("JobGPT:", "")
+            current_is_user = False
+        else:
+            current_message = current_message + "\n" + historical_message
+    messages.append([current_message, current_is_user])
+    for message_to_send, is_user in messages:
+        if message_to_send.strip() != "":
+            message(message_to_send, is_user=is_user)
+    message(input_text, is_user=True)
+    message(response['prediction'], is_user=False)
+    return 0
+
 
 def setup():
     """
@@ -26,7 +49,6 @@ def setup():
     """
     st.markdown(hide_streamlit_style, unsafe_allow_html=True)
 
-
 def main():
     '''
      Main function of the app.
@@ -38,12 +60,21 @@ def main():
     st.sidebar.markdown("---")
     #selector
     page = st.sidebar.selectbox(
-        "Select a page", ["Job Description Generator", "Job Description Fixer"])
-    if page == "Job Description Generator":
+        "Select a page", ["Home", "Job Description Generator", "Job Description Fixer"])
+    if page == "Home":
+        st.title("JobGPT")
+        st.write("Select a page in the sidebar to get started.")
+        st.write("### Available options:")
+        st.write("1. Job Description Generator")
+        st.write("2. Job Description Fixer")
+        st.markdown("---")
+
+    elif page == "Job Description Generator":
         c1 = st.container()
-        c1.title("JobGPT: A Job Description Generating Chatbot")
+        c1.title("A Job Description Generating Chatbot")
         c1.markdown(
-            "JobGPT is a chatbot that generates job descriptions. This is built just for demo purpose."
+            "JobGPT is a chatbot that generates job descriptions. \
+                This is built just for demo purpose."
         )
         input_text = c1.text_input(
             "Prompt",
@@ -57,43 +88,17 @@ def main():
         clear_session = st.sidebar.button("New Chat")
 
         if clear_session:
-            st.session_state['conversation'] = conversation
+            st.session_state['generator_conversation'] = conversation
         c1.markdown("---")
 
         initial_message = "Hello, how can I help you?"
         message(initial_message)
 
         if button:
-            messages = []
-            current_message = ""
-            current_is_user = True
-            #clear prompt textbox
             response = predict_job_description(input_text,
-                                               st.session_state['conversation'])
-            for historical_message in response['history']:
-                if "human" in historical_message.lower():
-                    messages.append([current_message, current_is_user])
-                    current_message = historical_message.replace("Human:", "")
-                    current_is_user = True
-                    # message(historical_message.replace("Human:", ""),
-                    #         is_user=True)
-                elif "JobGPT" in historical_message:
-                    messages.append([current_message, current_is_user])
-                    current_message = historical_message.replace("JobGPT:", "")
-                    current_is_user = False
-                else:
-                    current_message = current_message + "\n" + historical_message
-                # if message_to_send != "":
-                #     message(message_to_send.replace("JobGPT:",
-                #                                     "").replace("Human:", ""),
-                #             is_user=is_user)
-            messages.append([current_message, current_is_user])
-            for message_to_send, is_user in messages:
-                if message_to_send.strip() != "":
-                    message(message_to_send, is_user=is_user)
-            message(input_text, is_user=True)
-            message(response['prediction'], is_user=False)
-    else:
+                                               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(
@@ -118,35 +123,8 @@ def main():
         message(initial_message)
 
         if button:
-            messages = []
-            current_message = ""
-            current_is_user = True
-            #clear prompt textbox
             response = fix_job_description(
                 input_text, st.session_state['fixer_conversation'])
-            for historical_message in response['history']:
-                if "human" in historical_message.lower():
-                    messages.append([current_message, current_is_user])
-                    current_message = historical_message.replace("Human:", "")
-                    current_is_user = True
-                    # message(historical_message.replace("Human:", ""),
-                    #         is_user=True)
-                elif "JobGPT" in historical_message:
-                    messages.append([current_message, current_is_user])
-                    current_message = historical_message.replace("JobGPT:", "")
-                    current_is_user = False
-                else:
-                    current_message = current_message + "\n" + historical_message
-                # if message_to_send != "":
-                #     message(message_to_send.replace("JobGPT:",
-                #                                     "").replace("Human:", ""),
-                #             is_user=is_user)
-            messages.append([current_message, current_is_user])
-            for message_to_send, is_user in messages:
-                if message_to_send.strip() != "":
-                    message(message_to_send, is_user=is_user)
-            message(input_text, is_user=True)
-            message(response['prediction'], is_user=False)
-
+            message_writer(input_text, response)
 
 main()