UI additions
This commit is contained in:
Madhu
parent
8bb26cca49
commit
5450e8c9fa
3 changed files with 82 additions and 39 deletions
|
|
@ -0,0 +1,82 @@
|
|||
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
|
||||
import re
|
||||
|
||||
async def generate_meme(query: str, api_key: str) -> None:
|
||||
llm = ChatAnthropic(
|
||||
model="claude-3-5-sonnet-20240620",
|
||||
api_key=api_key
|
||||
)
|
||||
|
||||
task_description = (
|
||||
"You are a meme generator expert. You are given a query and you need to generate a meme for it.\n"
|
||||
"1. Go to https://imgflip.com/memetemplates \n"
|
||||
"2. Click on the Search bar in the middle and search for ONLY ONE MAIN ACTION VERB (like 'bully', 'laugh', 'cry') in this query: '{0}'\n"
|
||||
"3. Choose any meme template that metaphorically fits the meme topic: '{0}'\n"
|
||||
" by clicking on the 'Add Caption' button below it\n"
|
||||
"4. Write a Top Text (setup/context) and Bottom Text (punchline/outcome) related to '{0}'.\n"
|
||||
"5. Check the preview making sure it is funny and a meaningful meme. Adjust text directly if needed. \n"
|
||||
"6. Look at the meme and text on it, if it doesnt make sense, PLEASE retry by filling the text boxes with different text. \n"
|
||||
"7. Click on the Generate meme button to generate the meme\n"
|
||||
"8. Copy the image link and give it as the output\n"
|
||||
).format(query)
|
||||
|
||||
agent = Agent(
|
||||
task=task_description,
|
||||
llm=llm,
|
||||
max_actions_per_step=5,
|
||||
max_failures=25
|
||||
)
|
||||
|
||||
history = await agent.run()
|
||||
|
||||
# Extract final result from agent history
|
||||
final_result = history.final_result()
|
||||
|
||||
# Use regex to find the meme URL in the result
|
||||
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.")
|
||||
|
||||
# Configuration Settings
|
||||
with st.sidebar:
|
||||
st.header("⚙️ Configuration Settings")
|
||||
api_key = st.text_input("Enter your Claude API Key", type="password")
|
||||
|
||||
# Main content area
|
||||
st.markdown('<p style="font-size: 20px; font-weight: bold;">🎨 Describe the Meme You Want to Generate</p>', 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')", "")
|
||||
|
||||
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 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")
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
@ -1,39 +0,0 @@
|
|||
import asyncio
|
||||
from browser_use import Agent, SystemPrompt
|
||||
from langchain_google_genai import ChatGoogleGenerativeAI
|
||||
from langchain_openai import ChatOpenAI
|
||||
from langchain_anthropic import ChatAnthropic
|
||||
|
||||
async def generate_meme(query: str) -> None:
|
||||
api_key = "sk-A" # Replace with your actual DeepSeek API key
|
||||
llm = ChatAnthropic(
|
||||
model="claude-3-5-sonnet-20240620",
|
||||
api_key=api_key
|
||||
)
|
||||
|
||||
task_description = (
|
||||
"You are a meme generator expert. You are given a query and you need to generate a meme for it.\n"
|
||||
"1. Go to https://imgflip.com/memetemplates \n"
|
||||
"2. Click on the Search bar in the middle and search for ONLY ONE MAIN ACTION VERB (like 'bully', 'laugh', 'cry') in this query: '{0}'\n"
|
||||
"3. Choose any meme template that metaphorically fits the meme topic: '{0}'\n"
|
||||
" by clicking on the 'Add Caption' button below it\n"
|
||||
"4. Write a Top Text (setup/context) and Bottom Text (punchline/outcome) related to '{0}'.\n"
|
||||
"5. Check the preview making sure it is funny and a meaningful meme. Adjust text directly if needed. \n"
|
||||
"6. Look at the meme and text on it, if it doesnt make sense, PLEASE retry by filling the text boxes with different text. \n"
|
||||
"7. Click on the Generate meme button to generate the meme\n"
|
||||
"8. Copy the image link and give it as the output\n"
|
||||
).format(query)
|
||||
|
||||
agent = Agent(
|
||||
task=task_description,
|
||||
llm=llm,
|
||||
max_actions_per_step=5,
|
||||
max_failures=25
|
||||
)
|
||||
|
||||
history = await agent.run()
|
||||
history.final_result()
|
||||
|
||||
if __name__ == '__main__':
|
||||
query = "Elon Musk bullying sam altman about trump winning the election"
|
||||
asyncio.run(generate_meme(query))
|
||||
Loading…
Reference in a new issue