add imgbb cdn

.gitignore

@@ -0,0 +1,115 @@
+# Ignore environment variable files
+.env
+
+# Ignore log files
+bot.log
+
+# Ignore Python test scripts
+test.py
+output_image.jpg
+# Ignore cache directory
+cache/
+archive/
+
+# Igonre pycache
+__pycache__
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# Virtual environments
+venv/
+ENV/
+env/
+venv.bak/
+venv.bak/
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is intended to run in multiple environments; 
+#   otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies 
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not 
+#   install all needed dependencies.
+Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/

app.py

@@ -4,8 +4,9 @@ import os
 import threading
 import yaml
 from image_payload_reader import read_payload_from_image
-from telegram import start_telegram_bot
+#from telegram import start_telegram_bot
 from utils import read_config
+from imgbb import extract_original_image_urls
 
 app = Flask(__name__)
 templates_dir = "templates"
@@ -39,8 +40,8 @@ def start_bots():
     telegram_bot_enabled = info['telegram_bot_enabled']
     
     if telegram_bot_enabled:
-        start_telegram_bot()
-
+        #start_telegram_bot()
+        pass
     # Note: WhatsApp bot functionality is not implemented in this Flask version
 
 # Start the Telegram bot in a separate thread when the application starts
@@ -48,7 +49,7 @@ threading.Thread(target=start_bots).start()
 
 # Function to fetch image files and corresponding YAML data from 'cache' directory
 def fetch_image_files():
-    files = os.listdir(CACHE_DIR)
+    files = extract_original_image_urls()
     image_files = [f for f in files if f.endswith(('.jpg', '.jpeg', '.png'))]
     response_data = []
 
@@ -82,7 +83,7 @@ def get_images():
 
 @app.route('/cache/<image_name>')
 def get_image(image_name):
-    file_path = os.path.join(CACHE_DIR, image_name)
+    file_path = image_name
     if os.path.exists(file_path):
         return send_file(file_path)
     return jsonify({"error": "File not found"}), 404
@@ -96,7 +97,7 @@ def get_static_image(image_name):
 
 @app.route('/yaml/<image_name>')
 def get_embedded_yaml(image_name):
-    file_path = os.path.join(CACHE_DIR, image_name)
+    file_path = image_name
     if not os.path.exists(file_path):
         return jsonify({"error": "Image file not found"}), 404
     

captioner.py

@@ -0,0 +1,61 @@
+import os
+import requests
+import logging
+from dotenv import load_dotenv
+import time
+
+# Load environment variables from a .env file
+load_dotenv()
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Define constants
+API_ENDPOINT = os.getenv("API_ENDPOINT", "https://chandimaprabath-florence-2-sd3-captioner-cpu.hf.space/generate")
+
+def get_image_caption(image_path):
+    """
+    Send a POST request to the API endpoint with the image and return the caption.
+
+    Args:
+        image_path (str): Path to the image file.
+        api_endpoint (str): URL of the API endpoint.
+
+    Returns:
+        str: The caption generated by the API.
+    """
+    try:
+        with open(image_path, "rb") as image_file:
+            files = {"image": image_file}
+            
+            # Start timer
+            start_time = time.time()
+            
+            response = requests.post(API_ENDPOINT, files=files)
+            response.raise_for_status()  # Raise HTTPError for bad responses
+            
+            # End timer
+            end_time = time.time()
+            elapsed_time = end_time - start_time
+            
+            data = response.json()
+            caption = data.get('caption', 'No caption found')
+            
+            return caption, elapsed_time
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Request failed: {e}")
+        return None, None
+    except Exception as e:
+        logger.error(f"An error occurred: {e}")
+        return None, None
+
+if __name__ == "__main__":
+    IMAGE_PATH = "alchemy_refiner_alchemy_magic_0_3ed4af60-070e-47ed-b3de-a158103efa7b_0.jpg"
+    # Get caption for the specified image
+    caption, elapsed_time = get_image_caption(IMAGE_PATH)
+    if caption:
+        logger.info(f"Caption: {caption}")
+        logger.info(f"Time taken: {elapsed_time:.2f} seconds")
+    else:
+        logger.error("Failed to get caption.")

