Update app.py

app.py

@@ -312,16 +312,60 @@ def process_vk_link(link: str) -> tuple[str, int]:
 
     result = f"1. [{title}]({link}) — {human_format_views(views)}"  # Изменено на нумерованный список
     return result, views
-
-# ... (остальной код остаётся без изменений до формирования Markdown)
-
-# В функции main, где формируется итоговый Markdown:
+# Функция для определения платформы
+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
 def main():
-    # ... (остальной код остаётся без изменений до формирования Markdown)
+    st.title("PR Tool made by @Dina")
+
+    # Если токены не получены, показываем авторизацию
+    if not st.session_state.get("VK_ACCESS_TOKEN"):
+        vk_auth()
+        return
+
+    # Если токены есть, показываем парсер
+    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
+    """)
+
+    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)
 
-    md_lines = []
+        md_lines = []
 
-    # --- Telegram ---
+        # --- Telegram ---
     md_lines.append("## Telegram")
     md_lines.append(f"**Суммарный охват постов — {human_format_views(tg_total_views)} просмотров**")
     if tg_results:
@@ -329,9 +373,8 @@ def main():
             md_lines.append(f"{i}. {r_line[3:]}")  # Убираем "1. " из строки и добавляем новый номер
     else:
         md_lines.append("_Нет ссылок на Telegram_")
-    md_lines.append("")
-
-    # --- ВКонтакте ---
+    md_lines.append("") 
+        # --- ВКонтакте ---
     md_lines.append("## ВКонтакте")
     md_lines.append(f"**Суммарный охват постов — {human_format_views(vk_total_views)} просмотров**")
     if vk_results:
@@ -370,8 +413,6 @@ def main():
     """
     components.html(copy_button_html, height=80)
 
-# ... (остальной код остаётся без изменений)
-
 # Запуск приложения
 if __name__ == "__main__":
     main()