add new markdown

1 files changed, 121 insertions, 9 deletions

diff --git a/callbacks.py b/callbacks.py
index 8a06541..c411d75 100644
--- a/callbacks.py
+++ b/callbacks.py
@@ -121,7 +121,15 @@ async def set_model_callback(callback: CallbackQuery):
             
     if user_id == current_user_id:
         current_config = get_user_config(user_id)
-        set_user_config(user_id, new_model, current_config.get("stream", True), current_config.get("call"), current_config.get("prompt"), current_config.get("notify"))
+        set_user_config(
+            user_id, 
+            new_model, 
+            current_config.get("stream", True), 
+            current_config.get("call"), 
+            current_config.get("prompt"), 
+            current_config.get("notify"),
+            current_config.get("markdown")
+        )
         await callback.message.edit_text(f"выбрана модель: `{new_model}`")
         await callback.answer(f"выбрано: {new_model}")
     else:
@@ -188,7 +196,15 @@ async def set_prompt_callback(callback: CallbackQuery):
     new_status = callback.data.split("%")[2]
 
     config = get_user_config(user_id)
-    set_user_config(user_id, config.get("model"), config.get("stream"), config.get("call"), new_status, config.get("notify"))
+    set_user_config(
+        user_id, 
+        config.get("model"), 
+        config.get("stream"), 
+        config.get("call"), 
+        new_status, 
+        config.get("notify"),
+        config.get("markdown")
+    )
     await callback.answer("da поменял на " + new_status)
     await settings_handler(callback)
 
@@ -217,12 +233,26 @@ async def set_stream_callback(callback: CallbackQuery):
     new_status = callback.data.split("%")[2]
 
     config = get_user_config(user_id)
-    if new_status == "native":
-        set_user_config(user_id, config.get("model"), "native", config.get("call"), config.get("prompt"), config.get("notify"))
-    elif new_status == "edit":
-        set_user_config(user_id, config.get("model"), "edit", config.get("call"), config.get("prompt"), config.get("notify"))
+    if new_status == "native" or new_status == "edit":
+        set_user_config(
+            user_id, 
+            config.get("model"), 
+            new_status, 
+            config.get("call"), 
+            config.get("prompt"), 
+            config.get("notify"),
+            config.get("markdown"),
+        )
     else:
-        set_user_config(user_id, config.get("model"), "none", config.get("call"), config.get("prompt"), config.get("notify"))
+        set_user_config(
+            user_id, 
+            config.get("model"), 
+            "none", 
+            config.get("call"), 
+            config.get("prompt"), 
+            config.get("notify"),
+            config.get("markdown"),
+        )
     await settings_handler(callback)
 
 
@@ -238,7 +268,15 @@ async def set_call_callback(callback: CallbackQuery):
     new_status = callback.data.split("%")[2]
 
     config = get_user_config(user_id)
-    set_user_config(user_id, config.get("model"), config.get("stream"), new_status, config.get("prompt"), config.get("notify"))
+    set_user_config(
+        user_id, 
+        config.get("model"), 
+        config.get("stream"), 
+        new_status, 
+        config.get("prompt"), 
+        config.get("notify"),
+        config.get("markdown"),
+    )
     await settings_handler(callback)
 
 
@@ -254,10 +292,84 @@ async def set_notify_callback(callback: CallbackQuery):
     new_status = callback.data.split("%")[2]
 
     config = get_user_config(user_id)
-    set_user_config(user_id, config.get("model"), config.get("stream"), config.get("call"), config.get("prompt"), new_status)
+    set_user_config(
+        user_id, 
+        config.get("model"), 
+        config.get("stream"), 
+        config.get("call"), 
+        config.get("prompt"), 
+        new_status,
+        config.get("markdown"),
+    )
+    await settings_handler(callback)
+
+
+@router.callback_query(F.data.startswith("set_markdown%"))
+async def set_markdown_callback(callback: CallbackQuery):
+    current_user_id = str(callback.from_user.id)
+    data = callback.data.split("%", 2)
+    user_id = data[1]
+    if user_id != current_user_id:
+        await callback.answer("нет, не трогай это не твое")
+        return
+
+    new_status = data[2]
+
+    config = get_user_config(user_id)
+    set_user_config(
+        user_id, 
+        config.get("model"), 
+        config.get("stream"), 
+        config.get("call"), 
+        config.get("prompt"), 
+        config.get("notify"), 
+        new_status,
+    )
     await settings_handler(callback)
 
 
+@router.callback_query(F.data.startswith("new_md%"))
+async def set_markdown_callback(callback: CallbackQuery):
+    current_user_id = str(callback.from_user.id)
+    data = callback.data.split("%", 2)
+    user_id = data[1]
+    if user_id != current_user_id:
+        await callback.answer("нет, не трогай это не твое")
+        return
+    config = get_user_config(user_id)
+    set_user_config(
+        user_id, 
+        config.get("model"), 
+        config.get("stream"), 
+        config.get("call"), 
+        config.get("prompt"), 
+        config.get("notify"), 
+        "new",
+    )
+    await callback.answer("выбран новый стиль!!")
+
+
+@router.callback_query(F.data.startswith("old_md%"))
+async def set_markdown_callback(callback: CallbackQuery):
+    current_user_id = str(callback.from_user.id)
+    data = callback.data.split("%", 2)
+    user_id = data[1]
+    if user_id != current_user_id:
+        await callback.answer("нет, не трогай это не твое")
+        return
+    config = get_user_config(user_id)
+    set_user_config(
+        user_id, 
+        config.get("model"), 
+        config.get("stream"), 
+        config.get("call"), 
+        config.get("prompt"), 
+        config.get("notify"), 
+        "old",
+    )
+    await callback.answer("выбран старый стиль!!")
+
+
 @router.callback_query(F.data == "start")
 async def start_callback(callback: CallbackQuery):
     await callback.message.edit_text(