config.yaml

@@ -21,7 +21,7 @@ config:
     cfg_scale: 7.0
     imgbb_upload: false
   app:
-    whatsapp_bot_enabled: true
-    telegram_bot_enabled: true
+    whatsapp_bot_enabled: false
+    telegram_bot_enabled: false
     version: "0.6 Beta"
     debug: true

imgbb.py

@@ -1,7 +1,7 @@
 import requests
 from bs4 import BeautifulSoup
 
-def extract_original_image_urls(url):
+def extract_original_image_urls(url='https://unicone-studio.imgbb.com/'):
     try:
         # Make a GET request to fetch the raw HTML content
         response = requests.get(url)

old.telegram.py

@@ -0,0 +1,108 @@
+import logging
+import os
+import threading
+import telebot
+from llm import generate_llm  # Assuming these are your custom functions
+from sd import generate_sd    # Replace with actual imports as needed
+import time
+
+# Load environment variables
+def load_env():
+    global TELEGRAM_BOT_TOKEN
+    TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+    if not TELEGRAM_BOT_TOKEN:
+        raise ValueError("TELEGRAM_BOT_TOKEN environment variable is not set")
+
+load_env()
+
+# Initialize the bot
+bot = telebot.TeleBot(TELEGRAM_BOT_TOKEN)
+
+# Configure logging
+LOG_FILE = "bot.log"
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler(LOG_FILE),
+        logging.StreamHandler()
+    ]
+)
+logger = logging.getLogger(__name__)
+
+# Configure SOCKS5 Proxy (replace with your proxy details)
+PROXY_HOST = '23.19.244.109'
+PROXY_PORT = 1080
+
+telebot.apihelper.proxy = {'https': f'socks5://{PROXY_HOST}:{PROXY_PORT}'}
+
+def response_text(chat_id, prompt):
+    """
+    Generate a response using the LLM and send it to the user.
+    """
+    try:
+        msg = generate_llm(prompt)  # Example function call to generate a response
+        bot.send_message(chat_id, msg)
+        logger.info(f"Chat ID: {chat_id} - Sent Message: {msg}")
+    except Exception as e:
+        logger.error(f"Chat ID: {chat_id} - Error in response_text: {e}")
+        bot.send_message(chat_id, "There was an error processing your request.")
+
+@bot.message_handler(commands=['imagine'])
+def handle_imagine_command(message):
+    """
+    Handle the /imagine command by generating an image from the provided prompt.
+    """
+    prompt = message.text.replace('/imagine', '').strip()
+    if not prompt:
+        bot.reply_to(message, "Please provide a prompt after /imagine.")
+        logger.info(f"Chat ID: {message.chat.id} - Sent Message: Please provide a prompt after /imagine.")
+        return
+
+    generating_message = bot.send_message(message.chat.id, "Generating...")
+    logger.info(f"Chat ID: {message.chat.id} - Sent Message: Generating...")
+
+    def handle_image_generation():
+        try:
+            # Example function call to generate image data
+            image_data, image_path = generate_sd(prompt)
+            bot.delete_message(chat_id=message.chat.id, message_id=generating_message.message_id)
+
+            if image_data:
+                bot.send_photo(message.chat.id, image_data)
+                logger.info(f"Chat ID: {message.chat.id} - Sent Image {image_path}")
+            else:
+                bot.reply_to(message, "Failed to generate image. Please try again later.")
+                logger.error(f"Chat ID: {message.chat.id} - Failed to generate image.")
+        except Exception as e:
+            logger.error(f"Chat ID: {message.chat.id} - Error in handle_image_generation: {e}")
+            bot.reply_to(message, "There was an error generating the image. Please try again later.")
+
+    threading.Thread(target=handle_image_generation).start()
+
+@bot.message_handler(func=lambda message: True)
+def handle_message(message):
+    """
+    Handle all text messages by generating a response and sending it.
+    """
+    chat_id = message.chat.id
+    user_message = message.text
+    logger.info(f"Chat ID: {chat_id} - Received Message: {user_message}")
+
+    threading.Thread(target=response_text, args=(chat_id, user_message)).start()
+
+def start_telegram_bot():
+    """
+    Start the Telegram bot.
+    """
+    while True:
+        try:
+            logger.info("Starting bot")
+            bot.polling(none_stop=True)
+        except Exception as e:
+            logger.critical(f"Critical error: {e}")
+            time.sleep(5)  # Wait for a few seconds before restarting
+
+# Ensure the bot starts when this script is executed directly
+if __name__ == '__main__':
+    start_telegram_bot()

