msaifee commited on
Commit
f503142
·
1 Parent(s): 1de356a
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -61,11 +61,11 @@ def init_graph(api_key: str):
61
 
62
  builder = StateGraph(BlogState)
63
 
64
- builder.add_node("title_generator", generate_title)
65
- builder.add_node("search_web", search_web)
66
- builder.add_node("content_generator", generate_content)
67
- builder.add_node("content_reviewer", review_content)
68
- builder.add_node("quality_check", evaluate_content) # New evaluation node
69
 
70
  builder.add_edge(START, "title_generator")
71
  builder.add_edge(START, "search_web")
@@ -176,7 +176,7 @@ def route_based_on_verdict(state: BlogState):
176
  return "Pass" if state["is_blog_ready"] == "Pass" else "Fail"
177
 
178
  # Streamlit UI components
179
- st.title("🚀 BlogForge Pro")
180
  st.markdown("""
181
  **Smart Blog Generation with Auto-Refinement**
182
  *From first draft to final edit - AI-assisted writing meets professional standards*
 
61
 
62
  builder = StateGraph(BlogState)
63
 
64
+ builder.add_node("title_generator", generate_title) ## Generate Title
65
+ builder.add_node("search_web", search_web) ## Search Web using Tavily based in the topic
66
+ builder.add_node("content_generator", generate_content) ## Generate Content using the output of title_generator and search_web
67
+ builder.add_node("content_reviewer", review_content) ## Review Content and generate feedback
68
+ builder.add_node("quality_check", evaluate_content) ## Validate the content based on feedback and generate verdict
69
 
70
  builder.add_edge(START, "title_generator")
71
  builder.add_edge(START, "search_web")
 
176
  return "Pass" if state["is_blog_ready"] == "Pass" else "Fail"
177
 
178
  # Streamlit UI components
179
+ st.title("🚀 BlogForge Pro Agent")
180
  st.markdown("""
181
  **Smart Blog Generation with Auto-Refinement**
182
  *From first draft to final edit - AI-assisted writing meets professional standards*