better design

app.py

@@ -28,54 +28,139 @@ logger = logging.getLogger(__name__)
 profile_extractor=pe.ProfileExtractor()
 grammar_corrector=gc.GrammarCorrector()
 def collect_missing_data(profile):
-    """
-    Collects missing data from user input when automatic extraction fails.
-    
-    Args:
-        profile: The profile object with potentially missing data
-    
-    Returns:
-        Updated profile object
-    """
-    st.subheader("Complete Your Profile")
-    st.write("Please fill in any missing information below:")
+    """Collects missing data from user input using a step-by-step form"""
     
-    # Profile image upload
-    st.subheader("Profile Image")
-    uploaded_file = st.file_uploader("Upload a profile image (optional)", type=["jpg", "jpeg", "png"])
-    if uploaded_file is not None:
-        # Convert the file to base64 for storage
-        bytes_data = uploaded_file.getvalue()
-        encoded = base64.b64encode(bytes_data).decode()
-        profile.profileImg = f"data:image/{uploaded_file.type.split('/')[-1]};base64,{encoded}"
-        st.image(uploaded_file, caption="Profile Image Preview", width=150)
+    # Initialize step in session state if not exists
+    if 'form_step' not in st.session_state:
+        st.session_state.form_step = 1
     
-    # Essential information
-    st.subheader("Basic Information")
-    profile.name = st.text_input("Your Full Name:", value=profile.name if profile.name != "N/A" else "")
-    profile.title = st.text_input("Your Professional Title:", value=profile.title if profile.title != "N/A" else "")
-    profile.email = st.text_input("Your Email Address:", value=profile.email if profile.email != "N/A" else "")
-    profile.bio = st.text_area("Professional Bio (50-100 words):", value=profile.bio if profile.bio != "N/A" else "")
+    # Progress bar
+    total_steps = 6
+    progress = st.progress((st.session_state.form_step - 1) / total_steps)
     
-    if profile.bio and profile.bio != "N/A":
-        if st.button("Improve Bio Grammar"):
-            profile.bio = grammar_corrector.correct_grammar(profile.bio)
-            st.success("Grammar corrected!")
+    # Step navigation
+    st.write(f"Step {st.session_state.form_step} of {total_steps}")
     
-    # Optional information
-    profile.tagline = st.text_input("Professional Tagline (short catchy phrase):", value=profile.tagline if profile.tagline else "")
+    # Create columns for navigation buttons
+    col1, col2, col3 = st.columns([1, 2, 1])
     
-    # Social media
-    if not profile.social:
-        profile.social = SocialMedia()
+    with col1:
+        if st.session_state.form_step > 1:
+            if st.button("← Previous"):
+                st.session_state.form_step -= 1
+                st.rerun()
+                
+    with col3:
+        if st.session_state.form_step < total_steps:
+            if st.button("Next →"):
+                st.session_state.form_step += 1
+                st.rerun()
+
+    # Step content
+    if st.session_state.form_step == 1:
+        st.subheader("📸 Profile Image & Basic Information")
         