requirements.txt

@@ -6,4 +6,7 @@ pillow
 pyTelegramBotAPI
 PyYAML
 watchdog
-PySocks
+telethon
+asyncio
+python-socks[asyncio]
+beautifulsoup4

telegram.py

@@ -1,23 +1,22 @@
 import logging
 import os
-import threading
-import telebot
+import asyncio
+from telethon import TelegramClient, events, sync
 from llm import generate_llm  # Assuming these are your custom functions
 from sd import generate_sd    # Replace with actual imports as needed
-import time
+import socks
 
 # Load environment variables
 def load_env():
-    global TELEGRAM_BOT_TOKEN
+    global TELEGRAM_API_ID, TELEGRAM_API_HASH, TELEGRAM_BOT_TOKEN
+    TELEGRAM_API_ID = int(os.getenv("TELEGRAM_API_ID"))
+    TELEGRAM_API_HASH = os.getenv("TELEGRAM_API_HASH")
     TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
-    if not TELEGRAM_BOT_TOKEN:
-        raise ValueError("TELEGRAM_BOT_TOKEN environment variable is not set")
+    if not all([TELEGRAM_API_ID, TELEGRAM_API_HASH, TELEGRAM_BOT_TOKEN]):
+        raise ValueError("One or more required environment variables are not set")
 
 load_env()
 
