pretty print json objects

crew/til.py

@@ -3,6 +3,7 @@ from langchain_core.messages import SystemMessage
 from pydantic import BaseModel, Field
 from langchain_core.output_parsers import JsonOutputParser
 from langchain_openai import ChatOpenAI
+import pprint
 
 HIGH_IMPACT_THRESHOLD = 8
 LOW_IMPACT_THRESHOLD = 7
@@ -20,7 +21,6 @@ class TilCrew:
                 "feedback": "not_ok",
                 "feedback_criteria": "factuality_feedback",
                 "reason": self.feedback["factuality_reason"],
-                "suggestion": self.feedback["final_suggestion"],
             }
 
         if self.feedback["insightful_score"] < HIGH_IMPACT_THRESHOLD:
@@ -28,7 +28,6 @@ class TilCrew:
                 "feedback": "not_ok",
                 "feedback_criteria": "insightful_feedback",
                 "reason": self.feedback["insightful_reason"],
-                "suggestion": self.feedback["final_suggestion"],
             }
 
         if self.feedback["simplicity_score"] < LOW_IMPACT_THRESHOLD:
@@ -53,9 +52,10 @@ class TilCrew:
 
     def _gather_feedback(self):
         feedback_chain = self._build_feedback_chain()
-        print("Analysing the TIL.....")
+        pprint.pp("Analysing the TIL.....")
         self.feedback = feedback_chain.invoke({"til_content": self.content})
-        print("Feedback: ", self.feedback)
+        print("Feedback: ")
+        pprint.pp(self.feedback)
 
     def _build_feedback_chain(self):
         feedback_parser = JsonOutputParser(pydantic_object=TilFeedbackResult)
@@ -73,7 +73,8 @@ class TilCrew:
             ),
             HumanMessagePromptTemplate.from_template("{til_content}")
         ])
-        print("Prompt: ", feedback_prompt)
+        print("Prompt: ")
+        pprint.pp(feedback_prompt, width=80)
         llm = ChatOpenAI(model=OPENAI_MODEL, temperature=0.2)
         analysis_chain = feedback_prompt | llm | feedback_parser
 
@@ -84,18 +85,14 @@ class TilFeedbackResult(BaseModel):
     insightful_score: int = Field(
         description="TIL score on insightful criteria")
     insightful_reason: str = Field(description="Reason for insightful_score")
-    # insightful_pass: bool = Field(description="True if it passes the insightful criteria, else Fales")
     factuality_score: int = Field(
         description="TIL score on factuality criteria")
     factuality_reason: str = Field(description="Reason for factuality_score")
-    # factuality_pass: bool = Field(description="True if it passes the factuality criteria, else Fales")
     simplicity_score: int = Field(
         description="TIL score on simplicity criteria")
     simplicity_reason: str = Field(description="Reason for simplicity_score")
-    # simplicity_pass: bool = Field(description="True if it passes the simplicity criteria, else Fales")
     grammatical_score: int = Field(
         description="TIL score on grammatical criteria")
     grammatical_reason: str = Field(description="Reason for grammatical_score")
-    # grammatical_pass: bool = Field(description="True if it passes the grammatical criteria, else Fales")
     final_suggestion: str = Field(
         description="Final suggested version of the TIL")

test.py

@@ -43,7 +43,8 @@ def main():
         if result['feedback'] == "not_ok":
           st.markdown(f"**Criteria:** {result['feedback_criteria']}")
           st.markdown(f"**Reason:** {result['reason']}")
-          st.markdown(f"**Suggestion:** {result['suggestion']}")
+          if result.get('suggestion') is not None:
+            st.markdown(f"**Suggestion:** {result['suggestion']}")
 
 if __name__ == "__main__":
     main()