фикс я хуй знает много чего я добавил

1 files changed, 292 insertions, 1 deletions

diff --git a/helpers.py b/helpers.py
index 57f8b30..a65890a 100644
--- a/helpers.py
+++ b/helpers.py
@@ -1,11 +1,14 @@
+import re
 import time
+import shelve
 import requests
 
 from aiogram.methods import AnswerGuestQuery, SendPoll
 from aiogram.types import InlineQueryResultArticle, InputTextMessageContent, InputRichMessageContent, InputRichMessage, Message
 from aiogram.enums import ChatType
 
-from supergenerator import *
+from vars import *
+from config import *
 
 
 def is_blocked(user_id):
@@ -162,3 +165,291 @@ async def mans(message, text):
     else:
         return await message.reply(text)
     
+def graph_emoji(status):
+    match status:
+        case "done":
+            return "✅"
+        case "error":
+            return "🚫"
+        case "make":
+            return "💭"
+        case "wait":
+            return "⏳"
+        case _:
+            return "🥺"
+
+
+def make_graph(message, process):
+    text = ""
+    replied = message.reply_to_message or None
+    user_id = message.from_user.id
+    config = get_user_config(user_id)
+    im_model = IMAGE_MODEL.split("*", 1)
+    model = config["model"].split("*", 1)
+    photo_a = process.get("photo_a", None)
+    photo = process.get("photo", None)
+    stt = process.get("stt", None)
+    t = process.get("text", None)
+    
+    if photo_a:
+        text += f"{graph_emoji(photo_a)} анализ фото (в ответе)\n"
+    if photo:
+        text += f"{graph_emoji(photo)} анализ фото\n"
+    if stt:
+        text += f"{graph_emoji(stt)} stt (голос -> текст)\n"
+        
+    text += f"{graph_emoji(t)} запрос `{model[1]}` (`{model[0]}`)\n"
+    
+    if replied and replied.document:
+        text += f"↳ {replied.document.file_size} файл (в ответе)\n"
+    if message.document:
+        text += f"↳ {message.document.file_size} файл\n"
+        
+    if replied and replied.photo:
+        text += "↳ фото (в ответе)\n"
+    if message.photo:
+        text += "↳ фото\n"
+        
+    if replied and replied.location:
+        text += "↳ геолокация (в ответе)\n"
+    if message.location:
+        text += "↳ геолокация\n"
+        
+    if replied and replied.poll:
+        text += "↳ опрос (в ответе)\n"
+    if message.poll:
+        text += "↳ опрос\n"  
+      
+    if replied and replied.voice:
+        text += "↳ голосовое сообщение (в ответе)\n"
+    if message.voice:
+        text += "↳ голосовое сообщение\n"
+      
+    if replied and replied.text:
+        text += f"↳ {len(replied.text)} текст в ответе\n"
+
+    return text
+
+
+def clean_tail_repeats(text: str, max_repeat: int = 10):
+    if not text:
+        return text, 0
+
+    original_length = len(text)
+    cleaned = text
+
+    pattern = rf"(.)\1{{{max_repeat},}}$"
+
+    def limit_repeats(match):
+        char = match.group(1)
+        return char * max_repeat
+
+    cleaned = re.sub(pattern, limit_repeats, cleaned)
+
+    pattern_space = rf"((\S)\s+)\2{{{max_repeat},}}$"
+    cleaned = re.sub(
+        pattern_space,
+        lambda m: (m.group(2) + " ") * min(max_repeat, len(m.group(1))),
+        cleaned,
+    )
+
+    words = cleaned.split()
+    if len(words) >= 2:
+        for i in range(1, min(len(words), 10)):
+            if all(words[-j] == words[-j - 1] for j in range(i)):
+                unique_words = words[: -(i + 1)]
+                cleaned = " ".join(unique_words + [words[-1]])
+                break
+
+    removed_count = original_length - len(cleaned)
+
+    cleaned = re.sub(r" +", " ", cleaned)
+    cleaned = cleaned.strip()
+
+    return cleaned, removed_count
+
+
+def parse_response(text: str):
+    if "connection error." in text:
+        return {
+            "type": "connection",
+            "message": "ошибка подключения к провайдеру. подожди или смени провайдера в настройках узбекгпт.",
+        }
+    elif "model quota exceeded" in text:
+        return {
+            "type": "quota",
+            "message": "лимит токенов закончился. подожди или смени модель.",
+        }
+    elif "tier capacity exceeded." in text:
+        return {"type": "tier", "message": "попробуй ещё раз!"}
+    elif "internal server" in text:
+        return {
+            "type": "int",
+            "message": "произошла ошибка на стороне провайдера. подожди или смени провайдера в настройках узбекгпт.",
+        }
+    elif "(incomplete chunked read)" in text:
+        return {
+            "type": "incomplete",
+            "message": "сервер петух и оборвал соединение. подожди или смени провайдера в настройках узбекгпт.",
+        }
+    elif "provider error" in text:
+        return {
+            "type": "provider",
+            "message": "врат ошибка на стороне провайдера модели. смени провайдера или модель.",
+        }
+    elif "rate limit" in text:
+        return {
+            "type": "quota",
+            "message": "лимит токенов закончился у провайдера врат. подожди или смени модель.",
+        }
+    else:
+        return {
+            "type": "unknown",
+            "message": f"произошла какая-то ошибка при генерации... {text}",
+        }
+
+
+def galockinator(text):
+    for _ in range(random.randint(1, 3)):
+        if random.random() < 0.3:
+            text += "☝️"
+        else:
+            text += "✅"
+
+    return text
+
+
+def remove_think_tags(text):
+    result = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL)
+    result = re.sub(r"<thought>.*?</thought>", "", text, flags=re.DOTALL)
+    return result
+
+
+def get_all_users(only_notify=False):
+    with shelve.open("users") as db:
+        return {k: db[k] for k in db.keys() if not only_notify or db[k].get("notify") == "da"}
+
+
+def get_all_user_ids(only_notify=False):
+    ids = []
+    with shelve.open("users") as db:
+        for user_id in db.keys():
+            if only_notify and db[user_id].get("notify") != "da":
+                continue
+            try:
+                ids.append(int(user_id))
+            except ValueError:
+                pass
+    return ids
+
+
+def set_user_config(
+    user_id, 
+    model_name, 
+    stream=DEFAULT_STREAM, 
+    call=DEFAULT_CALL, 
+    prompt=DEFAULT_PROMPT,
+    notify=DEFAULT_NOTIFY,
+    markdown=DEFAULT_MARKDOWN,
+    prompts={},
+):
+    with shelve.open("users") as db:
+        db[str(user_id)] = {
+            "model": model_name,
+            "stream": stream,
+            "call": call,
+            "prompt": prompt,
+            "notify": notify,
+            "markdown": markdown,
+            "prompts": prompts,
+        }
+
+
+
+def set_user_setting(
+    user_id, 
+    model=None, 
+    stream=None, 
+    call=None, 
+    prompt=None,
+    notify=None,
+    markdown=None,
+    prompts=None,
+):
+    with shelve.open("users") as db:
+        user_id = str(user_id)
+        if user_id not in db:
+            db[user_id] = {}
+        
+        settings = db[user_id]
+        
+        if model is not None: settings["model"] = model
+        if stream is not None: settings["stream"] = stream
+        if call is not None: settings["call"] = call
+        if prompt is not None: settings["prompt"] = prompt
+        if notify is not None: settings["notify"] = notify
+        if markdown is not None: settings["markdown"] = markdown
+        if prompts is not None: settings["prompts"] = prompts
+
+        db[user_id] = settings
+
+
+def delete_user_from_db(user_id):
+    with shelve.open("users") as db:
+        if str(user_id) in db:
+            del db[str(user_id)]
+
+
+def reset_user_setting(user_id, key, value):
+    with shelve.open("users") as db:
+        user = db[str(user_id)]
+        user[key] = value
+        db[str(user_id)] = user
+
+
+def ensure_user(
+    user_id,
+    model_name=DEFAULT_MODEL,
+    stream_mode=DEFAULT_STREAM,
+    call_mode=DEFAULT_CALL,
+    prompt_name=DEFAULT_PROMPT,
+    notify_mode=DEFAULT_NOTIFY,
+    markdown=DEFAULT_MARKDOWN,
+    prompts={},
+):
+    with shelve.open("users") as db:
+        key = str(user_id)
+
+        if key not in db:
+            db[key] = {
+                "model": model_name,
+                "stream": stream_mode,
+                "call": call_mode,
+                "prompt": prompt_name,
+                "notify": notify_mode,
+                "markdown": markdown,
+                "prompts": prompts,
+            }
+
+
+def get_user_config(user_id):
+    with shelve.open("users") as db:
+        config = db.get(str(user_id), {})
+        defaults = {
+            "model": DEFAULT_MODEL,
+            "stream": DEFAULT_STREAM,
+            "call": DEFAULT_CALL,
+            "prompt": DEFAULT_PROMPT,
+            "notify": DEFAULT_NOTIFY,
+            "markdown": DEFAULT_MARKDOWN,
+            "prompts": {},
+        }
+        updated = False
+        for key, value in defaults.items():
+            if key not in config:
+                config[key] = value
+                updated = True
+        if updated:
+            db[str(user_id)] = config
+    return config
+