-    st.subheader("Social Media Links")
-    profile.social.linkedin = st.text_input("LinkedIn URL:", value=profile.social.linkedin if profile.social and profile.social.linkedin else "")
-    profile.social.github = st.text_input("GitHub URL:", value=profile.social.github if profile.social and profile.social.github else "")
-    profile.social.instagram = st.text_input("Instagram URL:", value=profile.social.instagram if profile.social and profile.social.instagram else "")
-    
-    # Education
-    st.subheader("Education")
+        # Profile image upload with preview
+        uploaded_file = st.file_uploader("Upload a profile image", type=["jpg", "jpeg", "png"])
+        if uploaded_file:
+            bytes_data = uploaded_file.getvalue()
+            encoded = base64.b64encode(bytes_data).decode()
+            profile.profileImg = f"data:image/{uploaded_file.type.split('/')[-1]};base64,{encoded}"
+            st.image(uploaded_file, caption="Profile Preview", width=150)
+        
+        # Basic info with validation
+        profile.name = st.text_input("Full Name*", value=profile.name if profile.name != "N/A" else "", 
+                                   help="Enter your full name as you want it to appear on your portfolio")
+        if not profile.name:
+            st.warning("Name is required")
+
+        profile.title = st.text_input("Professional Title*", value=profile.title if profile.title != "N/A" else "")
+        if not profile.title:
+            st.warning("Professional title is required")
+
+    elif st.session_state.form_step == 2:
+        st.subheader("📝 Bio & Contact")
+        profile.email = st.text_input("Email Address*", value=profile.email if profile.email != "N/A" else "",
+                                    help="This will be used as your contact email")
+        
+        profile.bio = st.text_area("Professional Bio*", value=profile.bio if profile.bio != "N/A" else "",
+                                 help="Write a brief professional summary (50-100 words)")
+        
+        if profile.bio:
+            if st.button("✨ Improve Bio Grammar"):
+                profile.bio = grammar_corrector.correct_grammar(profile.bio)
+                st.success("Grammar corrected!")
+
+    elif st.session_state.form_step == 3:
+        st.subheader("🎓 Education")
+        education_editor(profile)
+
+    elif st.session_state.form_step == 4:
+        st.subheader("💼 Work Experience")
+        experience_editor(profile)
+
+    elif st.session_state.form_step == 5:
+        st.subheader("🚀 Projects")
+        project_editor(profile)
+
+    elif st.session_state.form_step == 6:
+        st.subheader("🎯 Skills & Social Media")
+        skills_editor(profile)
+        social_media_editor(profile)
+
+    return profile
+
+def display_success_message(inserted_id, frontend_url):
+    """Displays an enhanced success message with portfolio links"""
+    
+    st.balloons()  # Add some celebration!
+    
+    success_container = st.container()
+    with success_container:
+        st.markdown("""
+        <div style='padding: 20px; border-radius: 10px; border: 2px solid #28a745; background-color: #f8f9fa;'>
+            <h2 style='color: #28a745;'>🎉 Congratulations!</h2>
+            <p>Your professional portfolio has been created successfully.</p>
+        </div>
+        """, unsafe_allow_html=True)
+        
+        st.markdown("### 🔗 Your Portfolio Links")
+        
+        # Portfolio link
+        if frontend_url:
+            portfolio_url = f"{frontend_url}/{inserted_id}"
+            st.markdown(f"""
+            - 🌟 **Main Portfolio:** [View Your Portfolio]({portfolio_url})
+            - 🆔 **Portfolio ID:** `{inserted_id}`
+            """)
+        
+        # API endpoints in expandable section
+        with st.expander("🔧 API Endpoints"):
+            api_url = f"{settings.PORTFOLIO_URL}/{inserted_id}"
+            st.markdown(f"""
+            - **Profile Data:** `{api_url}`
+            - **Profile Image:** `{api_url}/image`
+            """)
+
+        # Share buttons
+        st.markdown("### 📤 Share Your Portfolio")
+        col1, col2, col3 = st.columns(3)
+        with col1:
+            if frontend_url:
+                st.markdown(f"""
+                <a href="https://twitter.com/intent/tweet?text=Check%20out%20my%20professional%20portfolio!&url={frontend_url}/{inserted_id}" target="_blank">
+                    Share on Twitter
+                </a>
+                """, unsafe_allow_html=True)
+        with col2:
+            if frontend_url:
+                st.markdown(f"""
+                <a href="https://www.linkedin.com/sharing/share-offsite/?url={frontend_url}/{inserted_id}" target="_blank">
+                    Share on LinkedIn
+                </a>
+                """, unsafe_allow_html=True)
+
+def education_editor(profile):
     education_data = []
     
     # Display existing education entries with edit options
@@ -124,9 +209,64 @@ def collect_missing_data(profile):
     for edu_data in education_data:
         if edu_data["school"]:  # Only add if school is provided
             profile.educations.append(Education(**edu_data))
