Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import asyncio
 import nest_asyncio
-nest_asyncio.apply()  # Позволяет переиспользовать один event loop
+nest_asyncio.apply()
 
 import re
 import os
@@ -53,32 +53,52 @@ def process_telegram_link(link: str) -> tuple[str, int]:
     link = link.strip()
     if not link:
         return "", 0
-    # Распознаем ссылки вида t.me и telegram.me (без учета регистра)
+
+    # Поддержка ссылок вида 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))
-        if msg is None:
-            return f"**Ошибка (TG)**: сообщение не найдено — {link}", 0
+    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))
-        channel_title = channel_entity.title
-        views = msg.views or 0
-        result = f"- [{channel_title}]({link}) — {human_format_views(views)}"
-        return result, views
-    except Exception as e:
-        return f"**Ошибка (TG)** при обработке [{link}]({link}): {e}", 0
+    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}"
@@ -135,33 +155,6 @@ def detect_platform(link: str) -> str:
         return "vk"
     return ""
 
-# ----------- Кастомные стили (CSS) -----------
-st.markdown(
-    """
-    <style>
-    .section-header {
-        font-size: 1.4rem;
-        color: #333;
-        margin-bottom: 5px;
-    }
-    .total-views {
-        font-size: 1.1rem;
-        font-weight: bold;
-        color: #444;
-        margin-bottom: 10px;
-    }
-    .post-item {
-        margin-left: 20px;
-        margin-bottom: 4px;
-    }
-    .result-title {
-        font-size: 1.6rem;
-        color: #222;
-        margin-top: 20px;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
 # ---------- Интерфейс Streamlit ----------
 st.title("PR Tool made by @Dina")
 st.write("Вставьте ссылки (каждая на новой строке) — как Telegram, так и ВКонтакте. Пример:")
@@ -171,12 +164,10 @@ https://vk.com/wall-12256840_3220312
 https://telegram.me/za_traktor/27423
 https://vk.com/wall-237426_1643234
 """)
-
 links_input = st.text_area("Ссылки:", height=200)
 
 if st.button("Обработать"):
     lines = links_input.strip().split('\n')
-    
     tg_results = []
     tg_total_views = 0
     vk_results = []
@@ -198,31 +189,31 @@ if st.button("Обработать"):
     tg_results.sort(key=lambda x: x[1], reverse=True)
     vk_results.sort(key=lambda x: x[1], reverse=True)
 
-    # Формируем разделы в два столбца
-    st.markdown("### Результат:")
-    col1, col2 = st.columns(2)
-    with col1:
-        st.markdown("<div class='section-header'>Telegram</div>", unsafe_allow_html=True)
-        st.markdown(f"<div class='total-views'>Суммарный охват постов — {human_format_views(tg_total_views)} просмотров</div>", unsafe_allow_html=True)
-        for r_line, _ in tg_results:
-            st.markdown(f"<div class='post-item'>{r_line}</div>", unsafe_allow_html=True)
-    with col2:
-        st.markdown("<div class='section-header'>ВКонтакте</div>", unsafe_allow_html=True)
-        st.markdown(f"<div class='total-views'>Суммарный охват постов — {human_format_views(vk_total_views)} просмотров</div>", unsafe_allow_html=True)
-        for r_line, _ in vk_results:
-            st.markdown(f"<div class='post-item'>{r_line}</div>", unsafe_allow_html=True)
-
-    # Формируем итоговый Markdown-код для копирования
+    # Формирование итогового Markdown-вывода
     md_lines = []
+
     md_lines.append("## Telegram")
     md_lines.append(f"**Суммарный охват постов — {human_format_views(tg_total_views)} просмотров**")
-    for r_line, _ in tg_results:
-        md_lines.append(r_line)
+    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)} просмотров**")
-    for r_line, _ in vk_results:
-        md_lines.append(r_line)
+    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)
 
     # HTML/JS-компонент для копирования в буфер обмена с изменением текста кнопки