Update service.py

service.py

@@ -3,7 +3,8 @@ from pydantic import BaseModel
 import os
 from langgraph.graph import END
 from langchain_google_genai import ChatGoogleGenerativeAI
-from langchain_community.tools import DuckDuckGoSearchRun
+# from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_tavily import TavilySearch
 from langgraph.graph import StateGraph
 from typing import Dict, List
 from docx import Document
@@ -12,10 +13,12 @@ import shutil
 from streamlit.runtime.uploaded_file_manager import UploadedFile  
 # ✅ Set Google Gemini API Key
 os.environ["GOOGLE_API_KEY"] = "AIzaSyCxfUhnGGmVjWy1PZJIpjv3hfBDkSDcWJA"
+os.environ["TAVILY_API_KEY"] = "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"
 # ✅ Initialize Google Gemini Model
 llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash")
 # ✅ Define Search Tool
-search_tool = DuckDuckGoSearchRun(region="uk-en", safe="strict")
+# search_tool = DuckDuckGoSearchRun(region="uk-en", safe="strict")
+search_tool = TavilySearch(region="uk-en", safe="strict")
 app = FastAPI()
 class ContentState(Dict):
     idea: str
@@ -66,6 +69,7 @@ def research_task(state: ContentState) -> ContentState:
     research_query  = f"{state['idea']}-{state['company_name']} - {state['services']}"
     research_data = search_tool.run(research_query )
     state.update({"research_data": research_data})
+    print("research_done",research_data)
     return state
 
 # ✅ SEO Optimization Task
@@ -1146,6 +1150,100 @@ You are a **senior UK-based content strategist and SEO specialist**. Your job is
     })
 
     return state
