Update Functions.py

Functions.py

@@ -10,7 +10,7 @@ from Prompts_and_Chains import LLMChains
 from Templates import json_structure
 import json
 from Utils import estimate_to_value
-
+from Utils import is_key_value_present
 
 class RFPProcessor:
     def __init__(self):
@@ -18,6 +18,41 @@ class RFPProcessor:
         self.openai_api_key = os.getenv("OPENAI_API_KEY")
         self.chains_obj = LLMChains()
 
+    def generate_roadmap(self):
+        
+        roadmap_data = self.chains_obj.roadmap_chain.run({"project_input":st.session_state["estimation_data_json"]})
+        print(roadmap_data)
+        st.session_state["roadmap_data_json"] = roadmap_data
+        data = json.loads(roadmap_data)
+        
+        try:
+          decoded_data = json.loads(roadmap_data)
+          print(decoded_data)
+        except json.decoder.JSONDecodeError as e:
+          print(f"JSON decoding error: {e}")
+        
+        for phases_data in data['phases']:
+          phase = phases_data["name"]
+          for milestone in phases_data['milestones']:
+               milestone_name = milestone["name"]
+               total_time = milestone["totalTime"]
+               for feature in milestone["features"]:
+                  featue_name = feature["name"]
+                  featue_rationale = feature["rationale"]
+                  featue_effort = feature["effort"]
+                  phase_key_present = is_key_value_present(st.session_state["roadmap_data"], "phases", phase)
+          
+                  if phase_key_present:
+                     milestone_key_present = is_key_value_present(st.session_state["roadmap_data"], "milestones", milestone_name)
+                     if milestone_key_present:
+                         st.session_state.roadmap_data.append({"phases": "","milestones": "","total_time": "","features": featue_name,"rational": featue_rationale,"effort": featue_effort,})
+                     else:
+                         st.session_state.roadmap_data.append({"phases": "","milestones": milestone_name,"total_time": total_time,"features": featue_name,"rational": featue_rationale,"effort": featue_effort,})
+                  else:
+                      st.session_state.roadmap_data.append({"phases": phase,"milestones": milestone_name,"total_time": total_time,"features": featue_name,"rational": featue_rationale,"effort": featue_effort,}) 
+              
+        st.session_state["is_roadmap_data_created"] = True
+
     def generate_estimations(self, tech_leads, senior_developers, junior_developers):
         print(st.session_state["user_stories_json"])
         inputs = {
@@ -29,6 +64,9 @@ class RFPProcessor:
         }
 
         data = self.chains_obj.estimations_chain.run(inputs)
+        
+        st.session_state["estimation_data_json"]= data
+        
         estimation_json_data = json.loads(data)
 
         for epic_data in estimation_json_data["epics"]:
@@ -45,7 +83,7 @@ class RFPProcessor:
                             "Estimate": story["estimate"],
                             "Person Days Range": story["effort"],
                             "Story Rationale": story["rationale"],
-                            "Person Days": average
+                            "Person Days": average,
                         }
                     )
         st.session_state["is_estimation_data_created"] = True