-# Initialize the bot
-bot = telebot.TeleBot(TELEGRAM_BOT_TOKEN)
-
 # Configure logging
 LOG_FILE = "bot.log"
 logging.basicConfig(
@@ -30,79 +29,68 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-# Configure SOCKS5 Proxy (replace with your proxy details)
-PROXY_HOST = '23.19.244.109'
-PROXY_PORT = 1080
-
-telebot.apihelper.proxy = {'https': f'socks5://{PROXY_HOST}:{PROXY_PORT}'}
-
-def response_text(chat_id, prompt):
-    """
-    Generate a response using the LLM and send it to the user.
-    """
-    try:
-        msg = generate_llm(prompt)  # Example function call to generate a response
-        bot.send_message(chat_id, msg)
-        logger.info(f"Chat ID: {chat_id} - Sent Message: {msg}")
-    except Exception as e:
-        logger.error(f"Chat ID: {chat_id} - Error in response_text: {e}")
-        bot.send_message(chat_id, "There was an error processing your request.")
-
-@bot.message_handler(commands=['imagine'])
-def handle_imagine_command(message):
-    """
-    Handle the /imagine command by generating an image from the provided prompt.
-    """
-    prompt = message.text.replace('/imagine', '').strip()
-    if not prompt:
-        bot.reply_to(message, "Please provide a prompt after /imagine.")
-        logger.info(f"Chat ID: {message.chat.id} - Sent Message: Please provide a prompt after /imagine.")
-        return
-
-    generating_message = bot.send_message(message.chat.id, "Generating...")
-    logger.info(f"Chat ID: {message.chat.id} - Sent Message: Generating...")
-
-    def handle_image_generation():
-        try:
-            # Example function call to generate image data
-            image_data, image_path = generate_sd(prompt)
-            bot.delete_message(chat_id=message.chat.id, message_id=generating_message.message_id)
-
-            if image_data:
-                bot.send_photo(message.chat.id, image_data)
-                logger.info(f"Chat ID: {message.chat.id} - Sent Image {image_path}")
-            else:
-                bot.reply_to(message, "Failed to generate image. Please try again later.")
-                logger.error(f"Chat ID: {message.chat.id} - Failed to generate image.")
-        except Exception as e:
-            logger.error(f"Chat ID: {message.chat.id} - Error in handle_image_generation: {e}")
-            bot.reply_to(message, "There was an error generating the image. Please try again later.")
-
-    threading.Thread(target=handle_image_generation).start()
-
-@bot.message_handler(func=lambda message: True)
-def handle_message(message):
-    """
-    Handle all text messages by generating a response and sending it.
-    """
-    chat_id = message.chat.id
-    user_message = message.text
-    logger.info(f"Chat ID: {chat_id} - Received Message: {user_message}")
-
-    threading.Thread(target=response_text, args=(chat_id, user_message)).start()
-
-def start_telegram_bot():
-    """
-    Start the Telegram bot.
-    """
-    while True:
-        try:
-            logger.info("Starting bot")
-            bot.polling(none_stop=True)
-        except Exception as e:
-            logger.critical(f"Critical error: {e}")
-            time.sleep(5)  # Wait for a few seconds before restarting
+proxy_server = '142.93.68.63'
+proxy_port = 2434
+proxy_secret = 'ee32b920dffb51643028e2f6b878d4eac1666172616b61762e636f6d'
+proxy_dc_id = 2  # This is usually 2 for MTProto proxies
+
+proxy = (
+    socks.SOCKS5,
+    proxy_server,
+    proxy_port,
+    True,
+    'vpn',
+    'unlimited'
+)
 
-# Ensure the bot starts when this script is executed directly
+async def main():
+    # Initialize Telethon client
+    client = TelegramClient('bot', TELEGRAM_API_ID, TELEGRAM_API_HASH, proxy=proxy).start(TELEGRAM_BOT_TOKEN)
+
+    @client.on(events.NewMessage)
+    async def handle_message(event):
+        chat_id = event.chat_id
+        user_message = event.raw_text
+        logger.info(f"Chat ID: {chat_id} - Received Message: {user_message}")
+
+        # Handle /imagine command
+        if user_message.startswith('/imagine'):
+            prompt = user_message.replace('/imagine', '').strip()
+            if not prompt:
+                await event.reply("Please provide a prompt after /imagine.")
+                logger.info(f"Chat ID: {chat_id} - Sent Message: Please provide a prompt after /imagine.")
+                return
+
+            generating_message = await event.reply("Generating...")
+            logger.info(f"Chat ID: {chat_id} - Sent Message: Generating...")
+
+            try:
+                # Example function call to generate image data
+                image_data, image_path = generate_sd(prompt)
+                await client.delete_messages(chat_id, [generating_message.id])
+
+                if image_data:
+                    await client.send_file(chat_id, image_data)
+                    logger.info(f"Chat ID: {chat_id} - Sent Image {image_path}")
+                else:
+                    await event.reply("Failed to generate image. Please try again later.")
+                    logger.error(f"Chat ID: {chat_id} - Failed to generate image.")
+            except Exception as e:
+                logger.error(f"Chat ID: {chat_id} - Error in handle_image_generation: {e}")
+                await event.reply("There was an error generating the image. Please try again later.")
+
+        else:
+            # Default response handling
+            try:
+                msg = generate_llm(user_message)  # Example function call to generate a response
+                await event.reply(msg)
+                logger.info(f"Chat ID: {chat_id} - Sent Message: {msg}")
+            except Exception as e:
+                logger.error(f"Chat ID: {chat_id} - Error in response_text: {e}")
+                await event.reply("There was an error processing your request.")
+
+    await client.run_until_disconnected()
+
+# Run the main function
 if __name__ == '__main__':
-    start_telegram_bot()
+    asyncio.run(main())

templates/index.html

@@ -268,7 +268,7 @@ document.addEventListener('DOMContentLoaded', () => {
         container.className = 'image-item';
 
         const img = document.createElement('img');
-        img.src = `/cache/${image}`;
+        img.src = `${image}`;
         img.alt = image;
         img.onerror = () => {
             container.style.display = 'none';