Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
-from urllib.parse import urlparse
+from urllib.parse import urlparse, urljoin
+from bs4 import BeautifulSoup
 
 def is_valid_url(url):
     """Проверяет, является ли строка валидным URL."""
@@ -10,9 +11,29 @@ def is_valid_url(url):
     except:
         return False
 
+def extract_additional_resources(url):
+    """Извлекает ссылки на CSS, JS и изображения из HTML-кода."""
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+        soup = BeautifulSoup(response.text, "html.parser")
+
+        # Извлекаем ссылки на CSS
+        css_links = [urljoin(url, link["href"]) for link in soup.find_all("link", rel="stylesheet") if "href" in link.attrs]
+
+        # Извлекаем ссылки на JS
+        js_links = [urljoin(url, script["src"]) for script in soup.find_all("script") if "src" in script.attrs]
+
+        # Извлекаем ссылки на изображения
+        img_links = [urljoin(url, img["src"]) for img in soup.find_all("img") if "src" in img.attrs]
+
+        return css_links, js_links, img_links
+    except Exception as e:
+        return [], [], []
+
 def convert_to_text(url):
     if not is_valid_url(url):
-        return "Ошибка: Введите корректный URL.", "", None  # Возвращаем сообщение об ошибке, пустую строку и None для файла
+        return "Ошибка: Введите корректный URL.", "", None, [], [], []  # Возвращаем сообщение об ошибке и пустые данные
 
     try:
         # Устанавливаем заголовки, чтобы имитировать запрос от браузера
@@ -32,9 +53,12 @@ def convert_to_text(url):
         with open(file_path, "w", encoding="utf-8") as file:
             file.write(response.text)
 
-        return results, response.text, file_path  # Возвращаем результаты, содержимое и путь к файлу
+        # Извлекаем дополнительные ресурсы
+        css_links, js_links, img_links = extract_additional_resources(url)
+
+        return results, response.text, file_path, css_links, js_links, img_links
     except requests.exceptions.RequestException as e:
-        return f"Ошибка: {e}", "", None  # Возвращаем сообщение об ошибке, пустую строку и None для файла
+        return f"Ошибка: {e}", "", None, [], [], []  # Возвращаем сообщение об ошибке и пустые данные
 
 # HTML и JavaScript для кнопки "Скопировать код"
 copy_button_html = """
@@ -71,7 +95,22 @@ with gr.Blocks(css=css) as demo:
         file_output = gr.File(label="Скачать файл", visible=False)  # Скрытый компонент для скачивания файла
     
     submit_button = gr.Button("Загрузить")
-    submit_button.click(fn=convert_to_text, inputs=url_input, outputs=[results_output, text_output, file_output])
+    submit_button.click(
+        fn=convert_to_text,
+        inputs=url_input,
+        outputs=[results_output, text_output, file_output, gr.Textbox(label="CSS-файлы"), gr.Textbox(label="JS-файлы"), gr.Textbox(label="Изображения")]
+    )
+
+    # Добавляем Accordion для скрытия/показа дополнительных ресурсов
+    with gr.Accordion("Показать/Скрыть дополнительные ресурсы", open=False):
+        gr.Markdown("### CSS-файлы")
+        css_output = gr.Textbox(label="CSS-файлы", interactive=False)
+        
+        gr.Markdown("### JS-файлы")
+        js_output = gr.Textbox(label="JS-файлы", interactive=False)
+        
+        gr.Markdown("### Изображения")
+        img_output = gr.Textbox(label="Изображения", interactive=False)
 
 # Запускаем интерфейс
 demo.launch()