Template Image v0

.gitattributes

@@ -36,3 +36,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 assets/Montserrat-Bold.ttf filter=lfs diff=lfs merge=lfs -text
 assets/Montserrat-Regular.ttf filter=lfs diff=lfs merge=lfs -text
 assets/template_natura_empty.jpg filter=lfs diff=lfs merge=lfs -text
+assets/template_1.png filter=lfs diff=lfs merge=lfs -text
+assets/template_2.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -19,7 +19,7 @@ The system uses a Gradio interface (`app.py`) with three main tabs:
     *   **Expert Perfume Analyst and Web Data Extractor:** This agent extracts detailed perfume information (notes, accords, longevity, sillage, similar fragrances, reviews) from the Fragrantica page.
     *   **Fragrance Expert Woman and Perfume Analysis Reporter:** This agent synthesizes the extracted data into a human-friendly report, including graded evaluations and personalized recommendations.
 
-3.  **Image Ad Generator:** This tab allows users to generate a promotional image for a product. It takes the product name, original price, final price, a coupon code, and a product image URL as input. The tool then generates a promotional image with this information, based on a template.
+3.  **Image Ad Generator:** This tab allows users to generate a promotional image for a product. It takes the product name, original price, final price, a coupon code, and a product image URL as input. The tool then generates a promotional image with this information, based on a template. The generated image is saved as a temporary file and displayed in the interface.
 
 ## Merchant Support
 
@@ -62,5 +62,7 @@ The application now supports generating ad copy for both **Natura** and **Mercad
 - [x] Add support for any model/api key supported by LiteLLM.
 - [x] Add Fragrantica support, where user will input a Fragrantica URL and the agent will extract and generate a Perfume Analysis report.
 - [x] Support Mercado Livre Merchant
-- [wip] Add image templates
+- [x] Add image templates
+- [wip] Add share button to images, that work with instagram on mobile. Android/IPhone.
+- [] Make image template trigger/working seamless with data from generated post.
 - [] Create carroussel images for Fragrantica post

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import os
 import requests
+import base64
+from io import BytesIO
+from PIL import Image
 from crewai import Agent, Task, Crew, Process, LLM
 from dotenv import load_dotenv
 from stealth_scrape_tool import StealthScrapeTool
@@ -140,6 +143,42 @@ def clean_env_vars():
     os.environ.pop("OPENAI_BASE_URL", None)
     os.environ.pop("OPENAI_MODEL_NAME", None)
 
+js_share_logic = """
+async (image_data_url) => {
+    if (!image_data_url) {
+        console.error("Share button clicked, but no image data URL found.");
+        alert("No image to share. Please generate an image first.");
+        return;
+    }
+
+    const dataUrl = image_data_url;
+
+    try {
+        const response = await fetch(dataUrl);
+        const blob = await response.blob();
+        const file = new File([blob], "shared_image.png", { type: "image/png" });
+
+        if (navigator.share && navigator.canShare({ files: [file] })) {
+            await navigator.share({
+                files: [file],
+                title: "Check out this image!",
+                text: "Created with a cool Gradio app.",
+            });
+        } else {
+            alert("Sharing not supported on this browser.");
+        }
+    } catch (error) {
+        if (error.name === 'AbortError') {
+            console.log("Share dialog cancelled by user.");
+            // Do nothing, as this is a user-initiated cancellation
+        } else {
+            console.error("Error sharing:", error);
+            alert("An error occurred while trying to share.");
+        }
+    }
+}
+"""
+
 # --- Gradio Interface ---
 def generate_ad(product_url: str, store_name: str, main_cupom: str, main_cupom_discount_percentage: float, cupom_1: str, original_price: float, discounted_price: float, openai_api_key: str, natura_api_token: str, openai_base_url: str, openai_model_name: str):
     yield gr.update(interactive=False, value="Generating..."), gr.Markdown(value="⏳ Generating ad... Please wait.")
