Update main.py

main.py

@@ -20,63 +20,42 @@ from fastapi.responses import FileResponse
 from fastapi.responses import StreamingResponse
 from langchain_core.prompts import PromptTemplate
 from dotenv import load_dotenv
+import os
+import logging
 from fastapi import Request
-import time
-from io import BytesIO
-import uuid
 import tempfile
-import httpx
-import asyncioasync def generate_image(prompt: str):
-    url = "https://api.bfl.ml/v1/flux-pro-1.1"
-    headers = {
-        "accept": "application/json",
-        "x-key": "4f69d408-4979-4812-8ad2-ec6d232c9ddf",
-        "Content-Type": "application/json"
-    }
-    payload = {
-        "prompt": prompt,
-        "width": 1024,
-        "height": 1024,
-        "guidance_scale": 1,
-        "num_inference_steps": 50,
-        "max_sequence_length": 512,
-    }
 
-    async with httpx.AsyncClient() as client:
-        response = await client.post(url, headers=headers, json=payload)
-        response_data = response.json()
-        if "id" not in response_data:
-            raise Exception("Error generating image: ID missing from response")
-
-        request_id = response_data["id"]
-
-        # Poll for the image result
-        while True:
-            await asyncio.sleep(0.5)
-            result = await client.get(
-                "https://api.bfl.ml/v1/get_result",
-                headers=headers,
-                params={"id": request_id},
-            )
-            result_data = result.json()
-
-            if result_data["status"] == "Ready":
-                if "result" in result_data and "sample" in result_data["result"]:
-                    image_url = result_data["result"]["sample"]
-
-                    # Download the image
-                    image_response = await client.get(image_url)
-                    if image_response.status_code == 200:
-                        img = Image.open(BytesIO(image_response.content))
-                        filename = f"generated_{uuid.uuid4()}.png"
-                        filepath = save_image_locally(img, filename)
-                        return filepath
-                    else:
-                        raise Exception("Failed to download the image")
-                else:
-                    raise Exception("Expected 'sample' key not found in the result")
-            elif result_data["status"] == "Error":
-                raise Exception(f"Image generation failed: {result_data.get('error', 'Unknown error')}")
+logging.basicConfig(level=logging.DEBUG)
+
+
+load_dotenv()
+
+app = FastAPI()
+
+class ImagePromptRequest(BaseModel):
+    blog_post_content: str
+    previous_image_prompts: str
+
+API_KEY = os.getenv("API_KEY")
+
+llm = ChatGroq(
+    model="llama-3.3-70b-versatile", 
+    temperature=0.3, 
+    max_tokens=1024, 
+    api_key=API_KEY)
+
+class blog_request(BaseModel):
+    TypeOf : str
+    target_audience: str
+    tone: str
+    point_of_view: str
+    target_country: str
+    keywords: List[str]
+
+def fetch_google_results(keywords: List[str], target_country: str) -> List[str]:
+    username=os.getenv("USERNAME")
+    password=os.getenv("PASSWORD")
+    all_results_dict = {}
     for keyword in keywords:
         payload = {'source': 'google_search','query': keyword, 'domain': 'com', 'geo_location': target_country, 'locale': 'en-us', 'parse': True, 'start_page': 1, 'pages': 5,  'limit': 10, }
     try:
@@ -256,13 +235,14 @@ Examples:
     prompt = prompt_template.format(blog_post_content=content,previous_image_prompts=previous_prompts)
     response = llm.invoke(prompt)
     return response.content  # Extract prompts
-    
+
 def save_image_locally(image, filename):
     filepath = os.path.join(image_storage_dir, filename)
     image.save(filepath, format="PNG")
     return filepath
-
-async def generate_image(prompt: str):
+    
+def generate_image(prompt: str):
+    # API call to image generation service
     url = "https://api.bfl.ml/v1/flux-pro-1.1"
     headers = {
         "accept": "application/json",
@@ -270,7 +250,7 @@ async def generate_image(prompt: str):
         "Content-Type": "application/json"
     }
     payload = {
-        "prompt": prompt,
+        "prompt": refined_prompt,
         "width": 1024,
         "height": 1024,
         "guidance_scale": 1,
@@ -278,42 +258,39 @@ async def generate_image(prompt: str):
         "max_sequence_length": 512,
     }
 
