Spaces:
Sleeping
Sleeping
Commit ·
5d970ca
1
Parent(s): 0dcbd75
remove print statement
Browse files- src/blogGenerator/LLMs/groqllm.py +1 -1
- src/blogGenerator/graph/graph_builder.py +1 -2
- src/blogGenerator/nodes/content_generator_node.py +1 -1
- src/blogGenerator/nodes/route_deciding_node.py +3 -3
- src/blogGenerator/nodes/yt_transcript_node.py +1 -1
- src/blogGenerator/ui/streamlit/display_result.py +1 -1
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[
|
| 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)
|