vraj04patel commited on
Commit
5d970ca
·
1 Parent(s): 0dcbd75

remove print statement

Browse files
src/blogGenerator/LLMs/groqllm.py CHANGED
@@ -6,7 +6,7 @@ from langchain_groq import ChatGroq
6
  class GroqLLM:
7
  def __init__(self, user_controls_input):
8
  self.user_controls_input = user_controls_input
9
- print(f"#### self.user_controls_input : {self.user_controls_input}")
10
 
11
  def get_llm_model(self):
12
  try:
 
6
  class GroqLLM:
7
  def __init__(self, user_controls_input):
8
  self.user_controls_input = user_controls_input
9
+ # print(f"#### self.user_controls_input : {self.user_controls_input}")
10
 
11
  def get_llm_model(self):
12
  try:
src/blogGenerator/graph/graph_builder.py CHANGED
@@ -16,7 +16,7 @@ class GraphBuilder:
16
  Build a blog generation graph
17
  """
18
 
19
- print("Build a blog generation graph.......")
20
  try:
21
  self.route_deciding_node = RouteDecidingNode(self.llm)
22
  self.yt_transcript_node = YTTranscriptNode()
@@ -44,6 +44,5 @@ class GraphBuilder:
44
 
45
  def setup_graph(self):
46
  self.blog_generator_build_graph()
47
- print("setup_graph.......")
48
 
49
  return self.graph_builder.compile()
 
16
  Build a blog generation graph
17
  """
18
 
19
+ # print("Build a blog generation graph.......")
20
  try:
21
  self.route_deciding_node = RouteDecidingNode(self.llm)
22
  self.yt_transcript_node = YTTranscriptNode()
 
44
 
45
  def setup_graph(self):
46
  self.blog_generator_build_graph()
 
47
 
48
  return self.graph_builder.compile()
src/blogGenerator/nodes/content_generator_node.py CHANGED
@@ -15,7 +15,7 @@ class ContentGeneratorNode:
15
 
16
  def process(self, state: State):
17
 
18
- print(f"Node Called : ContentGeneratorNode ---> {state["decision"]}")
19
 
20
  blog_llm = self.llm.with_structured_output(BlogContent)
21
 
 
15
 
16
  def process(self, state: State):
17
 
18
+ # print(f"Node Called : ContentGeneratorNode ---> {state['decision']}")
19
 
20
  blog_llm = self.llm.with_structured_output(BlogContent)
21
 
src/blogGenerator/nodes/route_deciding_node.py CHANGED
@@ -21,7 +21,7 @@ class RouteDecidingNode:
21
  def process(self, state: State):
22
 
23
  # print(f"state['user_message'] : {state['user_message']}")
24
- print(f"Node Called : RouteDecidingNode")
25
 
26
  route = self.llm.with_structured_output(Route)
27
 
@@ -32,7 +32,7 @@ class RouteDecidingNode:
32
  ]
33
  )
34
 
35
- print(f"decision : {decision}")
36
 
37
  if decision.step == "youtube":
38
  extract_url = self.llm.invoke(
@@ -50,7 +50,7 @@ class RouteDecidingNode:
50
 
51
  def route_decision(self, state: State):
52
 
53
- print(f"state : {state}")
54
  # Return the node name you want to visit next
55
  if state["decision"] == "youtube":
56
  return "youtube"
 
21
  def process(self, state: State):
22
 
23
  # print(f"state['user_message'] : {state['user_message']}")
24
+ # print(f"Node Called : RouteDecidingNode")
25
 
26
  route = self.llm.with_structured_output(Route)
27
 
 
32
  ]
33
  )
34
 
35
+ # print(f"decision : {decision}")
36
 
37
  if decision.step == "youtube":
38
  extract_url = self.llm.invoke(
 
50
 
51
  def route_decision(self, state: State):
52
 
53
+ # print(f"state : {state}")
54
  # Return the node name you want to visit next
55
  if state["decision"] == "youtube":
56
  return "youtube"
src/blogGenerator/nodes/yt_transcript_node.py CHANGED
@@ -10,7 +10,7 @@ class YTTranscriptNode:
10
  def process(self, state: State) -> dict:
11
  """Fetches transcript from a given YouTube URL"""
12
 
13
- print(f"Node Called : yt_transcipt")
14
 
15
  video_id = state["yt_url"].replace("https://www.youtube.com/watch?v=", "")
16
 
 
10
  def process(self, state: State) -> dict:
11
  """Fetches transcript from a given YouTube URL"""
12
 
13
+ # print(f"Node Called : yt_transcipt")
14
 
15
  video_id = state["yt_url"].replace("https://www.youtube.com/watch?v=", "")
16
 
src/blogGenerator/ui/streamlit/display_result.py CHANGED
@@ -10,7 +10,7 @@ class DisplayResultStreamlit:
10
  graph = self.graph
11
  user_message = self.user_message
12
 
13
- print(f"### user_message : {user_message}")
14
 
15
  with st.chat_message("user"):
16
  st.write(user_message)
 
10
  graph = self.graph
11
  user_message = self.user_message
12
 
13
+ # print(f"### user_message : {user_message}")
14
 
15
  with st.chat_message("user"):
16
  st.write(user_message)