Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import asyncio | |
| from search_engine import search, advanced_search | |
| from osint_engine import create_report | |
| def format_results(results): | |
| if not results: | |
| return "No results found." | |
| if isinstance(results, list): | |
| # Format web search results | |
| formatted_results = [] | |
| for result in results: | |
| formatted_result = f""" | |
| ### [{result['title']}]({result['url']}) | |
| {result['summary']} | |
| **Source:** {result['url']} | |
| **Published:** {result.get('published_date', 'N/A')} | |
| """ | |
| formatted_results.append(formatted_result) | |
| return "\n---\n".join(formatted_results) | |
| elif isinstance(results, dict): | |
| # Format OSINT results | |
| if "error" in results: | |
| return f"Error: {results['error']}" | |
| formatted = [] | |
| # Web results | |
| if "web" in results: | |
| formatted.append(format_results(results["web"])) | |
| # Username/Platform results | |
| if "platforms" in results: | |
| platforms = results["platforms"] | |
| if platforms: | |
| formatted.append("\n### 🔍 Platform Results\n") | |
| for platform in platforms: | |
| formatted.append(f""" | |
| - **Platform:** {platform['platform']} | |
| **URL:** [{platform['url']}]({platform['url']}) | |
| **Status:** {'Found ✅' if platform.get('exists', False) else 'Not Found ❌'} | |
| """) | |
| # Image analysis | |
| if "analysis" in results: | |
| analysis = results["analysis"] | |
| if analysis: | |
| formatted.append("\n### 🖼️ Image Analysis\n") | |
| for key, value in analysis.items(): | |
| formatted.append(f"- **{key.title()}:** {value}") | |
| # Similar images | |
| if "similar_images" in results: | |
| similar = results["similar_images"] | |
| if similar: | |
| formatted.append("\n### 🔍 Similar Images\n") | |
| for img in similar: | |
| formatted.append(f"- [{img['source']}]({img['url']})") | |
| # Location info | |
| if "location" in results: | |
| location = results["location"] | |
| if location and not isinstance(location, str): | |
| formatted.append("\n### 📍 Location Information\n") | |
| for key, value in location.items(): | |
| if key != 'raw': | |
| formatted.append(f"- **{key.title()}:** {value}") | |
| # Domain info | |
| if "domain" in results: | |
| domain = results["domain"] | |
| if domain and not isinstance(domain, str): | |
| formatted.append("\n### 🌐 Domain Information\n") | |
| for key, value in domain.items(): | |
| formatted.append(f"- **{key.title()}:** {value}") | |
| # Historical data | |
| if "historical" in results: | |
| historical = results["historical"] | |
| if historical: | |
| formatted.append("\n### 📅 Historical Data\n") | |
| for entry in historical[:5]: # Limit to 5 entries | |
| formatted.append(f""" | |
| - **Date:** {entry.get('timestamp', 'N/A')} | |
| **URL:** [{entry.get('url', 'N/A')}]({entry.get('url', '#')}) | |
| **Type:** {entry.get('mime_type', 'N/A')} | |
| """) | |
| return "\n".join(formatted) if formatted else "No relevant information found." | |
| else: | |
| return str(results) | |
| def safe_search(query, search_type="web", max_results=5, platform=None, | |
| image_url=None, phone=None, location=None, domain=None): | |
| """Safe wrapper for search functions""" | |
| try: | |
| kwargs = { | |
| "max_results": max_results, | |
| "platform": platform, | |
| "phone": phone, | |
| "location": location, | |
| "domain": domain | |
| } | |
| if search_type == "web": | |
| results = search(query, max_results) | |
| else: | |
| # For async searches | |
| if search_type == "image" and image_url: | |
| query = image_url | |
| loop = asyncio.new_event_loop() | |
| asyncio.set_event_loop(loop) | |
| results = loop.run_until_complete(advanced_search(query, search_type, **kwargs)) | |
| loop.close() | |
| return format_results(results) | |
| except Exception as e: | |
| return f"Error: {str(e)}" | |
| # Create Gradio interface | |
| with gr.Blocks(theme=gr.themes.Soft()) as demo: | |
| gr.Markdown("# 🔍 Intelligent Search Engine") | |
| gr.Markdown(""" | |
| An AI-powered search engine with advanced OSINT capabilities. | |
| Features: | |
| - Web search with AI summaries | |
| - Username search across platforms | |
| - Image search and analysis | |
| - Social media profile search | |
| - Personal information gathering | |
| - Historical data search | |
| """) | |
| with gr.Tab("Web Search"): | |
| with gr.Row(): | |
| query_input = gr.Textbox( | |
| label="Search Query", | |
| placeholder="Enter your search query...", | |
| lines=2 | |
| ) | |
| max_results = gr.Slider( | |
| minimum=1, | |
| maximum=10, | |
| value=5, | |
| step=1, | |
| label="Number of Results" | |
| ) | |
| search_button = gr.Button("Search") | |
| results_output = gr.Markdown(label="Search Results") | |
| search_button.click( | |
| fn=lambda q, n: safe_search(q, "web", n), | |
| inputs=[query_input, max_results], | |
| outputs=results_output | |
| ) | |
| with gr.Tab("Username Search"): | |
| username_input = gr.Textbox( | |
| label="Username", | |
| placeholder="Enter username to search..." | |
| ) | |
| username_button = gr.Button("Search Username") | |
| username_output = gr.Markdown(label="Username Search Results") | |
| username_button.click( | |
| fn=lambda u: safe_search(u, "username"), | |
| inputs=username_input, | |
| outputs=username_output | |
| ) | |
| with gr.Tab("Image Search"): | |
| image_url = gr.Textbox( | |
| label="Image URL", | |
| placeholder="Enter image URL to search..." | |
| ) | |
| image_button = gr.Button("Search Image") | |
| image_output = gr.Markdown(label="Image Search Results") | |
| image_button.click( | |
| fn=lambda u: safe_search(u, "image", image_url=u), | |
| inputs=image_url, | |
| outputs=image_output | |
| ) | |
| with gr.Tab("Social Media Search"): | |
| with gr.Row(): | |
| social_username = gr.Textbox( | |
| label="Username", | |
| placeholder="Enter username..." | |
| ) | |
| platform = gr.Dropdown( | |
| choices=["all", "instagram", "twitter", "reddit"], | |
| value="all", | |
| label="Platform" | |
| ) | |
| social_button = gr.Button("Search Social Media") | |
| social_output = gr.Markdown(label="Social Media Results") | |
| social_button.click( | |
| fn=lambda u, p: safe_search(u, "social", platform=p), | |
| inputs=[social_username, platform], | |
| outputs=social_output | |
| ) | |
| with gr.Tab("Personal Info"): | |
| with gr.Row(): | |
| phone = gr.Textbox(label="Phone Number", placeholder="+1234567890") | |
| location = gr.Textbox(label="Location", placeholder="City, Country") | |
| domain = gr.Textbox(label="Domain", placeholder="example.com") | |
| personal_button = gr.Button("Gather Information") | |
| personal_output = gr.Markdown(label="Personal Information Results") | |
| personal_button.click( | |
| fn=lambda p, l, d: safe_search("", "personal", phone=p, location=l, domain=d), | |
| inputs=[phone, location, domain], | |
| outputs=personal_output | |
| ) | |
| with gr.Tab("Historical Data"): | |
| url_input = gr.Textbox( | |
| label="URL", | |
| placeholder="Enter URL to search historical data..." | |
| ) | |
| historical_button = gr.Button("Search Historical Data") | |
| historical_output = gr.Markdown(label="Historical Data Results") | |
| historical_button.click( | |
| fn=lambda u: safe_search(u, "historical"), | |
| inputs=url_input, | |
| outputs=historical_output | |
| ) | |
| gr.Markdown(""" | |
| ### Examples | |
| Try these example searches: | |
| - Web Search: "Latest developments in artificial intelligence" | |
| - Username: "johndoe" | |
| - Image URL: "https://images.app.goo.gl/w5BtxZKvzg6BdkGE8" | |
| - Social Media: "techuser" on Twitter | |
| - Historical Data: "example.com" | |
| """) | |
| # Launch the app | |
| if __name__ == "__main__": | |
| demo.launch() | |