Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import pandas as pd | |
| from main import SoundgasmScraper | |
| import time | |
| # Initialize the scraper | |
| scraper = SoundgasmScraper() | |
| def search_soundgasm(query, max_results=10): | |
| """ | |
| Search for audio on Soundgasm and return results in a formatted way | |
| """ | |
| if not query.strip(): | |
| return "Please enter a search query.", None | |
| try: | |
| # Show loading message | |
| results = scraper.search_audio(query, max_results=int(max_results)) | |
| if not results: | |
| return "No results found. Try a different search term.", None | |
| # Format results for display | |
| formatted_results = [] | |
| for i, result in enumerate(results, 1): | |
| formatted_result = f""" | |
| ### Result {i}: {result['audio_title'] or 'Untitled'} | |
| **Username:** {result['username'] or 'Unknown'} | |
| **Page URL:** [{result['url']}]({result['url']}) | |
| **Direct Audio URL:** {result['audio_url'] or 'Not found'} | |
| **Description:** {result['description'][:200] + '...' if result['description'] and len(result['description']) > 200 else result['description'] or 'No description available'} | |
| --- | |
| """ | |
| formatted_results.append(formatted_result) | |
| # Create a DataFrame for the table view | |
| df_data = [] | |
| for result in results: | |
| df_data.append({ | |
| 'Title': result['audio_title'] or 'Untitled', | |
| 'Username': result['username'] or 'Unknown', | |
| 'Page URL': result['url'], | |
| 'Audio URL': result['audio_url'] or 'Not found', | |
| 'Description': (result['description'][:100] + '...' if result['description'] and len(result['description']) > 100 else result['description']) or 'No description' | |
| }) | |
| df = pd.DataFrame(df_data) | |
| return "\n".join(formatted_results), df | |
| except Exception as e: | |
| return f"Error occurred during search: {str(e)}", None | |
| def search_by_user(username): | |
| """ | |
| Search for all audios by a specific username | |
| """ | |
| if not username.strip(): | |
| return "Please enter a username.", None | |
| try: | |
| results = scraper.search_by_username(username.strip()) | |
| if not results: | |
| return f"No audios found for user '{username}' or user doesn't exist.", None | |
| # Format results for display | |
| formatted_results = [] | |
| for i, result in enumerate(results, 1): | |
| formatted_result = f""" | |
| ### Audio {i}: {result['audio_title'] or 'Untitled'} | |
| **Page URL:** [{result['url']}]({result['url']}) | |
| **Direct Audio URL:** {result['audio_url'] or 'Not found'} | |
| **Description:** {result['description'][:200] + '...' if result['description'] and len(result['description']) > 200 else result['description'] or 'No description available'} | |
| --- | |
| """ | |
| formatted_results.append(formatted_result) | |
| # Create a DataFrame for the table view | |
| df_data = [] | |
| for result in results: | |
| df_data.append({ | |
| 'Title': result['audio_title'] or 'Untitled', | |
| 'Page URL': result['url'], | |
| 'Audio URL': result['audio_url'] or 'Not found', | |
| 'Description': (result['description'][:100] + '...' if result['description'] and len(result['description']) > 100 else result['description']) or 'No description' | |
| }) | |
| df = pd.DataFrame(df_data) | |
| return "\n".join(formatted_results), df | |
| except Exception as e: | |
| return f"Error occurred during user search: {str(e)}", None | |
| # Create the Gradio interface | |
| with gr.Blocks(title="Soundgasm Audio Search", theme=gr.themes.Base()) as app: | |
| gr.Markdown(""" | |
| # 🎧 Soundgasm Audio Search | |
| Search for audio content on Soundgasm.net and get direct links to the audio files. | |
| **Features:** | |
| - Search by keywords | |
| - Search by username | |
| - Get direct audio file URLs | |
| - View audio descriptions and metadata | |
| """) | |
| with gr.Tabs(): | |
| # Search by keywords tab | |
| with gr.TabItem("🔍 Search by Keywords"): | |
| with gr.Row(): | |
| with gr.Column(scale=3): | |
| search_input = gr.Textbox( | |
| label="Search Query", | |
| placeholder="Enter keywords (e.g., ASMR, relaxing, etc.)", | |
| lines=1 | |
| ) | |
| with gr.Column(scale=1): | |
| max_results_input = gr.Slider( | |
| minimum=1, | |
| maximum=20, | |
| value=10, | |
| step=1, | |
| label="Max Results" | |
| ) | |
| search_button = gr.Button("🔍 Search", variant="primary") | |
| with gr.Row(): | |
| with gr.Column(): | |
| search_output = gr.Markdown(label="Search Results") | |
| with gr.Column(): | |
| search_table = gr.Dataframe( | |
| label="Results Table", | |
| interactive=False, | |
| wrap=True | |
| ) | |
| # Search by username tab | |
| with gr.TabItem("👤 Search by Username"): | |
| username_input = gr.Textbox( | |
| label="Username", | |
| placeholder="Enter Soundgasm username", | |
| lines=1 | |
| ) | |
| user_search_button = gr.Button("🔍 Search User", variant="primary") | |
| with gr.Row(): | |
| with gr.Column(): | |
| user_output = gr.Markdown(label="User's Audios") | |
| with gr.Column(): | |
| user_table = gr.Dataframe( | |
| label="User's Audios Table", | |
| interactive=False, | |
| wrap=True | |
| ) | |
| # Instructions | |
| with gr.Accordion("📖 How to Use", open=False): | |
| gr.Markdown(""" | |
| ### Search by Keywords: | |
| 1. Enter keywords related to the audio you're looking for | |
| 2. Adjust the maximum number of results if needed | |
| 3. Click "Search" to find matching audios | |
| ### Search by Username: | |
| 1. Enter a Soundgasm username | |
| 2. Click "Search User" to see all audios from that user | |
| ### Results: | |
| - **Page URL**: Link to the Soundgasm page | |
| - **Audio URL**: Direct link to the audio file (m4a format) | |
| - **Description**: Audio description if available | |
| ### Note: | |
| - This tool searches using external search engines, so results may vary | |
| - Some audio URLs might not be extractable due to page structure changes | |
| - Always respect the content creators' rights and terms of service | |
| """) | |
| # Event handlers | |
| search_button.click( | |
| fn=search_soundgasm, | |
| inputs=[search_input, max_results_input], | |
| outputs=[search_output, search_table] | |
| ) | |
| user_search_button.click( | |
| fn=search_by_user, | |
| inputs=[username_input], | |
| outputs=[user_output, user_table] | |
| ) | |
| # Allow Enter key to trigger search | |
| search_input.submit( | |
| fn=search_soundgasm, | |
| inputs=[search_input, max_results_input], | |
| outputs=[search_output, search_table] | |
| ) | |
| username_input.submit( | |
| fn=search_by_user, | |
| inputs=[username_input], | |
| outputs=[user_output, user_table] | |
| ) | |
| # Launch the app | |
| if __name__ == "__main__": | |
| app.launch( | |
| server_name="0.0.0.0", | |
| server_port=7860, | |
| share=False, | |
| debug=True | |
| ) | |