Refactor app.py to enhance the Streamlit UI for the AI Job Application Email Generator. Updated the update_ui function to accept parameters for message, cv_file, and cv_sections, improving the handling of user inputs. Reorganized the layout to ensure a smoother flow for CV uploads and job description input, while maintaining clear feedback for users. This change streamlines the email generation process and improves overall user experience.

app.py

@@ -172,28 +172,8 @@ CV Summary:
 # Streamlit UI section
 st.title("AI Job Application Email Generator")
 
-# Add tabs for different sections
-tab1, tab2 = st.tabs(["Generate Email", "View CV Details"])
-
-with tab1:
-    # CV file upload
-    cv_file = st.file_uploader("Upload CV (PDF or DOCX)", type=["pdf", "docx"])
-    
-    if cv_file:
-        cv_sections = extract_cv_text(cv_file)
-        if isinstance(cv_sections, dict):
-            st.success("CV uploaded and parsed successfully!")
-        else:
-            st.error(cv_sections)
-    
-    # Job description input
-    st.markdown("### Job Description")
-    message = st.text_area("Paste the job description here:", height=200)
-    
-    # Call the updated UI function
-    update_ui()
-
-def update_ui():
+def update_ui(message, cv_file, cv_sections):
+    """Handle the UI updates for email generation."""
     # Create placeholder for the generated email
     email_placeholder = st.empty()
     
@@ -215,6 +195,30 @@ def update_ui():
         else:
             st.warning("Please upload a CV and enter a job description.")
 
+# Add tabs for different sections
+tab1, tab2 = st.tabs(["Generate Email", "View CV Details"])
+
+with tab1:
+    # CV file upload
+    cv_file = st.file_uploader("Upload CV (PDF or DOCX)", type=["pdf", "docx"])
+    
+    if cv_file:
+        cv_sections = extract_cv_text(cv_file)
+        if isinstance(cv_sections, dict):
+            st.success("CV uploaded and parsed successfully!")
+        else:
+            st.error(cv_sections)
+            cv_sections = None
+    else:
+        cv_sections = None
+    
+    # Job description input
+    st.markdown("### Job Description")
+    message = st.text_area("Paste the job description here:", height=200)
+    
+    # Call the updated UI function with parameters
+    update_ui(message, cv_file, cv_sections)
+
 with tab2:
     if cv_file and isinstance(cv_sections, dict):
         st.markdown("### Parsed CV Details")