Update app.py

app.py

@@ -1,240 +1,101 @@
 import streamlit as st
-import asyncio
-import nest_asyncio
-nest_asyncio.apply()
-
-import re
-import os
 import requests
-from telethon import TelegramClient
-from telethon.sessions import StringSession
-import streamlit.components.v1 as components
-
-# ----------- Настройки Telegram -----------
-TELEGRAM_API_ID = 23646321
-TELEGRAM_API_HASH = "102edf31d404cb72d17ee4831d4e8c7b"
-# Замените на вашу сгенерированную StringSession (одна строка без переносов)
-TELEGRAM_STRING_SESSION = "1ApWapzMBu8a9o9Wu1GexPdkO7Jsr6po1xym67EcfAxkEbVx1oxw6wW7LINb9Xf0VYcyEAP-b-D0RQ_4x0RDAW8Di_0P2r4bj7xNeV9xO3wBMcaeq7FJux3UMq5buq89gqG8JarZG04tIyWoVnH__dq1_6G2Zkr0GF4WJCS2UN7mmwBacIIUEjUryEA97tXYC8jD14J-1MDgFsNNZTJ6uurD5RzH2DIhphChebxOkDBdXmpLGlanEzYzVOlx4p6Mo2zYorRG968o64CxKEh-WiJaPSNrS3IEfIVWFVd_tHKWzQSPD1zHHhcw0v2_ZjCKbZnTMcT3uMYuyLwA55_dckzcw0wZgAZQ="
-
-# ----------- Настройки ВКонтакте -----------
-VK_ACCESS_TOKEN = os.environ.get("VK_ACCESS_TOKEN", "")
-if not VK_ACCESS_TOKEN:
-    st.warning("Не найден VK_ACCESS_TOKEN. Добавьте секрет 'VK_ACCESS_TOKEN' в настройках Spaces!")
-VK_API_VERSION = "5.131"
-
-# --- Инициализация Telethon-клиента (Telegram) ---
-def init_telethon():
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    client = TelegramClient(StringSession(TELEGRAM_STRING_SESSION), TELEGRAM_API_ID, TELEGRAM_API_HASH)
-    loop.run_until_complete(client.connect())
-    if not loop.run_until_complete(client.is_user_authorized()):
-        st.error("Telegram-сессия не авторизована. Проверьте StringSession или выполните login_telethon.py.")
-        st.stop()
-    return client, loop
-
-if "telethon_client" not in st.session_state:
-    st.session_state["telethon_client"], st.session_state["telethon_loop"] = init_telethon()
-tg_client = st.session_state["telethon_client"]
-tg_loop = st.session_state["telethon_loop"]
-
-def human_format_views(num: int) -> str:
-    if num < 1000:
-        return f"{num / 1000:.1f}K"
-    elif num < 10_000:
-        return f"{num / 1000:.1f}K"
-    elif num < 1_000_000:
-        return f"{int(num / 1000)}K"
-    else:
-        return f"{num / 1_000_000:.1f}M"
-
-# ------------- Парсер Telegram -------------
-def process_telegram_link(link: str) -> tuple[str, int]:
-    link = link.strip()
-    if not link:
-        return "", 0
-
-    # Поддержка ссылок вида t.me, telegram.me (без учета регистра)
-    match = re.match(r'https?://(?:t(?:elegram)?\.me)/([^/]+)/(\d+)', link, re.IGNORECASE)
-    if not match:
-        return f"**Ошибка (TG)**: не распознал ссылку — {link}", 0
-
-    channel_username = match.group(1)
-    message_id = int(match.group(2))
-    global tg_client, tg_loop
-    try:
-        msg = tg_loop.run_until_complete(tg_client.get_messages(channel_username, ids=message_id))
-    except RuntimeError as e:
-        if "Event loop is closed" in str(e):
-            tg_client, tg_loop = init_telethon()
-            st.session_state["telethon_client"] = tg_client
-            st.session_state["telethon_loop"] = tg_loop
-            msg = tg_loop.run_until_complete(tg_client.get_messages(channel_username, ids=message_id))
-        else:
-            return f"**Ошибка (TG)** при обработке [{link}]({link}): {e}", 0
-    if msg is None:
-        return f"**Ошибка (TG)**: сообщение не найдено — {link}", 0
-    try:
-        channel_entity = tg_loop.run_until_complete(tg_client.get_entity(channel_username))
-    except RuntimeError as e:
-        if "Event loop is closed" in str(e):
-            tg_client, tg_loop = init_telethon()
-            st.session_state["telethon_client"] = tg_client
-            st.session_state["telethon_loop"] = tg_loop
-            channel_entity = tg_loop.run_until_complete(tg_client.get_entity(channel_username))
-        else:
-            return f"**Ошибка (TG)** при обработке [{link}]({link}): {e}", 0
-    channel_title = channel_entity.title
-    views = msg.views or 0
-    result = f"- [{channel_title}]({link}) — {human_format_views(views)}"
-    return result, views
-
-# ------------- Парсер VK -------------
-def process_vk_link(link: str) -> tuple[str, int]:
-    link = link.strip()
-    if not link:
-        return "", 0
-
-    match = re.search(r'vk\.com/wall(-?\d+)_(\d+)', link)
-    if not match:
-        return f"**Ошибка (VK)**: не распознал ссылку — {link}", 0
-
-    owner_id = int(match.group(1))
-    post_id = match.group(2)
-    posts_param = f"{owner_id}_{post_id}"
-    api_url = "https://api.vk.com/method/wall.getById"
-    params = {
-        "posts": posts_param,
-        "access_token": VK_ACCESS_TOKEN,
-        "v": VK_API_VERSION,
-        "extended": 1
-    }
-    try:
-        resp = requests.get(api_url, params=params)
-        if resp.status_code != 200:
-            return f"**Ошибка (VK)**: HTTP {resp.status_code} — {link}", 0
-        data = resp.json()
-        if "error" in data:
-            error_msg = data["error"].get("error_msg", "Неизвестная ошибка")
-            return f"**Ошибка (VK)**: {error_msg} — {link}", 0
-        response_data = data.get("response", {})
-        items = response_data.get("items", [])
-        if not items:
-            return f"**Ошибка (VK)**: пост не найден — {link}", 0
-        post = items[0]
-        views = post.get("views", {}).get("count", 0)
-        post_owner_id = post.get("owner_id", owner_id)
-        title = None
-        if post_owner_id < 0:
-            group_id = -post_owner_id
-            for g in response_data.get("groups", []):
-                if g["id"] == group_id:
-                    title = g["name"]
-                    break
-        else:
-            user_id = post_owner_id
-            for p in response_data.get("profiles", []):
-                if p["id"] == user_id:
-                    first_name = p.get("first_name", "")
-                    last_name = p.get("last_name", "")
-                    title = (first_name + " " + last_name).strip()
-                    break
-        if not title:
-            title = "VK пост"
-        result = f"- [{title}]({link}) — {human_format_views(views)}"
-        return result, views
-    except Exception as e:
-        return f"**Ошибка (VK)** при обработке [{link}]({link}): {e}", 0
-
-# ------------- Определение платформы -------------
-def detect_platform(link: str) -> str:
-    link = link.strip().lower()
-    if "t.me/" in link or "telegram.me/" in link:
-        return "telegram"
-    elif "vk.com/wall" in link:
-        return "vk"
-    return ""
-
-# ---------- Интерфейс Streamlit ----------
-st.title("PR Tool made by @Dina")
-st.write("Вставьте ссылки (каждая на новой строке) — как Telegram, так и ВКонтакте. Пример:")
-st.code("""
-https://t.me/meduzalive/122479
-https://vk.com/wall-12256840_3220312
-https://telegram.me/za_traktor/27423
-https://vk.com/wall-237426_1643234
+import urllib.parse
+import hashlib
+import base64
+import secrets
+
+def generate_code_verifier(length=128):
+    return secrets.token_urlsafe(length)[:length]
+
+def generate_code_challenge(code_verifier):
+    hashed = hashlib.sha256(code_verifier.encode('utf-8')).digest()
+    return base64.urlsafe_b64encode(hashed).decode('utf-8').rstrip('=')
+
+st.title("VK ID Authorization with PKCE — Manual Input for PKCE Parameters")
+
+st.markdown("""
+Введите вручную сгенерированные вами значения **code_verifier** и **code_challenge**.
+**Пошагово:**
+1. Сгенерируйте с вашей стороны `code_verifier` и вычислите `code_challenge` (например, через Python или онлайн-инструмент).
+2. Вставьте их ниже.
+3. Нажмите кнопку для перехода на страницу авторизации.
+4. После авторизации VK вернёт вас на redirect_uri с параметрами `?code=...&device_id=...`.
+5. Скопируйте полученные `code` и `device_id` в соответствующие поля и нажмите «Обменять code», чтобы получить токены.
 """)
-links_input = st.text_area("Ссылки:", height=200)
-
-if st.button("Обработать"):
-    lines = links_input.strip().split('\n')
-    tg_results = []
-    tg_total_views = 0
-    vk_results = []
-    vk_total_views = 0
-
-    for link in lines:
-        platform = detect_platform(link)
-        if platform == "telegram":
-            res_line, views = process_telegram_link(link)
-            tg_results.append((res_line, views))
-            tg_total_views += views
-        elif platform == "vk":
-            res_line, views = process_vk_link(link)
-            vk_results.append((res_line, views))
-            vk_total_views += views
-        else:
-            st.markdown(f"**Ошибка**: не распознал платформу — {link}", unsafe_allow_html=True)
-
-    tg_results.sort(key=lambda x: x[1], reverse=True)
-    vk_results.sort(key=lambda x: x[1], reverse=True)
 
-    # Формирование итогового Markdown-вывода
-    md_lines = []
-
-    md_lines.append("## Telegram")
-    md_lines.append(f"**Суммарный охват постов — {human_format_views(tg_total_views)} просмотров**")
-    if tg_results:
-        for r_line, _ in tg_results:
-            md_lines.append(r_line)
-    else:
-        md_lines.append("_Нет ссылок на Telegram_")
-    md_lines.append("")
-
-    md_lines.append("## ВКонтакте")
-    md_lines.append(f"**Суммарный охват постов — {human_format_views(vk_total_views)} просмотров**")
-    if vk_results:
-        for r_line, _ in vk_results:
-            md_lines.append(r_line)
-    else:
-        md_lines.append("_Нет ссылок на ВКонтакте_")
-    md_lines.append("")
-
-    st.markdown("### Результат:")
-    for line in md_lines:
-        st.markdown(line, unsafe_allow_html=True)
-
-    markdown_result = "\n".join(md_lines)
+# Генерация code_verifier и code_challenge
+code_verifier = generate_code_verifier()
+code_challenge = generate_code_challenge(code_verifier)
+
+# Поля для ручного ввода PKCE-параметров
+code_verifier_input = st.text_input("Введите code_verifier", value=code_verifier)
+code_challenge_input = st.text_input("Введите code_challenge", value=code_challenge)
+
+# Параметры приложения
+client_id = st.text_input("client_id", value="53202706")
+client_secret = st.text_input("client_secret", value="piaMXHEDj0afa4QObJCD")
+redirect_uri = st.text_input("redirect_uri", value="https://huggingface.co/spaces/PRAnalytics/prtooltest")
+scope = st.text_input("scope", value="wall")
+
+st.markdown("### 1) Авторизация")
+auth_params = {
+    "client_id": client_id,
+    "redirect_uri": redirect_uri,
+    "response_type": "code",
+    "scope": scope,
+    "state": "my_state",
+    "code_challenge": code_challenge,
+    "code_challenge_method": "S256",
+    "response_mode": "query"
+}
+auth_url = "https://id.vk.com/authorize?" + urllib.parse.urlencode(auth_params, safe="-._~")
+st.markdown(f"[Перейти к авторизации]({auth_url})")
+
+st.markdown("""
+После авторизации VK вернёт вас на redirect_uri с параметрами:  
+`?code=...&device_id=...&state=...`
+""")
 
-    # HTML/JS-компонент для копирования в буфер обмена с изменением текста кнопки
-    copy_button_html = f"""
-    <html>
-      <head>
-        <script>
-          function copyToClipboard() {{
-            const btn = document.getElementById("copy-btn");
-            const text = `{markdown_result}`;
-            navigator.clipboard.writeText(text).then(function() {{
-              btn.innerText = "Скопировано успешно";
-            }}, function(err) {{
-              btn.innerText = "Ошибка при копировании";
-            }});
-          }}
-        </script>
-      </head>
-      <body>
-        <button id="copy-btn" onclick="copyToClipboard()" style="padding: 8px 16px; font-size: 16px;">Скопировать код</button>
-      </body>
-    </html>
-    """
-    components.html(copy_button_html, height=80)
+# Поля для ручного ввода полученных значений
+user_code = st.text_input("Введите полный code, полученный из URL", value="")
+user_device_id = st.text_input("Введите полный device_id, полученный из URL", value="")
+
+st.write("Введённый code:", user_code)
+st.write("Введённый device_id:", user_device_id)
+
+# Отладочный вывод перед обменом
+st.markdown("#### Отладочный вывод перед обменом:")
+st.write("client_id:", client_id)
+st.write("client_secret:", client_secret)
+st.write("redirect_uri:", redirect_uri)
+st.write("scope:", scope)
+st.write("code_verifier:", code_verifier)
+st.write("code_challenge:", code_challenge)
+
+if user_code:
+    st.markdown("### 2) Обмен кода на токены")
+    if st.button("Обменять code"):
+        url = "https://id.vk.com/access_token"  # Правильный endpoint для обмена токенов
+        data = {
+            "grant_type": "authorization_code",
+            "client_id": client_id,
+            "client_secret": client_secret,
+            "redirect_uri": redirect_uri,
+            "code": user_code,
+            "code_verifier": code_verifier
+        }
+        if user_device_id:
+            data["device_id"] = user_device_id
+        st.write("Отправляем запрос на:", url)
+        st.write("Параметры запроса:", data)
+        resp = requests.post(url, data=data)
+        st.write("Status code:", resp.status_code)
+        st.write("Response text:", resp.text)
+        try:
+            tokens = resp.json()
+            st.write("Ответ (JSON):")
+            st.json(tokens)
+        except Exception as e:
+            st.error(f"Ошибка при декодировании ответа: {e}")
+            st.text(resp.text)