-    async with httpx.AsyncClient() as client:
-        response = await client.post(url, headers=headers, json=payload)
-        response_data = response.json()
-        if "id" not in response_data:
-            raise Exception("Error generating image: ID missing from response")
-
-        request_id = response_data["id"]
-
-        # Poll for the image result
-        while True:
-            await asyncio.sleep(0.5)
-            result = await client.get(
-                "https://api.bfl.ml/v1/get_result",
-                headers=headers,
-                params={"id": request_id},
-            )
-            result_data = result.json()
-
-            if result_data["status"] == "Ready":
-                if "result" in result_data and "sample" in result_data["result"]:
-                    image_url = result_data["result"]["sample"]
-
-                    # Download the image
-                    image_response = await client.get(image_url)
-                    if image_response.status_code == 200:
-                        img = Image.open(BytesIO(image_response.content))
-                        filename = f"generated_{uuid.uuid4()}.png"
-                        filepath = save_image_locally(img, filename)
-                        return filepath
-                    else:
-                        raise Exception("Failed to download the image")
+    response = requests.post(url, headers=headers, json=payload).json()
+    if "id" not in response:
+        raise HTTPException(status_code=500, detail="Error generating image: ID missing from response")
+
+    request_id = response["id"]
+
+    # Poll for the image result
+    while True:
+        time.sleep(0.5)
+        result = requests.get(
+            "https://api.bfl.ml/v1/get_result",
+            headers=headers,
+            params={"id": request_id},
+        ).json()
+
+        if result["status"] == "Ready":
+            if "result" in result and "sample" in result["result"]:
+                image_url = result["result"]["sample"]
+
+                # Download the image
+                image_response = requests.get(image_url)
+                if image_response.status_code == 200:
+                    img = Image.open(BytesIO(image_response.content))
+                    filename = f"generated_{uuid.uuid4()}.png"
+                    filepath = save_image_locally(img, filename)
                 else:
