Refactor: Pass API keys directly to Gradio app and CrewAI agents besides use ENV.

app.py

@@ -4,30 +4,19 @@ import requests
 from crewai import Agent, Task, Crew, Process
 from crewai_tools import ScrapeWebsiteTool
 from crewai.tools import BaseTool
+from langchain_openai import ChatOpenAI
 
-os.environ["OPENAI_MODEL_NAME"] = "gpt-4o-mini"
-
-# --- API Key Management ---
-def get_api_keys():
-    return {
-        "OPENAI_API_KEY": os.getenv("OPENAI_API_KEY", ""),
-        "NATURA_API_TOKEN": os.getenv("NATURA_API_TOKEN", "")
-    }
-
-def set_api_key(openai_key: str, natura_token: str):
-    os.environ["OPENAI_API_KEY"] = openai_key
-    os.environ["NATURA_API_TOKEN"] = natura_token
-    return "API keys saved successfully!"
 
 # --- CrewAI Integration ---
 
 class ShortenerTool(BaseTool):
     name: str = "URL Shortener Tool"
     description: str = "Generates a short version of a given URL using an external API."
+    natura_api_token: str
 
     def _run(self, original_url: str) -> str:
         api_url = "https://sales-mgmt-cb-bff-apigw.prd.naturacloud.com/cb-bff-cms/cms/shortener"
-        headers = {"authorization": f"Bearer {os.getenv('NATURA_API_TOKEN')}", "content-type": "application/json"}
+        headers = {"authorization": f"Bearer {self.natura_api_token}", "content-type": "application/json"}
         payload = {"url": original_url}
 
         try:
@@ -57,9 +46,11 @@ class CalculateDiscountedPriceTool(BaseTool):
         return round(discounted_price, 2)
 
 class SocialMediaCrew:
-    def __init__(self):
+    def __init__(self, openai_api_key: str, natura_api_token: str):
+        self.openai_api_key = openai_api_key
+        self.natura_api_token = natura_api_token
         self.scrape_tool = ScrapeWebsiteTool()
-        self.shortener_tool = ShortenerTool()
+        self.shortener_tool = ShortenerTool(natura_api_token=self.natura_api_token)
         self.calculate_discounted_price_tool = CalculateDiscountedPriceTool()
 
         self.product_analyst = Agent(
@@ -68,7 +59,8 @@ class SocialMediaCrew:
             backstory=("You are an expert in analyzing product pages and extracting the most important information. You can identify the product name, its main features, and the target audience."),
             verbose=True,
             tools=[self.scrape_tool, self.shortener_tool, self.calculate_discounted_price_tool],
-            allow_delegation=False
+            allow_delegation=False,
+            llm=ChatOpenAI(api_key=self.openai_api_key, model="gpt-4o-mini")
         )
 
         self.social_media_copywriter = Agent(
@@ -76,7 +68,8 @@ class SocialMediaCrew:
             goal='Create a compelling social media post in Portuguese to sell the product',
             backstory=("You are a creative copywriter specialized in the beauty and fragrance market. You know how to craft posts that are engaging, persuasive, and tailored for a Portuguese-speaking audience. You are an expert in using emojis and hashtags to increase engagement."),
             verbose=True,
-            allow_delegation=False
+            allow_delegation=False,
+            llm=ChatOpenAI(api_key=self.openai_api_key, model="gpt-4o-mini")
         )
 
     def run_crew(self, product_url: str, main_cupom: str, main_cupom_discount_percentage: float, cupom_1: str, cupom_2: str) -> str:
@@ -136,7 +129,8 @@ Ensure a URL is always present in the output. Include a clear call to action and
         crew = Crew(
             agents=[self.product_analyst, self.social_media_copywriter],
             tasks=[analyze_product_task, create_post_task],
-            process=Process.sequential
+            process=Process.sequential,
+            openai_api_key=self.openai_api_key
         )
 
         print(f"Crew is kicking off for URL: {product_url}")
@@ -144,12 +138,11 @@ Ensure a URL is always present in the output. Include a clear call to action and
         return result
 
 # --- Gradio Interface ---
-def generate_ad(product_url: str, main_cupom: str, main_cupom_discount_percentage: float, cupom_1: str, cupom_2: str):
-    api_keys = get_api_keys()
-    if not api_keys["OPENAI_API_KEY"] or not api_keys["NATURA_API_TOKEN"]:
+def generate_ad(product_url: str, main_cupom: str, main_cupom_discount_percentage: float, cupom_1: str, cupom_2: str, openai_api_key: str, natura_api_token: str):
+    if not openai_api_key or not natura_api_token:
         return "Please configure your API keys in the settings section below."
 
-    social_media_crew = SocialMediaCrew()
+    social_media_crew = SocialMediaCrew(openai_api_key, natura_api_token)
     result = social_media_crew.run_crew(product_url, main_cupom, main_cupom_discount_percentage, cupom_1, cupom_2)
 
     if result == "INVALID_URL":
@@ -175,13 +168,13 @@ with gr.Blocks() as demo:
         
     with gr.Tab("Settings"):
         gr.Markdown("### ⚙️ API Key Settings")
-        gr.Markdown("Enter your API keys below. These will be stored as environment variables for the running application.")
-        openai_key_input = gr.Textbox(label="OPENAI_API_KEY", type="password", value=os.getenv("OPENAI_API_KEY"))
-        natura_token_input = gr.Textbox(label="NATURA_API_TOKEN", type="password", value=os.getenv("NATURA_API_TOKEN"))
-        save_button = gr.Button("Save Keys")
-        settings_message = gr.Markdown()
-        save_button.click(set_api_key, inputs=[openai_key_input, natura_token_input], outputs=settings_message)
-
-    generate_button.click(generate_ad, inputs=[url_input, main_cupom_input, main_cupom_discount_percentage_input, cupom_1_input, cupom_2_input], outputs=ad_output)
+        gr.Markdown("Enter your API keys below. These will be used for the current session.")
+        openai_key_input = gr.Textbox(label="OPENAI_API_KEY", type="password")
+        natura_token_input = gr.Textbox(label="NATURA_API_TOKEN", type="password")
+        
+        # No save button needed as keys are passed directly
+        gr.Markdown("API keys are used directly from these fields when you click 'Generate Ad'. They are not saved persistently.")
+
+    generate_button.click(generate_ad, inputs=[url_input, main_cupom_input, main_cupom_discount_percentage_input, cupom_1_input, cupom_2_input, openai_key_input, natura_token_input], outputs=ad_output)
 
 demo.launch()