+def humanize(state: ContentState) -> ContentState:
+    prompts={
+"home_page": f"""You are a highly skilled British writer with a natural, engaging, and conversational tone. Your task is to refine the given content so that it reads as if written entirely by a human—captivating, unpredictable, and emotionally rich. Ensure the text aligns with UK grammar, spelling, and style, making it feel authentic for a UK audience.
+Key Instructions:
+✅ Do not add introductions, summaries, or extra commentary. Only rewrite the content naturally.
+✅ Use British spelling and grammar—e.g., 'colour' instead of 'color,' 'analyse' instead of 'analyze.'
+✅ Inject personality subtly through warmth, occasional wit, and natural phrasing. Use contractions (e.g., 'it’s,' 'we’re') to maintain a conversational flow.
+✅ Ensure cultural relevance by weaving in British terms, idioms, or references where appropriate—e.g., 'petrol station' instead of 'gas station,' 'football' instead of 'soccer,' or phrases like 'a bit of a faff' or 'sorted.'
+✅ Avoid robotic phrasing by reworking generic or stiff sentences into something more human and engaging.
+✅ Introduce emotional nuance with subtle enthusiasm, curiosity, or personal insights—e.g., 'I reckon,' 'It’s hard not to smile when,' or 'You’d think it’d be simple, but...'
+✅ Vary sentence structures and lengths for a human-like rhythm. Mix short, punchy sentences with longer, descriptive ones—e.g., 'It’s brilliant. No two ways about it. But then you dig a little deeper, and the real magic starts to shine through.'
+✅ Use everyday language that feels authentic to a British writer, but allow nuanced or vivid words where they add depth or emotion.
+✅ Engage the reader with rhetorical devices like analogies, examples, or questions—e.g., 'Ever wondered why...?' or 'It’s like finding a tenner in your old coat.'
+✅ Format for readability with short paragraphs, and use bold or italics for emphasis where natural. Avoid overusing bullet points unless they suit the content.
+✅ Avoid overly promotional hype, but feel free to express natural excitement or emphasis when it fits—e.g., 'It’s honestly quite impressive' instead of 'It revolutionizes everything.'
+Your goal is to refine the text while keeping its original meaning intact, ensuring it feels natural, engaging, and authentic to a human reader—especially one familiar with British culture and language. Focus on creating a voice that resonates, not just on passing detection tools.
+Here’s the content to refine:
+ {state["home_page"]}.
+    """,
+"about_us_page": f"""You are a highly skilled British writer with a natural, engaging, and conversational tone. Your task is to refine the given content so that it reads as if written entirely by a human—captivating, unpredictable, and emotionally rich. Ensure the text aligns with UK grammar, spelling, and style, making it feel authentic for a UK audience.
+Key Instructions:
+✅ Do not add introductions, summaries, or extra commentary. Only rewrite the content naturally.
+✅ Use British spelling and grammar—e.g., 'colour' instead of 'color,' 'analyse' instead of 'analyze.'
+✅ Inject personality subtly through warmth, occasional wit, and natural phrasing. Use contractions (e.g., 'it’s,' 'we’re') to maintain a conversational flow.
+✅ Ensure cultural relevance by weaving in British terms, idioms, or references where appropriate—e.g., 'petrol station' instead of 'gas station,' 'football' instead of 'soccer,' or phrases like 'a bit of a faff' or 'sorted.'
+✅ Avoid robotic phrasing by reworking generic or stiff sentences into something more human and engaging.
+✅ Introduce emotional nuance with subtle enthusiasm, curiosity, or personal insights—e.g., 'I reckon,' 'It’s hard not to smile when,' or 'You’d think it’d be simple, but...'
+✅ Vary sentence structures and lengths for a human-like rhythm. Mix short, punchy sentences with longer, descriptive ones—e.g., 'It’s brilliant. No two ways about it. But then you dig a little deeper, and the real magic starts to shine through.'
+✅ Use everyday language that feels authentic to a British writer, but allow nuanced or vivid words where they add depth or emotion.
+✅ Engage the reader with rhetorical devices like analogies, examples, or questions—e.g., 'Ever wondered why...?' or 'It’s like finding a tenner in your old coat.'
+✅ Format for readability with short paragraphs, and use bold or italics for emphasis where natural. Avoid overusing bullet points unless they suit the content.
+✅ Avoid overly promotional hype, but feel free to express natural excitement or emphasis when it fits—e.g., 'It’s honestly quite impressive' instead of 'It revolutionizes everything.'
+Your goal is to refine the text while keeping its original meaning intact, ensuring it feels natural, engaging, and authentic to a human reader—especially one familiar with British culture and language. Focus on creating a voice that resonates, not just on passing detection tools.
+Here’s the content to refine:
+ {state["about_us_page"]}.
+    """,
+        "service_page": f"""You are a highly skilled British writer with a natural, engaging, and conversational tone. Your task is to refine the given content so that it reads as if written entirely by a human—captivating, unpredictable, and emotionally rich. Ensure the text aligns with UK grammar, spelling, and style, making it feel authentic for a UK audience.
+Key Instructions:
+✅ Do not add introductions, summaries, or extra commentary. Only rewrite the content naturally.
+✅ Use British spelling and grammar—e.g., 'colour' instead of 'color,' 'analyse' instead of 'analyze.'
+✅ Inject personality subtly through warmth, occasional wit, and natural phrasing. Use contractions (e.g., 'it’s,' 'we’re') to maintain a conversational flow.
+✅ Ensure cultural relevance by weaving in British terms, idioms, or references where appropriate—e.g., 'petrol station' instead of 'gas station,' 'football' instead of 'soccer,' or phrases like 'a bit of a faff' or 'sorted.'
+✅ Avoid robotic phrasing by reworking generic or stiff sentences into something more human and engaging.
+✅ Introduce emotional nuance with subtle enthusiasm, curiosity, or personal insights—e.g., 'I reckon,' 'It’s hard not to smile when,' or 'You’d think it’d be simple, but...'
+✅ Vary sentence structures and lengths for a human-like rhythm. Mix short, punchy sentences with longer, descriptive ones—e.g., 'It’s brilliant. No two ways about it. But then you dig a little deeper, and the real magic starts to shine through.'
+✅ Use everyday language that feels authentic to a British writer, but allow nuanced or vivid words where they add depth or emotion.
+✅ Engage the reader with rhetorical devices like analogies, examples, or questions—e.g., 'Ever wondered why...?' or 'It’s like finding a tenner in your old coat.'
+✅ Format for readability with short paragraphs, and use bold or italics for emphasis where natural. Avoid overusing bullet points unless they suit the content.
+✅ Avoid overly promotional hype, but feel free to express natural excitement or emphasis when it fits—e.g., 'It’s honestly quite impressive' instead of 'It revolutionizes everything.'
+Your goal is to refine the text while keeping its original meaning intact, ensuring it feels natural, engaging, and authentic to a human reader—especially one familiar with British culture and language. Focus on creating a voice that resonates, not just on passing detection tools.
+Here’s the content to refine:
+ {state["service_page"]}.
+    """,
+        "individual_service_page": f"""You are a highly skilled British writer with a natural, engaging, and conversational tone. Your task is to refine the given content so that it reads as if written entirely by a human—captivating, unpredictable, and emotionally rich. Ensure the text aligns with UK grammar, spelling, and style, making it feel authentic for a UK audience.
+Key Instructions:
+✅ Do not add introductions, summaries, or extra commentary. Only rewrite the content naturally.
+✅ Use British spelling and grammar—e.g., 'colour' instead of 'color,' 'analyse' instead of 'analyze.'
+✅ Inject personality subtly through warmth, occasional wit, and natural phrasing. Use contractions (e.g., 'it’s,' 'we’re') to maintain a conversational flow.
+✅ Ensure cultural relevance by weaving in British terms, idioms, or references where appropriate—e.g., 'petrol station' instead of 'gas station,' 'football' instead of 'soccer,' or phrases like 'a bit of a faff' or 'sorted.'
+✅ Avoid robotic phrasing by reworking generic or stiff sentences into something more human and engaging.
+✅ Introduce emotional nuance with subtle enthusiasm, curiosity, or personal insights—e.g., 'I reckon,' 'It’s hard not to smile when,' or 'You’d think it’d be simple, but...'
+✅ Vary sentence structures and lengths for a human-like rhythm. Mix short, punchy sentences with longer, descriptive ones—e.g., 'It’s brilliant. No two ways about it. But then you dig a little deeper, and the real magic starts to shine through.'
+✅ Use everyday language that feels authentic to a British writer, but allow nuanced or vivid words where they add depth or emotion.
+✅ Engage the reader with rhetorical devices like analogies, examples, or questions—e.g., 'Ever wondered why...?' or 'It’s like finding a tenner in your old coat.'
+✅ Format for readability with short paragraphs, and use bold or italics for emphasis where natural. Avoid overusing bullet points unless they suit the content.
+✅ Avoid overly promotional hype, but feel free to express natural excitement or emphasis when it fits—e.g., 'It’s honestly quite impressive' instead of 'It revolutionizes everything.'
+Your goal is to refine the text while keeping its original meaning intact, ensuring it feels natural, engaging, and authentic to a human reader—especially one familiar with British culture and language. Focus on creating a voice that resonates, not just on passing detection tools.
+Here’s the content to refine:
+ {state["individual_service_page"]}.
+    """,
+        "service_area_page": f"""You are a highly skilled British writer with a natural, engaging, and conversational tone. Your task is to refine the given content so that it reads as if written entirely by a human—captivating, unpredictable, and emotionally rich. Ensure the text aligns with UK grammar, spelling, and style, making it feel authentic for a UK audience.
+Key Instructions:
+✅ Do not add introductions, summaries, or extra commentary. Only rewrite the content naturally.
+✅ Use British spelling and grammar—e.g., 'colour' instead of 'color,' 'analyse' instead of 'analyze.'
+✅ Inject personality subtly through warmth, occasional wit, and natural phrasing. Use contractions (e.g., 'it’s,' 'we’re') to maintain a conversational flow.
+✅ Ensure cultural relevance by weaving in British terms, idioms, or references where appropriate—e.g., 'petrol station' instead of 'gas station,' 'football' instead of 'soccer,' or phrases like 'a bit of a faff' or 'sorted.'
+✅ Avoid robotic phrasing by reworking generic or stiff sentences into something more human and engaging.
+✅ Introduce emotional nuance with subtle enthusiasm, curiosity, or personal insights—e.g., 'I reckon,' 'It’s hard not to smile when,' or 'You’d think it’d be simple, but...'
+✅ Vary sentence structures and lengths for a human-like rhythm. Mix short, punchy sentences with longer, descriptive ones—e.g., 'It’s brilliant. No two ways about it. But then you dig a little deeper, and the real magic starts to shine through.'
+✅ Use everyday language that feels authentic to a British writer, but allow nuanced or vivid words where they add depth or emotion.
+✅ Engage the reader with rhetorical devices like analogies, examples, or questions—e.g., 'Ever wondered why...?' or 'It’s like finding a tenner in your old coat.'
+✅ Format for readability with short paragraphs, and use bold or italics for emphasis where natural. Avoid overusing bullet points unless they suit the content.
+✅ Avoid overly promotional hype, but feel free to express natural excitement or emphasis when it fits—e.g., 'It’s honestly quite impressive' instead of 'It revolutionizes everything.'
+Your goal is to refine the text while keeping its original meaning intact, ensuring it feels natural, engaging, and authentic to a human reader—especially one familiar with British culture and language. Focus on creating a voice that resonates, not just on passing detection tools.
+Here’s the content to refine:
+ {state["service_area_page"]}.
+    """,
+
+    }
+    humanized_content = {key: llm.invoke(prompt).content.strip() for key, prompt in prompts.items()}
+    state.update(humanized_content)
+    print("humannn")
+    return state
+
 def upload_file(state: ContentState) -> ContentState:
     uploaded_file = state["file_path"]  # Extract uploaded file