@@ -175,7 +214,7 @@ with gr.Blocks() as demo:
         original_price_input = gr.Number(label="Original Price (Optional)", value=0, minimum=0)
         discounted_price_input = gr.Number(label="Discounted Price (Optional)", value=0, minimum=0)
         with gr.Row():
-            generate_button = gr.Button("Generate Ad")
+            generate_ad_button = gr.Button("Generate Ad")
             clear_button = gr.Button("Clear")
         ad_output = gr.Markdown(label="Your Generated Ad", show_copy_button=True)
         
@@ -185,6 +224,25 @@ with gr.Blocks() as demo:
         analyze_fragrantica_button = gr.Button("Analyze Fragrantica Product")
         fragrantica_output = gr.Markdown(label="Fragrantica Analysis Report")
 
+    with gr.Tab("Images"):
+        gr.Markdown("### 🖼️ Generate Promotional Image")
+        with gr.Row():
+            with gr.Column():
+                image_product_url_input = gr.Textbox(label="Product Image URL", placeholder="Enter product image URL...")
+                image_product_name_input = gr.Textbox(label="Product Name", placeholder="Enter product name...")
+                image_original_price_input = gr.Number(label="Original Price", placeholder="Enter original price...")
+                image_final_price_input = gr.Number(label="Final Price", placeholder="Enter final price...")
+                image_coupon_code_input = gr.Textbox(label="Coupon Code", placeholder="Enter coupon code...")
+                gen_image_btn = gr.Button("Generate Image")
+            with gr.Column():
+                image_output = gr.Image(label="Generated Image", height=500, type="pil", interactive=False)
+                image_data_url_storage = gr.Textbox(
+                    label="Image Data URL (for debugging)",
+                    visible=True,
+                    elem_id="image_data_url_storage"
+                )
+                share_button = gr.Button("🚀 Share Image", interactive=False)
+
     with gr.Tab("Settings"):
         gr.Markdown("### ⚙️ API Key Settings")
         gr.Markdown("Enter your API keys below. These will be used for the current session.")
@@ -200,21 +258,57 @@ with gr.Blocks() as demo:
     def clear_fields():
         return "", 0, 0
 
-    generate_button.click(generate_ad, inputs=[url_input, store_name_input, main_cupom_input, main_cupom_discount_percentage_input, cupom_1_input, original_price_input, discounted_price_input, openai_key_input, natura_token_input, openai_base_url_input, openai_model_name_input], outputs=[generate_button, ad_output])
-    clear_button.click(clear_fields, inputs=[], outputs=[url_input, original_price_input, discounted_price_input])
-
-    # Placeholder for Fragrantica analysis function
     def analyze_fragrantica_url(url, openai_api_key, natura_api_token, openai_base_url, openai_model_name):
+        yield "⏳ Analyzing Fragrantica product... Please wait." # Loading message
         if not openai_api_key or not openai_model_name or not openai_base_url:
-            return "Please configure your API keys in the settings section below."
+            yield "Please configure your API keys in the settings section below."
+            return
         from fragrantica_crew import FragranticaCrew
         fragrantica_crew = FragranticaCrew(openai_api_key, openai_base_url, openai_model_name)
         report = fragrantica_crew.kickoff(url=url)
         if report == "SCRAPING_FAILED":