+
+def experience_editor(profile):
+    experience_data = []
+    
+    # Display existing experience entries with edit options
+    if profile.experiences:
+        for i, exp in enumerate(profile.experiences):
+            st.write(f"Experience #{i+1}")
+            company = st.text_input(f"Company {i+1}:", value=exp.company, key=f"company_{i}")
+            position = st.text_input(f"Position {i+1}:", value=exp.position, key=f"position_{i}")
+            start = st.text_input(f"Start Date {i+1}:", value=exp.startDate, key=f"exp_start_{i}")
+            end = st.text_input(f"End Date {i+1}:", value=exp.endDate if exp.endDate else "", key=f"exp_end_{i}")
+            description = st.text_area(f"Description {i+1}:", value=exp.description if exp.description else "", key=f"exp_desc_{i}")
+            
+            if company and position:  # Only add if company and position are provided
+                experience_data.append({
+                    "company": company,
+                    "position": position,
+                    "startDate": start,
+                    "endDate": end if end else None,
+                    "description": description if description else None
+                })
+    
+    # Option to add new experience entries
+    add_experience = st.checkbox("Add more work experience", key="add_exp")
+    if add_experience:
+        num_new_exp = st.number_input("Number of additional experiences:", min_value=1, max_value=10, value=1)
+        offset = len(profile.experiences) if profile.experiences else 0
+        
+        for i in range(int(num_new_exp)):
+            st.write(f"Additional Experience #{i+1}")
+            company = st.text_input(f"Company:", key=f"new_company_{offset+i}")
+            position = st.text_input(f"Position:", key=f"new_position_{offset+i}")
+            start = st.text_input(f"Start Date:", key=f"new_exp_start_{offset+i}")
+            end = st.text_input(f"End Date:", key=f"new_exp_end_{offset+i}")
+            description = st.text_area(f"Description:", key=f"new_exp_desc_{offset+i}")
+            
+            if company and position:  # Only add if company and position are provided
+                correct_grammar_btn = st.button(f"Correct Grammar for Experience #{i+1}")
+                if correct_grammar_btn and description:
+                    description = grammar_corrector.correct_grammar(description)
+                    st.success("Grammar corrected!")
+                    
+                experience_data.append({
+                    "company": company,
+                    "position": position,
+                    "startDate": start,
+                    "endDate": end if end else None,
+                    "description": description if description else None
+                })
     
-    # Projects
-    st.subheader("Projects")
+    # Update profile with experience data
+    profile.experiences = []
+    for exp_data in experience_data:
+        if exp_data["company"] and exp_data["position"]:  # Only add if company and position are provided
+            profile.experiences.append(Experience(**exp_data))
+
+def project_editor(profile):
     project_data = []
     
     # Display existing projects with edit options
@@ -181,10 +321,8 @@ def collect_missing_data(profile):
     for proj_data in project_data:
         if proj_data["title"] and proj_data["description"]:  # Only add if title and description are provided
             profile.projects.append(Project(**proj_data))
-    
-    # Skills with categories
-    st.subheader("Skills")
-    
+
+def skills_editor(profile):
     # Group existing skills by category
     technical_skills = [skill.name for skill in profile.skills if skill.category == Category.TECHNICAL]
     soft_skills = [skill.name for skill in profile.skills if skill.category == Category.SOFT_SKILLS]
@@ -222,65 +360,15 @@ def collect_missing_data(profile):
     add_skills_by_category(soft_skills_input, Category.SOFT_SKILLS)
     add_skills_by_category(domain_skills_input, Category.DOMAIN_KNOWLEDGE)
     add_skills_by_category(uncategorized_input, None)
