diff --git a/ai_agent_tutorials/ai_meme_generator_browseruse/ai_meme_generator.py b/ai_agent_tutorials/ai_meme_generator_browseruse/ai_meme_generator.py index 81a0d1e..7a20191 100644 --- a/ai_agent_tutorials/ai_meme_generator_browseruse/ai_meme_generator.py +++ b/ai_agent_tutorials/ai_meme_generator_browseruse/ai_meme_generator.py @@ -1,16 +1,31 @@ import asyncio import streamlit as st from browser_use import Agent, SystemPrompt -from langchain_google_genai import ChatGoogleGenerativeAI from langchain_openai import ChatOpenAI from langchain_anthropic import ChatAnthropic +from langchain_core.messages import HumanMessage import re -async def generate_meme(query: str, api_key: str) -> None: - llm = ChatAnthropic( - model="claude-3-5-sonnet-20241022", - api_key=api_key - ) +async def generate_meme(query: str, model_choice: str, api_key: str) -> None: + # Initialize the appropriate LLM based on user selection + if model_choice == "Claude": + llm = ChatAnthropic( + model="claude-3-5-sonnet-20241022", + api_key=api_key + ) + elif model_choice == "Deepseek": + llm = ChatOpenAI( + base_url='https://api.deepseek.com/v1', + model='deepseek-chat', + api_key=api_key, + temperature=0.3 + ) + else: # OpenAI + llm = ChatOpenAI( + model="gpt-4o", + api_key=api_key, + temperature=0.0 + ) task_description = ( "You are a meme generator expert. You are given a query and you need to generate a meme for it.\n" @@ -29,7 +44,8 @@ async def generate_meme(query: str, api_key: str) -> None: task=task_description, llm=llm, max_actions_per_step=5, - max_failures=25 + max_failures=25, + use_vision=(model_choice != "Deepseek") ) history = await agent.run() @@ -41,42 +57,74 @@ async def generate_meme(query: str, api_key: str) -> None: url_match = re.search(r'https://imgflip\.com/i/(\w+)', final_result) if url_match: meme_id = url_match.group(1) - # Convert to direct image URL format return f"https://i.imgflip.com/{meme_id}.jpg" return None def main(): - st.title("AI Meme Generator - Browser Use") - st.info("This AI browser agent does browser automation to generate memes based on your input with browser use. Please enter your API key and describe the meme you want to generate.") + # Custom CSS styling - # Configuration Settings + + st.title("🤖 AI Meme Generator - Browser Use Web Agent") + st.info("This AI browser agent does browser automation to generate memes based on your input with browser use. Please enter your API key and describe the meme you want to generate.") + + # Sidebar configuration with st.sidebar: - st.header("⚙️ Configuration Settings") - api_key = st.text_input("Enter your Claude API Key", type="password") + st.markdown('

⚙️ Model Configuration

', unsafe_allow_html=True) + + # Model selection + model_choice = st.selectbox( + "Select AI Model", + ["Claude", "Deepseek", "OpenAI"], + index=0, + help="Choose which LLM to use for meme generation" + ) + + # API key input based on model selection + api_key = "" + if model_choice == "Claude": + api_key = st.text_input("Claude API Key", type="password", + help="Get your API key from https://console.anthropic.com") + elif model_choice == "Deepseek": + api_key = st.text_input("Deepseek API Key", type="password", + help="Get your API key from https://platform.deepseek.com") + else: + api_key = st.text_input("OpenAI API Key", type="password", + help="Get your API key from https://platform.openai.com") # Main content area - st.markdown('

🎨 Describe the Meme You Want to Generate

', unsafe_allow_html=True) - query = st.text_input("Enter your meme idea (e.g., 'Ilya's SSI quietly looking at the OpenAI vs Deepseek debate while diligently working on ASI')", "") + st.markdown('

🎨 Describe Your Meme Concept

', unsafe_allow_html=True) + + query = st.text_input( + "Meme Idea Input", + placeholder="Example: 'Ilya's SSI quietly looking at the OpenAI vs Deepseek debate while diligently working on ASI'", + label_visibility="collapsed" + ) - if st.button("Generate Meme"): - if api_key and query: - with st.spinner("🤖 Generating your meme... This might take a minute"): - try: - meme_url = asyncio.run(generate_meme(query, api_key)) + if st.button("Generate Meme 🚀"): + if not api_key: + st.warning(f"Please provide the {model_choice} API key") + st.stop() + if not query: + st.warning("Please enter a meme idea") + st.stop() + + with st.spinner(f"🧠 {model_choice} is generating your meme..."): + try: + meme_url = asyncio.run(generate_meme(query, model_choice, api_key)) + + if meme_url: + st.success("✅ Meme Generated Successfully!") + st.image(meme_url, caption="Generated Meme Preview", use_container_width=True) + st.markdown(f""" + **Direct Link:** [Open in ImgFlip]({meme_url}) + **Embed URL:** `{meme_url}` + """) + else: + st.error("❌ Failed to generate meme. Please try again with a different prompt.") - if meme_url: - st.success("🎉 Meme Generated Successfully!") - st.image(meme_url, caption="Your Generated Meme", use_container_width=True) - - # Display clickable link - st.markdown(f"**Meme Link:** [Open in ImgFlip]({meme_url})") - else: - st.error("Could not retrieve meme URL. Please try again.") - - except Exception as e: - st.error(f"Error generating meme: {str(e)}") - else: - st.warning("⚠️ Please provide both API key and meme idea") + except Exception as e: + st.error(f"Error: {str(e)}") + st.info("💡 If using OpenAI, ensure your account has GPT-4o access") if __name__ == '__main__': main() \ No newline at end of file diff --git a/ai_agent_tutorials/ai_meme_generator_browseruse/requirements.txt b/ai_agent_tutorials/ai_meme_generator_browseruse/requirements.txt new file mode 100644 index 0000000..e69de29 diff --git a/ai_agent_tutorials/ai_meme_generator_browseruse/test.py b/ai_agent_tutorials/ai_meme_generator_browseruse/test.py deleted file mode 100644 index 38cb890..0000000 --- a/ai_agent_tutorials/ai_meme_generator_browseruse/test.py +++ /dev/null @@ -1,35 +0,0 @@ -import asyncio -import os - -from dotenv import load_dotenv -from langchain_openai import ChatOpenAI -from pydantic import SecretStr - -from browser_use import Agent - -# dotenv -load_dotenv() - -api_key = os.getenv('DEEPSEEK_API_KEY', '') -if not api_key: - raise ValueError('DEEPSEEK_API_KEY is not set') - - -async def run_search(): - agent = Agent( - task=('go to amazon.com, search for laptop, sort by best rating, and give me the price of the first result'), - llm=ChatOpenAI( - base_url='https://api.deepseek.com/v1', - model='deepseek-reasoner', - api_key=SecretStr(api_key), - ), - use_vision=False, - max_failures=2, - max_actions_per_step=1, - ) - - await agent.run() - - -if __name__ == '__main__': - asyncio.run(run_search()) \ No newline at end of file