Update app.py

app.py

@@ -5,58 +5,79 @@ import os
 
 API_URL = "https://de1.api.radio-browser.info/json/stations"
 
-# Function to fetch stations and save as .m3u
+# Fetch full list of stations from the API
 
-def fetch_and_save_m3u():
+def fetch_stations():
     try:
-        # Fetch full database of stations
         response = requests.get(API_URL)
         response.raise_for_status()
         stations = response.json()
         count = len(stations)
-        print(f"Начинаем экспорт {count} радиостанций...")
-
-        # Create temporary file
-        tmp_dir = tempfile.mkdtemp()
-        file_path = os.path.join(tmp_dir, "radio_stations.m3u")
-
-        # Write M3U file
-        with open(file_path, "w", encoding="utf-8") as f:
-            f.write("#EXTM3U\n")
-            for station in stations:
-                name = station.get("name", "Unknown")
-                url = station.get("url_resolved", station.get("url",""))
-                f.write(f"#EXTINF:-1,{name}\n")
-                f.write(f"{url}\n")
+        print(f"Starting fetch of {count} stations...")
+        return stations, None
+    except Exception as e:
+        print(f"Error fetching stations: {e}")
+        return None, str(e)
 
-        print(f"Экспорт завершен успешно. Всего записано {count} станций.")
-        return file_path, "✔️ Экспорт завершен.", count
+# Write stations to M3U file
 
-    except Exception as e:
-        print(f"Ошибка при экспорте: {e}")
-        return None, f"❌ Ошибка: {str(e)}", 0
+def save_as_m3u(stations):
+    tmp_dir = tempfile.mkdtemp()
+    file_path = os.path.join(tmp_dir, "radio_stations.m3u")
+    with open(file_path, "w", encoding="utf-8") as f:
+        f.write("#EXTM3U\n")
+        for station in stations:
+            name = station.get("name", "Unknown")
+            url = station.get("url_resolved", station.get("url", ""))
+            f.write(f"#EXTINF:-1,{name}\n")
+            f.write(f"{url}\n")
+    print(f"M3U file created at {file_path} with {len(stations)} entries.")
+    return file_path
 
 # Gradio interface
+
 def create_app():
     with gr.Blocks() as demo:
-        gr.Markdown("## Экспорт базы радиостанций в M3U\nНажмите кнопку, чтобы скачать полный список радиостанций из radio-browser.info в формате .m3u.")
-        download_button = gr.Button("Экспортировать в M3U")
-        output_file = gr.File(label="Скачать файл .m3u")
-        status = gr.Textbox(label="Статус")
-        count_box = gr.Textbox(label="Количество станций экспортировано")
-
-        def on_click():
-            file_path, msg, count = fetch_and_save_m3u()
-            if file_path:
-                return file_path, msg, str(count)
+        gr.Markdown(
+            "## Radio Station Database Export\n"
+            "Fetch the full list of radio stations from radio-browser.info and export in your chosen format."
+        )
+        format_selector = gr.Radio(
+            choices=["M3U", "Table"],
+            value="M3U",
+            label="Select output format"
+        )
+        export_button = gr.Button("Export")
+        download_file = gr.File(label="Download .m3u File")
+        station_table = gr.Dataframe(
+            headers=["Station Name", "Stream URL"],
+            label="Station Table"
+        )
+        status = gr.Textbox(label="Status")
+        count_box = gr.Textbox(label="Stations Exported")
+
+        def on_export(selected_format):
+            stations, error = fetch_stations()
+            if error:
+                return None, None, f"❌ Error: {error}", "0"
+            count = len(stations)
+            if selected_format == "M3U":
+                file_path = save_as_m3u(stations)
+                return file_path, None, "✔️ Export completed.", str(count)
             else:
-                return None, msg, "0"
+                table_data = [[s.get("name", ""), s.get("url_resolved", s.get("url", ""))] for s in stations]
+                print(f"Table generated with {count} stations.")
+                return None, table_data, "✔️ Table generated.", str(count)
 
-        download_button.click(fn=on_click, outputs=[output_file, status, count_box])
+        export_button.click(
+            fn=on_export,
+            inputs=format_selector,
+            outputs=[download_file, station_table, status, count_box]
+        )
 
     return demo
 
-# Запуск приложения
 if __name__ == "__main__":
     app = create_app()
     app.launch(share=False)
+