Update app.py

app.py

@@ -10,6 +10,7 @@ from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
 from streamlit_tags import st_tags
 from streamlit_vertical_slider import vertical_slider
 import pdf_generator
+import datetime
 
 # Set page config
 st.set_page_config(
@@ -54,12 +55,17 @@ if 'current_page' not in st.session_state:
 if 'answers' not in st.session_state:
     st.session_state.answers = {
         'workload_scope': {
-            'time_commitments': '',
-            'deadlines': '',
-            'feature_prioritization': ''
+            'feature_prioritization': ['', '', '', '', ''],
+            'preferred_start_period': '',
+            'team_composition': {
+                'partner': '',
+                'ibm': ''
+            }
         },
         'useful_assets': {
-            'assets': []
+            'solution_elements': [],
+            'ibm_software': [],
+            'open_source_supports': []
         },
         'pilot_evaluation': {
             'metrics': []
@@ -77,7 +83,7 @@ pages = [
         'input_type': 'custom'
     },
     {
-        'title': "Useful Assets and Libraries",
+        'title': "Solution Elements",
         'content': """
         Discuss the useful assets and libraries that can be utilized for the project. Consider Langchain, Langflow, ChromaDB, Jupyter Notebooks, etc.
         """,
@@ -130,62 +136,67 @@ if 'input_key' in current_page and current_page['input_key'] is not None:
         st.subheader("Workload Scope")
         col1, col2, col3 = st.columns(3)
         with col1:
-            st.session_state.answers['workload_scope']['time_commitments'] = st.text_area(
-                "Time Commitments:",
-                value=st.session_state.answers['workload_scope'].get('time_commitments', ""),
-                key="time_commitments",
-                height=150
-            )
+            st.session_state.answers['workload_scope']['feature_prioritization'] = [st.text_area(f"Feature Prioritization {i+1}", value=st.session_state.answers['workload_scope']['feature_prioritization'][i], height=100) for i in range(5)]
         with col2:
-            st.session_state.answers['workload_scope']['deadlines'] = st.text_area(
-                "Deadlines:",
-                value=st.session_state.answers['workload_scope'].get('deadlines', ""),
-                key="deadlines",
-                height=150
-            )
+            st.session_state.answers['workload_scope']['preferred_start_period'] = st.date_input("Preferred Start Period", value=(datetime.date.today(), datetime.date.today() + datetime.timedelta(days=30)))
         with col3:
-            st.session_state.answers['workload_scope']['feature_prioritization'] = st.text_area(
-                "Feature Prioritization:",
-                value=st.session_state.answers['workload_scope'].get('feature_prioritization', ""),
-                key="feature_prioritization",
-                height=150
-            )
+            st.session_state.answers['workload_scope']['team_composition']['partner'] = st.text_area("Partner", value=st.session_state.answers['workload_scope']['team_composition']['partner'], height=150)
+            st.session_state.answers['workload_scope']['team_composition']['ibm'] = st.text_area("IBM", value=st.session_state.answers['workload_scope']['team_composition']['ibm'], height=150)
 
     elif current_page['input_key'] == 'useful_assets':
-        st.subheader("Useful Assets and Libraries")
-        assets = st_tags(
-            label='Enter Useful Assets and Libraries:',
-            text='Press enter to add more',
-            value=st.session_state.answers['useful_assets'].get('assets', []),
-            suggestions=[],
-            maxtags=5,
-            key='useful_assets'
-        )
-        st.session_state.answers['useful_assets']['assets'] = assets
+        st.info(st.session_state.answers['workload_scope']['feature_prioritization'])
+        st.subheader("Solution Elements")
+        col1, col2 = st.columns(2)
+        with col1:
+            st.session_state.answers['useful_assets']['ibm_software'] = st_tags(label='Enter IBM Software:', text='Press enter to add more', value=st.session_state.answers['useful_assets']['ibm_software'], suggestions=[], maxtags=5, key='ibm_software')
+        with col2:
+            st.session_state.answers['useful_assets']['open_source_supports'] = st_tags(label='Enter Open Source Supports:', text='Press enter to add more', value=st.session_state.answers['useful_assets']['open_source_supports'], suggestions=[], maxtags=5, key='open_source_supports')
 
     elif current_page['input_key'] == 'pilot_evaluation':
-        st.subheader("Pilot Evaluation")
-        metrics = st_tags(
-            label='Enter Metrics to Track:',
-            text='Press enter to add more',
-            value=st.session_state.answers['pilot_evaluation'].get('metrics', []),
-            suggestions=[],
-            maxtags=5,
-            key='pilot_evaluation'
-        )
-        st.session_state.answers['pilot_evaluation']['metrics'] = metrics
+        st.subheader("Evaluation Criteria")
+        col1, col2 = st.columns(2)
+        with col1:
+            qualitative_criteria = st_tags(label='Enter Qualitative Criteria:', text='Press enter to add more', value=st.session_state.answers['useful_assets'].get('qualitative', []), suggestions=[], maxtags=5, key='qualitative_criteria')
+            st.session_state.answers['useful_assets']['qualitative'] = qualitative_criteria
+
+            # Add description fields for each qualitative criterion
+            for i, criterion in enumerate(qualitative_criteria):
+                description_key = f'qual_desc_{i}'
+                description = st.text_area(f"Description for {criterion}:", value=st.session_state.answers['useful_assets'].get(description_key, ""), key=description_key, height=100)
+                st.session_state.answers['useful_assets'][description_key] = description
+
+        with col2:
+            quantitative_criteria = st_tags(label='Enter Quantitative Criteria:', text='Use format: CriteriaName[min - max] (can use %, floats, or integers)', value=st.session_state.answers['useful_assets'].get('quantitative', []), suggestions=[], maxtags=5, key='quantitative_criteria')
+            st.session_state.answers['useful_assets']['quantitative'] = quantitative_criteria
+
+            if quantitative_criteria:
+                slider_cols = st.columns(len(quantitative_criteria))
+                for i, (criterion, slider_col) in enumerate(zip(quantitative_criteria, slider_cols)):
+                    parsed = pdf_generator.parse_quantitative_criteria(criterion)
+                    if parsed:
+                        name, min_val, max_val, is_percentage, is_integer = parsed
+                        current_value = st.session_state.answers['useful_assets'].get(f'quant_value_{i}', min_val)
+
+                        with slider_col:
+                            value = vertical_slider(label=name, key=f"quant_slider_{i}", min_value=min_val, max_value=max_val, step=(max_val - min_val) / 100, default_value=current_value, height=200, thumb_shape="circle", thumb_color="#9999FF", slider_color=('green', 'orange'), value_always_visible=True)
+
+                            st.session_state.answers['useful_assets'][f'quant_value_{i}'] = value
+
+                            if is_percentage:
+                                st.markdown(f"**{name}: {value*100:.1f}%**")
+                            elif is_integer:
+                                st.markdown(f"**{name}: {int(value)}**")
+                            else:
+                                st.markdown(f"**{name}: {value:.2f}**")
+                    else:
+                        st.warning(f"Invalid format for quantitative criterion: {criterion}")
+
 
 # Generate PDF button (only on the last page)
 if st.session_state.current_page == len(pages) - 1:
     if st.button("Generate and Download PDF", use_container_width=True):
         pdf = pdf_generator.generate_pdf(st.session_state)
-        st.download_button(
-            label="Download PDF",
-            data=pdf,
-            file_name="Pilot_Drafting.pdf",
-            mime="application/pdf",
-            use_container_width=True
-        )
+        st.download_button(label="Download PDF", data=pdf, file_name="Pilot_Drafting.pdf", mime="application/pdf", use_container_width=True)
 
 # Display progress
 st.progress((st.session_state.current_page + 1) / len(pages))