# Install necessary libraries
# pip install praw gradio pandas

import praw
import gradio as gr
import os

# Step 1: Set up Reddit API authentication using environment variables
reddit = praw.Reddit(
    client_id=os.getenv("REDDIT_CLIENT_ID"),  # Fetch from environment variables
    client_secret=os.getenv("REDDIT_CLIENT_SECRET"),  # Fetch from environment variables
    user_agent="my-reddit-app/1.0"  # User-agent can stay as is
)


# Step 2: Function to fetch posts from subreddits
def get_posts(subreddits, keywords, time_filter="day"):
    posts = []
    for subreddit in subreddits.split(","):
        subreddit_obj = reddit.subreddit(subreddit.strip())
        for post in subreddit_obj.search(" OR ".join(keywords.split(",")), time_filter=time_filter):
            posts.append({
                "title": post.title,
                "url": post.url,
                "created": post.created_utc,
                "selftext": post.selftext
            })
    return posts


# Step 3: Function to filter posts based on keywords
def filter_posts(posts, keywords):
    filtered_posts = []
    for post in posts:
        if any(keyword.lower() in (post['title'] + post['selftext']).lower() for keyword in keywords.split(",")):
            filtered_posts.append(post)
    return filtered_posts


# Step 4: Function to categorize posts into topics
def categorize_posts(posts):
    categories = {
        "AI": ["AI", "machine learning", "deep learning"],
        "Programming": ["Python", "JavaScript", "coding"],
        "Tech": ["gadgets", "technology", "innovation"]
    }
    categorized_posts = {category: [] for category in categories}

    for post in posts:
        for category, keywords in categories.items():
            if any(keyword.lower() in post['title'].lower() for keyword in keywords):
                categorized_posts[category].append(post)
    return categorized_posts


# Step 5: Main function to process Reddit data
def process_reddit(subreddits, keywords, time_filter):
    posts = get_posts(subreddits, keywords, time_filter)
    filtered = filter_posts(posts, keywords)
    categorized = categorize_posts(filtered)

    result = {}
    for category, posts in categorized.items():
        result[category] = [{"Title": post["title"], "URL": post["url"]} for post in posts]
    return result


# Step 6: Create Gradio interface
interface = gr.Interface(
    fn=process_reddit,
    inputs=[
        gr.Textbox(label="Subreddits (comma-separated)", placeholder="e.g., technology, python"),
        gr.Textbox(label="Keywords (comma-separated)", placeholder="e.g., AI, Python, gadgets"),
        gr.Dropdown(choices=["hour", "day", "week", "month", "year", "all"], value="day", label="Time Filter")
    ],
    outputs=gr.JSON(label="Categorized Posts"),
    title="Reddit Post Filter",
    description="Search specific subreddits for posts containing certain keywords and categorize them into topics."
)

# Launch the Gradio interface
if __name__ == "__main__":
    interface.launch()