-            return "❌ Scraping failed. The website could not be accessed or parsed. Please check the URL or try again later."
-        return report.raw
+            yield "❌ Scraping failed. The website could not be accessed or parsed. Please check the URL or try again later."
+            return
+        yield report.raw
+
+    def image_to_base64(pil_image):
+        print(f"image_to_base64 called. pil_image type: {type(pil_image)}")
+        if pil_image is None:
+            print("pil_image is None. Returning empty string and interactive=False.")
+            return "", gr.update(interactive=False)
+        buffered = BytesIO()
+        pil_image.save(buffered, format="PNG")
+        img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+        data_url = f"data:image/png;base64,{img_str}"
+        print("Returning data_url and interactive=True.")
+        return data_url, gr.update(interactive=True)
+
+    def generate_image(product_image_url, product_name, original_price, final_price, coupon_code):
+        tool = GenerateImageTool()
+        original_price_str = f"{original_price:.2f}".replace('.', ',')
+        final_price_str = f"{final_price:.2f}".replace('.', ',')
+        
+        yield gr.update(interactive=False, value="Generating..."), None
+
+        image_path = tool._run(
+            product_image_url=product_image_url,
+            product_name=product_name,
+            original_price=original_price_str,
+            final_price=final_price_str,
+            coupon_code=coupon_code
+        )
+        
+        pil_image = Image.open(image_path)
+        yield gr.update(interactive=True, value="Generate Image"), pil_image
 
+    generate_ad_button.click(generate_ad, inputs=[url_input, store_name_input, main_cupom_input, main_cupom_discount_percentage_input, cupom_1_input, original_price_input, discounted_price_input, openai_key_input, natura_token_input, openai_base_url_input, openai_model_name_input], outputs=[generate_ad_button, ad_output])
+    clear_button.click(clear_fields, inputs=[], outputs=[url_input, original_price_input, discounted_price_input])
     analyze_fragrantica_button.click(analyze_fragrantica_url, inputs=[fragrantica_url_input, openai_key_input, natura_token_input, openai_base_url_input, openai_model_name_input], outputs=fragrantica_output)
+    gen_image_btn.click(generate_image, 
+                                inputs=[image_product_url_input, image_product_name_input, image_original_price_input, image_final_price_input, image_coupon_code_input], 
+                                outputs=[gen_image_btn, image_output])
+    image_output.change(image_to_base64, inputs=image_output, outputs=[image_data_url_storage, share_button])
+    share_button.click(fn=None, inputs=[image_data_url_storage], js=js_share_logic)
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)

fragrantica_crew.py