-    
-    # Work Experience
-    st.subheader("Work Experience")
-    experience_data = []
-    
-    # Display existing experience entries with edit options
-    if profile.experiences:
-        for i, exp in enumerate(profile.experiences):
-            st.write(f"Experience #{i+1}")
-            company = st.text_input(f"Company {i+1}:", value=exp.company, key=f"company_{i}")
-            position = st.text_input(f"Position {i+1}:", value=exp.position, key=f"position_{i}")
-            start = st.text_input(f"Start Date {i+1}:", value=exp.startDate, key=f"exp_start_{i}")
-            end = st.text_input(f"End Date {i+1}:", value=exp.endDate if exp.endDate else "", key=f"exp_end_{i}")
-            description = st.text_area(f"Description {i+1}:", value=exp.description if exp.description else "", key=f"exp_desc_{i}")
-            
-            if company and position:  # Only add if company and position are provided
-                experience_data.append({
-                    "company": company,
-                    "position": position,
-                    "startDate": start,
-                    "endDate": end if end else None,
-                    "description": description if description else None
-                })
-    
-    # Option to add new experience entries
-    add_experience = st.checkbox("Add more work experience", key="add_exp")
-    if add_experience:
-        num_new_exp = st.number_input("Number of additional experiences:", min_value=1, max_value=10, value=1)
-        offset = len(profile.experiences) if profile.experiences else 0
+
+def social_media_editor(profile):
+    if not profile.social:
+        profile.social = SocialMedia()
         
-        for i in range(int(num_new_exp)):
-            st.write(f"Additional Experience #{i+1}")
-            company = st.text_input(f"Company:", key=f"new_company_{offset+i}")
-            position = st.text_input(f"Position:", key=f"new_position_{offset+i}")
-            start = st.text_input(f"Start Date:", key=f"new_exp_start_{offset+i}")
-            end = st.text_input(f"End Date:", key=f"new_exp_end_{offset+i}")
-            description = st.text_area(f"Description:", key=f"new_exp_desc_{offset+i}")
-            
-            if company and position:  # Only add if company and position are provided
-                correct_grammar_btn = st.button(f"Correct Grammar for Experience #{i+1}")
-                if correct_grammar_btn and description:
-                    description = grammar_corrector.correct_grammar(description)
-                    st.success("Grammar corrected!")
-                    
-                experience_data.append({
-                    "company": company,
-                    "position": position,
-                    "startDate": start,
-                    "endDate": end if end else None,
-                    "description": description if description else None
-                })
-    
-    # Update profile with experience data
-    profile.experiences = []
-    for exp_data in experience_data:
-        if exp_data["company"] and exp_data["position"]:  # Only add if company and position are provided
-            profile.experiences.append(Experience(**exp_data))
-    
-    return profile
+    st.subheader("Social Media Links")
+    profile.social.linkedin = st.text_input("LinkedIn URL:", value=profile.social.linkedin if profile.social and profile.social.linkedin else "")
+    profile.social.github = st.text_input("GitHub URL:", value=profile.social.github if profile.social and profile.social.github else "")
+    profile.social.instagram = st.text_input("Instagram URL:", value=profile.social.instagram if profile.social and profile.social.instagram else "")
 
 def display_profile(profile):
     """
@@ -506,6 +594,7 @@ def main():
                 
                 # Mark as saved in session state
                 st.session_state.profile_saved = True
+                st.session_state.inserted_id = inserted_id
                 
                 # Display the complete profile
                 display_profile(profile)
@@ -520,17 +609,16 @@ def main():
     
     # Final state - allow extracting another profile
     else:
-        st.success("Profile extraction complete!")
+        display_success_message(
+            inserted_id=st.session_state.get('inserted_id'),
+            frontend_url=os.environ.get("FRONTEND_URL", "")
+        )
         
-        # Show options to extract another profile or view the current one
-        if st.button("Extract Another Profile"):
-            # Reset session state
-            for key in ['profile', 'extraction_complete', 'user_input_complete', 'profile_saved']:
-                st.session_state[key] = False
+        if st.button("Create Another Profile"):
+            for key in ['profile', 'extraction_complete', 'user_input_complete', 'profile_saved', 'form_step']:
+                if key in st.session_state:
+                    del st.session_state[key]
             st.rerun()
-        else:
-            # Show the profile again
-            display_profile(st.session_state.profile)
 
 if __name__ == "__main__":
     main()