Create app.py

app.py

@@ -0,0 +1,154 @@
+import streamlit as st
+import os
+from dotenv import load_dotenv, set_key
+import requests
+import asyncio
+from telegram import Update
+from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes
+
+# Page config
+st.set_page_config(page_title="Meme Bot Setup", page_icon="🎭")
+
+# Load environment variables
+load_dotenv()
+
+# Initialize session state
+if 'bot_running' not in st.session_state:
+    st.session_state.bot_running = False
+
+def format_meme_prompt(user_input):
+    """Format user input into a meme-specific prompt"""
+    return f"Create a funny meme that shows: {user_input}. Make it in classic meme style with impact font text. Make it humorous and shareable. The image should be meme-worthy and entertaining."
+
+def save_credentials(telegram_token, openai_key):
+    """Save credentials to .env file"""
+    try:
+        with open('.env', 'w') as f:
+            f.write(f'TELEGRAM_BOT_TOKEN="{telegram_token}"\n')
+            f.write(f'IMAGE_API_KEY="{openai_key}"\n')
+        return True
+    except Exception as e:
+        st.error(f"Error saving credentials: {str(e)}")
+        return False
+
+async def test_meme_generation(prompt):
+    """Test meme generation with OpenAI API"""
+    api_key = os.getenv("IMAGE_API_KEY")
+    if not api_key:
+        return None, "OpenAI API key not found"
+    
+    formatted_prompt = format_meme_prompt(prompt)
+    headers = {"Authorization": f"Bearer {api_key}"}
+    data = {
+        "prompt": formatted_prompt,
+        "size": "1024x1024",
+        "model": "dall-e-3"
+    }
+    
+    try:
+        response = requests.post(
+            "https://api.openai.com/v1/images/generate",
+            json=data,
+            headers=headers,
+            timeout=15
+        )
+        if response.status_code == 200:
+            return response.json()["data"][0]["url"], None
+        else:
+            return None, f"API Error: {response.status_code} - {response.text}"
+    except Exception as e:
+        return None, f"Request failed: {str(e)}"
+
+# Main UI
+st.title("🎭 Meme Bot Setup")
+st.markdown("Configure your Telegram bot to generate memes!")
+
+# Credentials Section
+st.header("Bot Configuration")
+with st.form("credentials_form"):
+    telegram_token = st.text_input(
+        "Telegram Bot Token",
+        value=os.getenv("TELEGRAM_BOT_TOKEN", ""),
+        type="password",
+        help="Get this from @BotFather on Telegram"
+    )
+    openai_key = st.text_input(
+        "OpenAI API Key",
+        value=os.getenv("IMAGE_API_KEY", ""),
+        type="password",
+        help="Get this from OpenAI Platform"
+    )
+    save_button = st.form_submit_button("Save Credentials")
+
+if save_button:
+    if telegram_token and openai_key:
+        if save_credentials(telegram_token, openai_key):
+            st.success("Credentials saved successfully!")
+            load_dotenv()
+    else:
+        st.error("Please provide both API keys")
+
+# Test Meme Generation
+st.header("Test Meme Generation")
+test_prompt = st.text_input(
+    "Enter your meme idea",
+    placeholder="e.g., 'When you forget to save your work'"
+)
+
+if st.button("Generate Test Meme"):
+    if not os.getenv("IMAGE_API_KEY"):
+        st.error("Please set the OpenAI API key first")
+    else:
+        with st.spinner("Creating your meme..."):
+            image_url, error = asyncio.run(test_meme_generation(test_prompt))
+            if error:
+                st.error(error)
+            else:
+                st.image(image_url, caption="Your Generated Meme")
+                st.success("Meme generated successfully!")
+
+# Bot Status
+st.header("Bot Status")
+col1, col2 = st.columns(2)
+
+with col1:
+    if st.button("Start Bot" if not st.session_state.bot_running else "Stop Bot"):
+        if not st.session_state.bot_running:
+            if not os.getenv("TELEGRAM_BOT_TOKEN"):
+                st.error("Please set the Telegram Bot Token first")
+            else:
+                st.session_state.bot_running = True
+                st.success("Bot started!")
+                st.rerun()
+        else:
+            st.session_state.bot_running = False
+            st.warning("Bot stopped")
+            st.rerun()
+
+with col2:
+    status = "🟢 Running" if st.session_state.bot_running else "🔴 Stopped"
+    st.write(f"Status: {status}")
+
+# Instructions
+with st.expander("How to Use"):
+    st.markdown("""
+    ### Setup Steps:
+    1. Get a Telegram Bot Token from [@BotFather](https://t.me/botfather)
+    2. Get an OpenAI API key from [OpenAI Platform](https://platform.openai.com)
+    3. Enter both keys above and save
+    4. Start the bot
+    
+    ### Using the Bot:
+    1. Find your bot on Telegram
+    2. Send /start to begin
+    3. Send any text to generate a meme
+    
+    ### Example Prompts:
+    - "When the code works on first try"
+    - "Me explaining why I need another monitor"
+    - "That moment when you forget to git commit"
+    """)
+
+# Footer
+st.markdown("---")
+st.markdown("Made for generating awesome memes 🎭")