@@ -12,7 +12,8 @@ class FragranticaCrew:
         llm = LLM(
             api_key=self.openai_api_key, 
             model=self.openai_model_name, 
-            base_url=self.openai_base_url
+            base_url=self.openai_base_url,
+            max_tokens=32768
         )
 
         self.research_agent = Agent(
@@ -39,7 +40,7 @@ class FragranticaCrew:
     def kickoff(self, url: str) -> str:
         research_task = Task(
             description=(
-                f"""1. Scrape the content of the URL: {url} using the 'Stealth Web Scraper' tool with `website_url` as {url} and `css_element` as "#main-content". If the scraping tool fails or returns empty content ONCE, try the `css_element` as "body", If they also fail when you pass `css_element` as "body", then you MUST return the exact string "SCRAPING_FAILED".
+                f"""1. Scrape the content of the URL: {url} using the 'Stealth Web Scraper' tool with `website_url` as {url}, `css_element` as "#main-content", and `wait_for_selectors` as "['#all-reviews']". If the scraping tool fails or returns empty content ONCE, try the `css_element` as "body", If they also fail when you pass `css_element` as "body", then you MUST return the exact string "SCRAPING_FAILED".
 
                 2. If scraping is successful, carefully analyze the entire page content to extract the following information:
 
@@ -54,7 +55,7 @@ class FragranticaCrew:
 
                    - Este Perfume me Lembra do: Find the section titled "Este perfume me lembra do", and list the perfume names mentioned there.
 
-                   - Opinião dos usuários: Look for a section containing detailed user reviews, such as "Todas as Resenhas por Data" or similar, and synthesize a detailed summary from these reviews.
+                   - Opinião dos usuários: Locate the HTML div with ID "all-reviews" within the scraped content. From this div, extract the *full text content* of all individual user reviews. Then, synthesize a detailed summary from these extracted texts, focusing on common themes, sentiments, and key observations from the users.
 
                 3. Present the extracted information in a clear, structured format, ready for reporting. If any specific piece of information cannot be found, check again to make sure they are not found, after check again, if you truly do not find the info, state 'N/A' for that field. If the entire scraping process fails, return "SCRAPING_FAILED".
 """

image_generator_tool.py

@@ -3,7 +3,7 @@ from pydantic import BaseModel, Field
 from PIL import Image, ImageDraw, ImageFont
 import requests
 from io import BytesIO
-import uuid
+import base64
 
 class GenerateImageToolInput(BaseModel):
     """Input for the Generate Image Tool."""
@@ -14,18 +14,14 @@ class GenerateImageToolInput(BaseModel):
     coupon_code: str = Field(..., description="Coupon code to be displayed on the image.")
 
 import tempfile
-import os
 
 class GenerateImageTool(BaseTool):
     name: str = "Generate Image Tool"
-    description: str = "Generates a promotional image for a product using a template."
+    description: str = "Generates a promotional image for a product using a template and returns the file path."
     args_schema = GenerateImageToolInput
 
     def _run(self, product_image_url: str, product_name: str, original_price: str, final_price: str, coupon_code: str) -> str:
-
-        template_path = 'assets/template_natura_empty.jpg'
-        temp_dir = tempfile.gettempdir()
-        output_path = os.path.join(temp_dir, f'{uuid.uuid4()}.png')
+        template_path = 'assets/template_1.png'
 
         try:
             template_image = Image.open(template_path).convert("RGBA")
@@ -64,15 +60,15 @@ class GenerateImageTool(BaseTool):
             black_color = "#000000"
 
             draw.text((360, 710), product_name, font=font_name, fill=white_color, anchor="ms")
-            draw.text((360, 800), original_price, font=font_price_from, fill=white_color, anchor="ms")
-            draw.text((360, 860), final_price, font=font_price, fill=yellow_color, anchor="ms")
+            draw.text((360, 800), f"De: R$ {original_price}", font=font_price_from, fill=white_color, anchor="ms")
+            draw.text((360, 860), f"Por: R$ {final_price}", font=font_price, fill=yellow_color, anchor="ms")
             draw.text((360, 993), coupon_code, font=font_cupom, fill=black_color, anchor="ms")
 
-            template_image.save(output_path)
-            
-            return output_path
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+                template_image.save(temp_file.name)
+                return temp_file.name
 
         except FileNotFoundError:
             return f"Error: The template file '{template_path}' was not found."
         except Exception as e:
-            return f"An error occurred: {e}"
+            return f"An error occurred: {e[:1000]}"

stealth_scrape_tool.py

@@ -8,38 +8,99 @@ class StealthScrapeTool(BaseTool):
     name: str = "Stealth Web Scraper"
     description: str = "A tool for stealthily scraping content from a given URL using Playwright and a CSS selector."
 
-    async def _arun(self, website_url: str, css_element = "body") -> str:
+    async def _arun(self, website_url: str, css_element = "body", wait_for_selectors: list[str] = None) -> str:
         try:
             async with Stealth().use_async(async_playwright()) as p:
                 browser = await p.chromium.launch(headless=True)
                 page = await browser.new_page()
                 
+                print(f"StealthScrapeTool: Starting scraping for {website_url}...")
+                print(f"StealthScrapeTool: Navigating to {website_url}")
                 await page.goto(website_url, timeout=120000)
+                await asyncio.sleep(5)
+
+                # Scroll to the bottom of the page repeatedly to load all dynamic content
+                print("StealthScrapeTool: Scrolling through the page to load dynamic content...")
+                print("StealthScrapeTool: Getting initial scrollHeight...")
+                last_height = await page.evaluate("document.body.scrollHeight")
+                print(f"StealthScrapeTool: Initial scrollHeight: {last_height}")
+                scroll_attempts = 0
+                max_scroll_attempts = 10
+
+                while scroll_attempts < max_scroll_attempts:
+                    print(f"StealthScrapeTool: Scroll attempt {scroll_attempts + 1}")
+                    print("StealthScrapeTool: Scrolling to bottom...")
+                    await page.evaluate("window.scrollTo(0, document.body.scrollHeight)")
+                    print("StealthScrapeTool: Scrolled. Waiting for content to load...")
+
+                    await asyncio.sleep(5)
+
+                    print("StealthScrapeTool: Getting new scrollHeight...")
+                    new_height = await page.evaluate("document.body.scrollHeight")
+                    print(f"StealthScrapeTool: New scrollHeight: {new_height}")
+                    if new_height == last_height:
+                        print("StealthScrapeTool: ScrollHeight unchanged. Breaking scroll loop.")
+                        break
+                    last_height = new_height
+                    scroll_attempts += 1
+                print("StealthScrapeTool: Finished scrolling.")
+                    
+                print(f"StealthScrapeTool: Page loaded. Attempting to find element with selector '{css_element}'")
+                
+                # Element waiting logic
+                selectors_to_wait_for = []
+                if wait_for_selectors:
+                    print("StealthScrapeTool: Additional selectors to wait for provided.")
+                    selectors_to_wait_for.extend(wait_for_selectors)
+                
+                # Always include css_element in the list of selectors to wait for
+                selectors_to_wait_for.append(css_element)
+
+                combined_selector = ", ".join(selectors_to_wait_for)
+                print(f"StealthScrapeTool: Waiting for selectors: {combined_selector}")
+                await page.wait_for_selector(combined_selector, timeout=60000, state='attached')
                 
-                try:
-                    # Wait for the specific element to be present
-                    await page.wait_for_selector(css_element, timeout=30000)
-                except Exception as e:
-                    # If timeout error, try again with "body" as css_element
-                    if "Timeout" in str(e) and css_element != "body":
-                        await page.wait_for_selector("body", timeout=60000)
-                        css_element = "body"
-                    else:
-                        raise e
                 
+                print("StealthScrapeTool: Required elements found. Extracting content...")
                 html_content = await page.content()
                 soup = BeautifulSoup(html_content, 'html.parser')
-                
+               
+                # Debug print to confirm if waited-for elements are in the scraped content
+                if soup.select_one("#all-reviews"):
+                    print("StealthScrapeTool: #all-reviews found in scraped content.")
+                else:
+                    print("StealthScrapeTool: #all-reviews NOT found in scraped content.")
+
                 target_element = soup.select_one(css_element)
                 if target_element:
+                    # Clean the HTML content
+                    print(f"Successfully found element with selector '{css_element}'. Cleaning content...")
+                    for script in target_element.find_all("script"):
+                        script.decompose()
+                    for style_tag in target_element.find_all("style"):
+                        style_tag.decompose()
+                    for img in target_element.find_all("img"):
+                        img.decompose()
+                    for svg in target_element.find_all("svg"):
+                        svg.decompose()
+                    for iframe in target_element.find_all("iframe"):
+                        iframe.decompose()
+                    for source_tag in target_element.find_all("source"):
+                        source_tag.decompose()
+
+                    # Remove style attributes from all tags
+                    for tag in target_element.find_all(True):
+                        if 'style' in tag.attrs:
+                            del tag['style']
+
                     return target_element.prettify()
                 else:
                     return f"Error: Could not find element with selector '{css_element}' on the page."
         except Exception as e:
             return f"Error during stealth web scraping: {e}"
 
-    def _run(self, website_url: str, css_element: str) -> str:
+    def _run(self, website_url: str, css_element: str, wait_for_selectors: list[str] = None) -> str:
         # This method is for synchronous execution, which is not ideal for Playwright.
         # CrewAI typically calls _arun for async tools.
         # For simplicity, we'll just call the async version here.
-        return asyncio.run(self._arun(website_url, css_element))
+        return asyncio.run(self._arun(website_url, css_element, wait_for_selectors))