Create app.py

app.py

@@ -0,0 +1,390 @@
+import streamlit as st
+import pandas as pd
+import json
+import os
+import datetime
+import base64
+from pathlib import Path
+from PIL import Image
+import time
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from webdriver_manager.chrome import ChromeDriverManager
+from selenium.webdriver.chrome.service import Service
+import re
+
+# Set page configuration
+st.set_page_config(
+    page_title="AI Tools Directory",
+    page_icon="🤖",
+    layout="wide"
+)
+
+# Define paths
+DATA_PATH = "data/ai_tools.json"
+IMAGES_PATH = "images"
+
+# Ensure directories exist
+os.makedirs(os.path.dirname(DATA_PATH), exist_ok=True)
+os.makedirs(IMAGES_PATH, exist_ok=True)
+
+# CSS for styling
+st.markdown("""
+<style>
+    .card {
+        border: 1px solid #e0e0e0;
+        border-radius: 8px;
+        padding: 1rem;
+        margin-bottom: 1rem;
+        background-color: white;
+        box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+        transition: transform 0.3s ease;
+    }
+    .card:hover {
+        transform: translateY(-5px);
+        box-shadow: 0 5px 15px rgba(0,0,0,0.1);
+    }
+    .logo-container {
+        display: flex;
+        align-items: center;
+        margin-bottom: 0.5rem;
+    }
+    .logo {
+        width: 30px;
+        height: 30px;
+        margin-right: 10px;
+    }
+    .tool-name {
+        font-size: 1.2rem;
+        font-weight: bold;
+    }
+    .votes {
+        margin-top: 0.5rem;
+        font-size: 0.9rem;
+        color: #666;
+    }
+    .capture-btn {
+        background-color: #4CAF50;
+        color: white;
+        border: none;
+        padding: 8px 12px;
+        text-align: center;
+        text-decoration: none;
+        display: inline-block;
+        border-radius: 4px;
+        margin-top: 0.5rem;
+        cursor: pointer;
+    }
+    .gallery-container {
+        display: flex;
+        flex-wrap: wrap;
+        gap: 10px;
+    }
+    .gallery-item {
+        border: 1px solid #ddd;
+        border-radius: 4px;
+        padding: 5px;
+        width: 150px;
+    }
+    .gallery-img {
+        width: 100%;
+        height: auto;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+
+# Function to initialize or load AI tools data
+def load_ai_tools():
+    if os.path.exists(DATA_PATH):
+        with open(DATA_PATH, 'r') as file:
+            return json.load(file)
+    else:
+        # Initialize with the original HTML data
+        tools = [
+            {"id": 1, "name": "ChatGPT", "url": "https://chat.openai.com/", "votes": 0},
+            {"id": 2, "name": "deepseek", "url": "https://www.deepseek.com/", "votes": 0},
+            {"id": 3, "name": "character.ai", "url": "https://character.ai/", "votes": 0},
+            {"id": 4, "name": "perplexity", "url": "https://www.perplexity.ai/", "votes": 0},
+            {"id": 5, "name": "JanitorAI", "url": "https://www.janitorai.com/", "votes": 0},
+            {"id": 6, "name": "Claude", "url": "https://claude.ai/", "votes": 0},
+            {"id": 7, "name": "QuillBot", "url": "https://quillbot.com/", "votes": 0},
+            {"id": 8, "name": "SUNO", "url": "https://www.suno.ai/", "votes": 0},
+            {"id": 9, "name": "SPICYCHAT.AI", "url": "https://spicychat.ai/", "votes": 0},
+            {"id": 10, "name": "Doubao", "url": "https://doubao.com/", "votes": 0},
+            {"id": 11, "name": "Moonshot AI", "url": "https://moonshot.ai/", "votes": 0},
+            {"id": 12, "name": "Hailuo AI", "url": "https://hailuoai.com/", "votes": 0},
+            {"id": 13, "name": "Hugging Face", "url": "https://huggingface.co/", "votes": 0},
+            {"id": 14, "name": "Poe", "url": "https://poe.com/", "votes": 0},
+            {"id": 15, "name": "Adot", "url": "https://www.adotdev.com/", "votes": 0},
+            {"id": 16, "name": "Eden AI", "url": "https://eden.ai/", "votes": 0},
+            {"id": 17, "name": "PolyBuzz", "url": "https://polybuzz.com/", "votes": 0},
+            {"id": 18, "name": "SERRAT.AI", "url": "https://serrat.ai/", "votes": 0},
+            {"id": 19, "name": "liner", "url": "https://liner.ai/", "votes": 0},
+            {"id": 20, "name": "KLING AI", "url": "https://kling.ai/", "votes": 0},
+            {"id": 21, "name": "CIVITAI", "url": "https://civitai.com/", "votes": 0},
+            {"id": 22, "name": "IIElevenLabs", "url": "https://11labs.io/", "votes": 0},
+            {"id": 23, "name": "Sora", "url": "https://sora.ai/", "votes": 0},
+            {"id": 24, "name": "Crushon AI", "url": "https://crushon.ai/", "votes": 0},
+            {"id": 25, "name": "BLACKBOX AI", "url": "https://blackbox.ai/", "votes": 0},
+            {"id": 26, "name": "DeepAI", "url": "https://deepai.org/", "votes": 0},
+            {"id": 27, "name": "Gamma", "url": "https://gamma.app/", "votes": 0},
+            {"id": 28, "name": "Leonardo.Ai", "url": "https://leonardo.ai/", "votes": 0},
+            {"id": 29, "name": "cutout.pro", "url": "https://cutout.pro/", "votes": 0},
+            {"id": 30, "name": "BRAINLY", "url": "https://brainly.com/", "votes": 0},
+            {"id": 31, "name": "Photoroom", "url": "https://photoroom.com/", "votes": 0},
+            {"id": 32, "name": "Moescape AI", "url": "https://moescape.ai/", "votes": 0},
+            {"id": 33, "name": "Midjourney", "url": "https://www.midjourney.com/", "votes": 0},
+            {"id": 34, "name": "candy.ai", "url": "https://candy.ai/", "votes": 0},
+            {"id": 35, "name": "zeemo", "url": "https://zeemo.ai/", "votes": 0},
+            {"id": 36, "name": "VEED", "url": "https://veed.io/", "votes": 0},
+            {"id": 37, "name": "invideo AI", "url": "https://invideo.ai/", "votes": 0},
+            {"id": 38, "name": "Pixelcut", "url": "https://pixelcut.ai/", "votes": 0},
+            {"id": 39, "name": "talkie", "url": "https://talkie.ai/", "votes": 0},
+            {"id": 40, "name": "PixAI", "url": "https://pixa.ai/", "votes": 0},
+            {"id": 41, "name": "Monica", "url": "https://monica.im/", "votes": 0},
+            {"id": 42, "name": "CURSOR", "url": "https://cursor.sh/", "votes": 0},
+            {"id": 43, "name": "ideogram", "url": "https://ideogram.ai/", "votes": 0},
+            {"id": 44, "name": "CHUB", "url": "https://chub.ai/", "votes": 0},
+            {"id": 45, "name": "Clipchamp", "url": "https://clipchamp.com/", "votes": 0},
+            {"id": 46, "name": "Meta AI", "url": "https://meta.ai/", "votes": 0},
+            {"id": 47, "name": "StudyX", "url": "https://www.studyx.ai/", "votes": 0},
+            {"id": 48, "name": "bolt", "url": "https://bolt.ai/", "votes": 0},
+            {"id": 49, "name": "PicWish", "url": "https://picwish.com/", "votes": 0},
+            {"id": 50, "name": "Joyland", "url": "https://joyland.ai/", "votes": 0}
+        ]
+        save_ai_tools(tools)
+        return tools
+
+
+# Function to save AI tools data
+def save_ai_tools(tools):
+    with open(DATA_PATH, 'w') as file:
+        json.dump(tools, file, indent=4)
+
+
+# Function to generate HTML from tools data
+def generate_html(tools):
+    html = """<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI Tools Table</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+        }
+        table {
+            width: 100%;
+            border-collapse: collapse;
+        }
+        td {
+            padding: 12px;
+            border: 1px solid #e0e0e0;
+            vertical-align: middle;
+        }
+        .tool {
+            display: flex;
+            align-items: center;
+        }
+        .number {
+            font-weight: bold;
+            margin-right: 12px;
+        }
+        .logo {
+            width: 24px;
+            height: 24px;
+            margin-right: 8px;
+            display: inline-flex;
+            align-items: center;
+            justify-content: center;
+        }
+        .name {
+            font-size: 16px;
+            font-weight: 500;
+        }
+        a {
+            color: #000;
+            text-decoration: none;
+            display: flex;
+            align-items: center;
+        }
+    </style>
+</head>
+<body>
+    <table>
+        <tbody>
+"""
+    
+    # Sort tools by votes in descending order
+    sorted_tools = sorted(tools, key=lambda x: x['votes'], reverse=True)
+    
+    # Create rows of 5 columns each
+    for i in range(0, len(sorted_tools), 5):
+        html += "            <tr>\n"
+        for j in range(5):
+            if i+j < len(sorted_tools):
+                tool = sorted_tools[i+j]
+                html += f"""                <td>
+                    <div class="tool">
+                        <span class="number">{i+j+1}.</span>
+                        <a href="{tool['url']}" target="_blank">
+                            <span class="logo">
+                                <img src="https://via.placeholder.com/24" alt="{tool['name']} logo" />
+                            </span>
+                            <span class="name">{tool['name']}</span>
+                        </a>
+                    </div>
+                </td>
+"""
+        html += "            </tr>\n"
+    
+    html += """        </tbody>
+    </table>
+</body>
+</html>"""
+    
+    return html
+
+
+# Function to take a screenshot of a website
+def capture_website(url, tool_name):
+    try:
+        chrome_options = Options()
+        chrome_options.add_argument("--headless")
+        chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.add_argument("--window-size=1920,1080")
+        
+        service = Service(ChromeDriverManager().install())
+        driver = webdriver.Chrome(service=service, options=chrome_options)
+        
+        # Navigate to the website
+        driver.get(url)
+        
+        # Allow time for the page to load
+        time.sleep(3)
+        
+        # Create a safe filename
+        safe_name = re.sub(r'[^\w\-_.]', '_', tool_name)
+        filename = f"{IMAGES_PATH}/{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}_{safe_name}.png"
+        
+        # Take screenshot
+        driver.save_screenshot(filename)
+        
+        # Close the browser
+        driver.quit()
+        
+        return filename
+    except Exception as e:
+        st.error(f"Error capturing website: {str(e)}")
+        return None
+
+
+# Display image gallery
+def display_gallery():
+    st.subheader("📸 Captured Screenshots Gallery")
+    
+    images = sorted(Path(IMAGES_PATH).glob("*.png"), reverse=True)
+    
+    if not images:
+        st.info("No captured screenshots yet. Use the Capture button on any AI tool to take a screenshot.")
+        return
+    
+    # Display images in a grid
+    cols = st.columns(3)
+    
+    for i, img_path in enumerate(images):
+        col_idx = i % 3
+        with cols[col_idx]:
+            img = Image.open(img_path)
+            st.image(img, caption=img_path.stem, use_column_width=True)
+            
+            # Extract tool name from filename
+            parts = img_path.stem.split('_')
+            if len(parts) > 2:
+                tool_name = '_'.join(parts[2:])
+            else:
+                tool_name = parts[-1]
+                
+            # Create download button for the image
+            with open(img_path, "rb") as file:
+                btn = st.download_button(
+                    label="Download",
+                    data=file,
+                    file_name=img_path.name,
+                    mime="image/png"
+                )
+
+
+# Main app function
+def main():
+    st.title("🤖 AI Tools Directory")
+    
+    # Load AI tools data
+    ai_tools = load_ai_tools()
+    
+    # Main navigation
+    tabs = st.tabs(["AI Tools", "Screenshot Gallery"])
+    
+    with tabs[0]:
+        # Create two-column layout
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            st.subheader("AI Tools (Ranked by Votes)")
+            
+            # Display AI tools list with voting and capture
+            for tool in sorted(ai_tools, key=lambda x: x['votes'], reverse=True):
+                with st.container():
+                    st.markdown(f"""
+                    <div class="card">
+                        <div class="logo-container">
+                            <img src="https://via.placeholder.com/30" class="logo" alt="{tool['name']} logo">
+                            <span class="tool-name">{tool['name']}</span>
+                        </div>
+                        <div class="votes">👍 {tool['votes']} votes</div>
+                    </div>
+                    """, unsafe_allow_html=True)
+                    
+                    # Add voting button and capture button in columns
+                    vote_col, capture_col = st.columns([1, 1])
+                    
+                    with vote_col:
+                        if st.button(f"👍 Upvote", key=f"vote_{tool['id']}"):
+                            # Increment votes
+                            for t in ai_tools:
+                                if t['id'] == tool['id']:
+                                    t['votes'] += 1
+                                    break
+                            
+                            # Save updated data
+                            save_ai_tools(ai_tools)
+                            st.experimental_rerun()
+                    
+                    with capture_col:
+                        if st.button(f"📸 Capture", key=f"capture_{tool['id']}"):
+                            with st.spinner(f"Capturing {tool['name']}..."):
+                                capture_website(tool['url'], tool['name'])
+                                st.success("Captured!")
+                    
+                    # Add visit button
+                    if st.button(f"🔗 Visit {tool['name']}", key=f"visit_{tool['id']}"):
+                        # Create iframe to visit the site
+                        st.markdown(f"""
+                        <iframe src="{tool['url']}" width="100%" height="600px" frameborder="0"></iframe>
+                        """, unsafe_allow_html=True)
+        
+        with col2:
+            # Generate and display HTML
+            html_content = generate_html(ai_tools)
+            st.subheader("HTML Rendered View")
+            st.components.v1.html(html_content, height=800, scrolling=True)
+    
+    with tabs[1]:
+        display_gallery()
+
+
+if __name__ == "__main__":
+    main()