-                    raise Exception("Expected 'sample' key not found in the result")
-            elif result_data["status"] == "Error":
-                raise Exception(f"Image generation failed: {result_data.get('error', 'Unknown error')}")
-                
+                    raise HTTPException(status_code=500, detail="Failed to download the image")
+            else:
+                raise HTTPException(status_code=500, detail="Expected 'sample' key not found in the result")
+        elif result["status"] == "Error":
+            raise HTTPException(status_code=500, detail=f"Image generation failed: {result.get('error', 'Unknown error')}")
+
+
 def selected_category(category: dict, search_results: list) -> str:
     prompt_template = """
     Based on the given search results, select the most appropriate category for the blog post.
@@ -362,6 +339,20 @@ def fetch_google_results_for_site(keywords: List[str]) -> List[Dict[str, int]]:
     except requests.RequestException as e:
         print(f"Error fetching results: {e}")
         return []
+
+def decide_to_generate_image(content: str, i:int) -> bool:
+    max=3
+    prompt_template = '''
+    You are an advanced language model tasked with deciding if an image should be generated based on the provided blog post. Analyze the blog content and respond with "Yes" or "No" only. Generate an image only if the content is rich, high-quality, and would benefit from it. Generate a maximum of 3 images: if {i} > {max}, respond with "No".
+    Blog post:
+    {blog_post}
+    Output:
+    Yes or No
+    '''
+    prompt=prompt_template.format(blog_post=content,i=i,max=max)
+    response = llm.invoke(prompt)
+    should_generate_image = response.content
+    return should_generate_image
         
 def generate_linkages(blog_post: str, search_results: list, keywords: List[str]) -> dict:
     Internal_search_results = fetch_google_results_for_site(keywords)
@@ -402,20 +393,20 @@ def generate_linkages(blog_post: str, search_results: list, keywords: List[str])
     
 @app.post("/generate_blog/", response_model=dict)
 def create_blog_pipeline(blog_request: blog_request):
-    output_dir = "/tmp/pic"
-    if not os.path.exists(output_dir):
-        os.makedirs(output_dir, exist_ok=True)
     try:
         word_file_path = "Generated_Blog_Post.docx"
         for file_path in [word_file_path]:
             if os.path.exists(file_path):
                 os.remove(file_path)
                 
-        category = {"Parenting Stages":["Baby & Toddler Years","Preschool & Early Childhood","Big Kids (6–12 Years)","Tweens & Teens","Newborn Care","Parenting After Baby #2 (or #3!)"],"Everyday Life with Kids":["Daily Routines & Organization","Mealtime & Nutrition","Playtime & Activities","Sleep Schedules & Tips","Family Time Ideas","Special Occasions & Holidays"],"Self-Care for Moms":["Health & Wellness","Mental Health & Stress Relief","Beauty & Self-Care Tips","Hobbies & “Me Time”","Personal Growth & Goal Setting"],"Parenting Tips & Tricks":["Time-Saving Hacks","Budgeting for Families","Quick Cleaning Tips","Home Organization with Kids","School & Homework Help","Tech Tools for Parenting"],"Mom Life (Humor & Reality)":["Honest Mom Moments","Laughs & Parenting Memes","Confessions & Fails","Real Life, Real Moms","Quotes & Relatable Stories"],"Parenting Styles & Philosophies":["Gentle Parenting & Positive Discipline","Attachment Parenting","Raising Independent Kids","Balancing Structure & Freedom","Parenting Trends & Research"],"Relationships & Family Dynamics":["Co-Parenting & Communication","Sibling Relationships","Family Bonding Activities","Grandparents & Extended Family","Blended Families & Step-Parenting"],"Learning & Development":["Early Childhood Education","Fun Learning at Home","Language & Social Skills","Emotional Development","Physical & Motor Skills"],"Health & Wellness":["Child Nutrition & Health","Exercise & Play for Kids","Sleep Health","Pediatric Check-Ups","Common Illnesses & Remedies","Childproofing & Safety"],"Mompreneurs & Working Moms":["Balancing Work & Family","Remote Work Tips","Side Hustles & Passions","Time Management for Busy Moms","Self-Care for Working Moms"],"School & Education":["Preschool & Daycare Choices","School Readiness & Transitions","Homework & Study Skills","Navigating School Friendships","Involvement in School Activities"],"Lifestyle & Home":["Home Décor for Families","Sustainable & Eco-Friendly Choices","Family Finances & Budgeting","Travel & Family Adventures","Pets & Kids"],"Parenting Challenges":["Tantrums & Discipline","Social Media & Screen Time","Bullying & Peer Pressure","Dealing with Picky Eaters","Navigating Kids’ Fears"],"Creative & Fun Ideas":["DIY Projects for Kids","Kid-Friendly Crafts","Fun Recipes & Snacks","Family Games & Activities","Fun Celebrations & Birthdays"],"Modern Parenting Topics":["Raising Kids in a Digital World","Cultural & Diversity Awareness","Gender-Neutral Parenting","Parenting and Social Media"],"The Wild World of Parenting":["Tiny Bosses: Life with Toddlers","Kid Logic: Decoding the Mind of a Child","Growing Up Fast: Navigating the Tween Years"],"The Mom Zone":["Surviving the Madness, One Coffee at a Time","Keeping It Real: The Beautiful Mess of Mom Life","Dear Diary: Honest Mom Confessions"],"Mastering the Art of Family Chaos":["Organized Chaos: Tips for Running a Household","Barely Hanging On: Parenting Hacks for the Real World","Kid-Proof Your Life (If That’s Even Possible)"],"Mom Self-Care, Simplified":["Time for You: Self-Care for Busy Moms","Staying Sane (Mostly) with Self-Care on a Budget","Love Yourself: From Self-Care to Self-Love"],"Making Memories, Keeping Your Sanity":["Everyday Magic: Fun Family Traditions","Making the Ordinary Extraordinary","The Cool Mom’s Guide to Family Fun"],"Mom Hacks & Life-Saving Tricks":["Shortcuts for Sanity: Clever Parenting Hacks","The No-Fuss Guide to Getting Stuff Done","Mom Brain Solutions: Little Tricks for Big Wins"],"When Kids Are…Kids!":["Real Talk: Tantrums, Tears & Tiny Attitudes","Kid Quirks: The Weird, Wonderful World of Children","Mini People, Mega Emotions"],"Relationships and Realities":["It Takes Two: Parenting Together (Even When You Don’t Agree)","Keeping Love Alive Amid the Chaos","Keeping the Family Peace, One Day at a Time"],"The School Scene":["Homework Without the Headache","From Preschool to Preteen Drama: Surviving School Years","Winning at School (Even If They Don’t Love It)"],"Digital World for Digital Kids":["Screen Time vs. Play Time: Finding the Balance","Raising Tech-Savvy Kids in a Tech-Obsessed World","Social Media & Selfies: Teaching Digital Smarts"],"Raising the Next Generation":["The Kindness Project: Raising Empathetic Kids","How to Raise Future World-Changers","The Power of Yes and No: Teaching Choices"],"Healthier, Happier Families":["Making Meals Easy & Fun (Yes, Really!)","Health Hacks for Kids Who Hate Veggies","Small Habits for Big Health Wins"],"The Organized Chaos Hub":["Declutter Like a Pro (Yes, Even with Kids)","Home Hacks for the Ultimate Kid-Friendly Space","Mastering the Family Schedule"],"Funny Mom Survival Kit":["Parenting Memes You’ll Feel in Your Soul","Surviving Kids’ Parties with Style","Confessions of a Bedtime Warrior"],"Big Dreams & Little Goals":["Goal-Getting for Moms Who Do It All","Dare to Dream Big (Even If You’re Tired)","Mom Goals: From ‘Just Survive’ to ‘Thrive’"],"For the Love of Learning":["Learning Through Play: Fun Ideas for Little Learners","Home Learning Hacks for Smart Kids","Raising Curious Kids: Sparking Little Imaginations"],"Tales from the Trenches":["Stories from the Wild World of Parenting","Lessons Learned from the Chaos","Hilarious Mom Stories You’ll Never Believe"],"Adventures Big and Small":["Tiny Adventures: Fun for Kids of All Ages","Family Vacations & Kid-Friendly Getaways","Staycations That Feel Like the Real Deal"],"The Support Network":["For the Love of Moms: Support & Community","Village of Moms: Finding Your Support Circle","Surviving & Thriving Together"],"Creative Kids Zone":["Arts & Crafts that Won’t Break the Bank","Imagination Station: Encouraging Creative Play","Rainy Day Fun: Indoor Ideas for Any Weather"]}
-
-        output_dir = "/tmp/pic"
-        if not os.path.exists(output_dir):
-            os.makedirs(output_dir, exist_ok=True)
+        category = {"Parenting Stages":["Baby & Toddler Years","Preschool & Early Childhood","Big Kids (6–12 Years)","Tweens & Teens","Newborn Care","Parenting After Baby #2 (or #3!)"],"Everyday Life with Kids":["Daily Routines & Organization","Mealtime & Nutrition","Playtime & Activities","Sleep Schedules & Tips","Family Time Ideas","Special Occasions & Holidays"],"Self-Care for Moms":["Health & Wellness","Mental Health & Stress Relief","Beauty & Self-Care Tips","Hobbies & “Me Time”","Personal Growth & Goal Setting"],"Parenting Tips & Tricks":["Time-Saving Hacks","Budgeting for Families","Quick Cleaning Tips","Home Organization with Kids","School & Homework Help","Tech Tools for Parenting"],"Mom Life (Humor & Reality)":["Honest Mom Moments","Laughs & Parenting Memes","Confessions & Fails","Real Life, Real Moms","Quotes & Relatable Stories"],"Parenting Styles & Philosophies":["Gentle Parenting & Positive Discipline","Attachment Parenting","Raising Independent Kids","Balancing Structure & Freedom","Parenting Trends & Research"],"Relationships & Family Dynamics":["Co-Parenting & Communication","Sibling Relationships","Family Bonding Activities","Grandparents & Extended Family","Blended Families & Step-Parenting"],"Learning & Development":["Early Childhood Education","Fun Learning at Home","Language & Social Skills","Emotional Development","Physical & Motor Skills"],"Health & Wellness":["Child Nutrition & Health","Exercise & Play for Kids","Sleep Health","Pediatric Check-Ups","Common Illnesses & Remedies","Childproofing & Safety"],"Mompreneurs & Working Moms":["Balancing Work & Family","Remote Work Tips","Side Hustles & Passions","Time Management for Busy Moms","Self-Care for Working Moms"],"School & Education":["Preschool & Daycare Choices","School Readiness & Transitions","Homework & Study Skills","Navigating School Friendships","Involvement in School Activities"],"Lifestyle & Home":["Home Décor for Families","Sustainable & Eco-Friendly Choices","Family Finances & Budgeting","Travel & Family Adventures","Pets & Kids"],"Parenting Challenges":["Tantrums & Discipline","Social Media & Screen Time","Bullying & Peer Pressure","Dealing with Picky Eaters","Navigating Kids’ Fears"],"Creative & Fun Ideas":["DIY Projects for Kids","Kid-Friendly Crafts","Fun Recipes & Snacks","Family Games & Activities","Fun Celebrations & Birthdays"],"Modern Parenting Topics":["Raising Kids in a Digital World","Cultural & Diversity Awareness","Gender-Neutral Parenting","Parenting and Social Media"],"The Wild World of Parenting":["Tiny Bosses: Life with Toddlers","Kid Logic: Decoding the Mind of a Child","Growing Up Fast: Navigating the Tween Years"],"The Mom Zone":["Surviving the Madness, One Coffee at a Time","Keeping It Real: The Beautiful Mess of Mom Life","Dear Diary: Honest Mom Confessions"],"Mastering the Art of Family Chaos":["Organized Chaos: Tips for Running a Household","Barely Hanging On: Parenting Hacks for the Real World","Kid-Proof Your Life (If That’s Even Possible)"],"Mom Self-Care, Simplified":["Time for You: Self-Care for Busy Moms","Staying Sane (Mostly) with Self-Care on a Budget","Love Yourself: From Self-Care to Self-Love"],"Making Memories, Keeping Your Sanity":["Everyday Magic: Fun Family Traditions","Making the Ordinary Extraordinary","The Cool Mom’s Guide to Family Fun"],"Mom Hacks & Life-Saving Tricks":["Shortcuts for Sanity: Clever Parenting Hacks","The No-Fuss Guide to Getting Stuff Done","Mom Brain Solutions: Little Tricks for Big Wins"],"When Kids Are…Kids!":["Real Talk: Tantrums, Tears & Tiny Attitudes","Kid Quirks: The Weird, Wonderful World of Children","Mini People, Mega Emotions"],"Relationships and Realities":["It Takes Two: Parenting Together (Even When You Don’t Agree)","Keeping Love Alive Amid the Chaos","Keeping the Family Peace, One Day at a Time"],"The School Scene":["Homework Without the Headache","From Preschool to Preteen Drama: Surviving School Years","Winning at School (Even If They Don’t Love It)"],"Digital World for Digital Kids":["Screen Time vs. Play Time: Finding the Balance","Raising Tech-Savvy Kids in a Tech-Obsessed World","Social Media & Selfies: Teaching Digital Smarts"],"Raising the Next Generation":["The Kindness Project: Raising Empathetic Kids","How to Raise Future World-Changers","The Power of Yes and No: Teaching Choices"],"Healthier, Happier Families":["Making Meals Easy & Fun (Yes, Really!)","Health Hacks for Kids Who Hate Veggies","Small Habits for Big Health Wins"],"The Organized Chaos Hub":["Declutter Like a Pro (Yes, Even with Kids)","Home Hacks for the Ultimate Kid-Friendly Space","Mastering the Family Schedule"],"Funny Mom Survival Kit":["Parenting Memes You’ll Feel in Your Soul","Surviving Kids’ Parties with Style","Confessions of a Bedtime Warrior"],"Big Dreams & Little Goals":["Goal-Getting for Moms Who Do It All","Dare to Dream Big (Even If You’re Tired)","Mom Goals: From ‘Just Survive’ to ‘Thrive’"],"For the Love of Learning":["Learning Through Play: Fun Ideas for Little Learners","Home Learning Hacks for Smart Kids","Raising Curious Kids: Sparking Little Imaginations"],"Tales from the Trenches":["Stories from the Wild World of Parenting","Lessons Learned from the Chaos","Hilarious Mom Stories You’ll Never Believe"],"Adventures Big and Small":["Tiny Adventures: Fun for Kids of All Ages","Family Vacations & Kid-Friendly Getaways","Staycations That Feel Like the Real Deal"],"The Support Network":["For the Love of Moms: Support & Community","Village of Moms: Finding Your Support Circle","Surviving & Thriving Together"],"Creative Kids Zone":["Arts & Crafts that Won’t Break the Bank","Imagination Station: Encouraging Creative Play","Rainy Day Fun: Indoor Ideas for Any Weather"]}        
+        
+        # Create a temporary directory inside the 'temp' folder
+        temp_base = "temp"
+        os.makedirs(temp_base, exist_ok=True)  # Ensure the 'temp' base folder exists
+        output_dir = tempfile.mkdtemp(dir=temp_base, prefix="pic_")
+        
+        print(f"Temporary directory created: {output_dir}")
 
 
         print('SEO Searching')
@@ -440,7 +431,7 @@ def create_blog_pipeline(blog_request: blog_request):
         image = generate_image(image_prompt)
         
         if image:
-            title_image_path = "/tmp/pic/image.png"
+            title_image_path = "pic/image.png"
             image.save(title_image_path)
             document.add_picture(title_image_path, width=Inches(6), height=Inches(6))
         else:
@@ -449,8 +440,6 @@ def create_blog_pipeline(blog_request: blog_request):
         
         subheadings = generate_blog_subheadings(title, search_results, selected_cat, blog_request )
         
-        print('processing Subheadings for blog post')
-        
         for i, subheading in enumerate(tqdm(subheadings, desc="Processing subheadings")):
             content = BlogPostPromptSingleSubheading(                
                 title, subheading, blog_request, search_results, selected_cat, blog_content            
@@ -464,7 +453,7 @@ def create_blog_pipeline(blog_request: blog_request):
             previous_image_prompts += image_prompt + " , "
             image = generate_image(image_prompt)
             if image:
-                subheading_image_path = f"/tmp/pic/image_{i}.png"
+                subheading_image_path = f"pic/image_{i}.png"
                 image.save(subheading_image_path)
                 document.add_picture(subheading_image_path, width=Inches(6), height=Inches(6))
             else:
@@ -479,18 +468,12 @@ def create_blog_pipeline(blog_request: blog_request):
         document.add_paragraph(formatted_linkages)
 
         document.save(word_file_path)
-        if os.path.exists("/tmp/pic/"):
-            shutil.rmtree("/tmp/pic/", ignore_errors=True)
-        else:
-             print("Directory '/tmp/pic/' does not exist. Skipping cleanup.")
-
+        shutil.rmtree("pic/", ignore_errors=True)
+        return {"docx_path": word_file_path}
     except Exception as e:
         print(f"An error occurred: {e}")
         return {"error": str(e)}
 
-    return {"docx_path": word_file_path}
-        
-
 # Update to include format_linkages
 def format_linkages(linkages: str) -> str:
     """
@@ -521,3 +504,9 @@ def download_file():
 async def root():
     return {"message": "API is up and running!"}
 
+@app.middleware("http")
+async def log_requests(request: Request, call_next):
+    print(f"Incoming request: {request.method} {request.url}")
+    response = await call_next(request)
+    print(f"Response status: {response.status_code}")
+    return response