Update UI to support models that are not from OpenAI, based on LiteLLM.

app.py

@@ -1,13 +1,12 @@
 import gradio as gr
 import os
 import requests
-from crewai import Agent, Task, Crew, Process
+from dotenv import load_dotenv
+
+load_dotenv()
+from crewai import Agent, Task, Crew, Process, LLM
 from crewai_tools import ScrapeWebsiteTool
 from crewai.tools import BaseTool
-from langchain_openai import ChatOpenAI
-
-
-# --- CrewAI Integration ---
 
 class ShortenerTool(BaseTool):
     name: str = "URL Shortener Tool"
@@ -46,13 +45,25 @@ class CalculateDiscountedPriceTool(BaseTool):
         return round(discounted_price, 2)
 
 class SocialMediaCrew:
-    def __init__(self, openai_api_key: str, natura_api_token: str):
+    def __init__(self, openai_api_key: str, natura_api_token: str, openai_base_url: str, openai_model_name: str):
         self.openai_api_key = openai_api_key
         self.natura_api_token = natura_api_token
+        self.openai_base_url = openai_base_url
+        self.openai_model_name = openai_model_name
         self.scrape_tool = ScrapeWebsiteTool()
         self.shortener_tool = ShortenerTool(natura_api_token=self.natura_api_token)
         self.calculate_discounted_price_tool = CalculateDiscountedPriceTool()
 
+        print("Initializing SocialMediaCrew with BASE URL:", self.openai_base_url)
+        print("Using OpenAI Model:", self.openai_model_name)
+        print("Using OpenAI Key:", self.openai_api_key[:10])
+
+        llm = LLM(
+            api_key=self.openai_api_key, 
+            model=self.openai_model_name, 
+            base_url=self.openai_base_url
+        )
+
         self.product_analyst = Agent(
             role='Product Analyst',
             goal='Analyze the provided URL and extract key product information',
@@ -60,7 +71,7 @@ class SocialMediaCrew:
             verbose=True,
             tools=[self.scrape_tool, self.shortener_tool, self.calculate_discounted_price_tool],
             allow_delegation=False,
-            llm=ChatOpenAI(api_key=self.openai_api_key, model="gpt-4o-mini")
+            llm=llm
         )
 
         self.social_media_copywriter = Agent(
@@ -69,7 +80,7 @@ class SocialMediaCrew:
             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,
-            llm=ChatOpenAI(api_key=self.openai_api_key, model="gpt-4o-mini")
+            llm=llm
         )
 
     def run_crew(self, product_url: str, main_cupom: str, main_cupom_discount_percentage: float, cupom_1: str, cupom_2: str) -> str:
@@ -111,7 +122,7 @@ class SocialMediaCrew:
 De ~~{{ORIGINAL PRICE}}~~
 🔥Por {{CUPOM DISCOUNTED PRICE}} 🔥
 
-❤️ USE O CUPOM >>> {main_cupom}
+🎟️ USE O CUPOM >>> {main_cupom}
 
 🛒 Link >>> {{short_url}}
 
@@ -129,20 +140,25 @@ 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,
-            openai_api_key=self.openai_api_key
+            process=Process.sequential
         )
 
         print(f"Crew is kicking off for URL: {product_url}")
         result = crew.kickoff()
         return result
 
+def clean_env_vars():
+    os.environ.pop("OPENAI_API_KEY", None)
+    os.environ.pop("NATURA_API_TOKEN", None)
+    os.environ.pop("OPENAI_BASE_URL", None)
+    os.environ.pop("OPENAI_MODEL_NAME", None)
+
 # --- Gradio Interface ---
-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:
+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, openai_base_url: str, openai_model_name: str):
+    if not openai_api_key or not natura_api_token or not openai_model_name or not openai_base_url:
         return "Please configure your API keys in the settings section below."
 
-    social_media_crew = SocialMediaCrew(openai_api_key, natura_api_token)
+    social_media_crew = SocialMediaCrew(openai_api_key, natura_api_token, openai_base_url, openai_model_name)
     result = social_media_crew.run_crew(product_url, main_cupom, main_cupom_discount_percentage, cupom_1, cupom_2)
 
     if result == "INVALID_URL":
@@ -169,12 +185,15 @@ with gr.Blocks() as demo:
     with gr.Tab("Settings"):
         gr.Markdown("### ⚙️ API Key Settings")
         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")
+        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", ""))
+        openai_base_url_input = gr.Textbox(label="OPENAI_BASE_URL", value=os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1"))
+        openai_model_name_input = gr.Textbox(label="OPENAI_MODEL_NAME", value=os.getenv("OPENAI_MODEL_NAME", "gpt-4o-mini"))
         
+        clean_env_vars()
         # 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)
+    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, openai_base_url_input, openai_model_name_input], outputs=ad_output)
 
 demo.launch()

requirements.txt

@@ -1,3 +1,4 @@
 crewai>=0.148.0
 crewai-tools>=0.55.0
-gradio>=5.38.0
+gradio>=5.38.0
+litellm