Update app.py

app.py

@@ -1,65 +1,8 @@
-# import requests
-# import xml.etree.ElementTree as ET
-# from openpyxl import Workbook
-#
-# def extract_links_from_xml(xml_content):
-#     root = ET.fromstring(xml_content)
-#     urls = []
-#
-#     # Find all <loc> elements and extract the text inside them
-#     for loc in root.iter('{http://www.sitemaps.org/schemas/sitemap/0.9}loc'):
-#         urls.append(loc.text.strip())
-#
-#     return urls
-#
-#
-# def check_sitemap(sitemap_url):
-#     # Send a GET request to the sitemap URL
-#     response = requests.get(sitemap_url)
-#
-#     if response.status_code == 200:
-#         urls = extract_links_from_xml(response.content)
-#
-#         # Create a new Excel workbook and select the active sheet
-#         workbook = Workbook()
-#         sheet = workbook.active
-#
-#         # Write headers to the first row of the sheet
-#         sheet['A1'] = 'URL'
-#         sheet['B1'] = 'Response'
-#
-#         row = 2  # Starting row to write URLs and responses
-#
-#         for url in urls:
-#             # Send a GET request to the URL in the sitemap
-#             url_response = requests.get(url)
-#
-#             # Write the URL and response status code to the next row of the sheet
-#             sheet.cell(row=row, column=1, value=url)
-#             sheet.cell(row=row, column=2, value=url_response.status_code)
-#
-#             row += 1
-#
-#         # Save the workbook as an Excel file
-#         workbook.save('sitemap_responses.xlsx')
-#         print("Excel file generated successfully.")
-#     else:
-#         print(f"Error retrieving sitemap: {response.status_code}")
-#
-#
-# # Example usage
-# sitemap_url = 'http://www.embedded-innovations.com/sitemap.xml'
-# check_sitemap(sitemap_url)
-
-
-sitemap_url = 'http://www.embedded-innovations.com/sitemap.xml'
-
 import requests
 import xml.etree.ElementTree as ET
 from openpyxl import Workbook
 import gradio as gr
 import base64
-import os
 
 def extract_links_from_xml(xml_content):
     root = ET.fromstring(xml_content)
@@ -108,13 +51,11 @@ def check_sitemap(sitemap_url):
 
 def download_xml(url):
     check_sitemap(url)
-    file_name = "sitemap_responses.xlsx"
-    file_size = os.path.getsize(file_name) / 1024  # Size in KB
-    with open(file_name, "rb") as f:
+    with open("sitemap_responses.xlsx", "rb") as f:
         data = f.read()
         base64_data = base64.b64encode(data).decode("utf-8")
-        return f'<a href="data:application/octet-stream;base64,{base64_data}" download="{file_name}">Download XML</a>', f"File Name: {file_name}", f"File Size: {file_size:.2f} KB"
+        return f'<a href="data:application/octet-stream;base64,{base64_data}" download="sitemap_responses.xlsx">Download XML</a>'
 
 # Gradio interface
-iface = gr.Interface(download_xml, inputs="text", outputs=[gr.outputs.HTML(), "text", "text"], title="Sitemap Checker", examples=[['https://example.com/sitemap.xml']])
-iface.launch()
+iface = gr.Interface(download_xml, inputs="text", outputs=gr.outputs.HTML(), title="Sitemap Checker")
+iface.launch()