827 lines
31 KiB
Python
827 lines
31 KiB
Python
"""
|
|
Reddit Scraper Dashboard - Streamlit Web UI
|
|
Run with: streamlit run dashboard/app.py
|
|
"""
|
|
import streamlit as st
|
|
import pandas as pd
|
|
from pathlib import Path
|
|
import sys
|
|
from datetime import datetime
|
|
|
|
# Add parent to path
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|
|
|
from analytics.sentiment import (
|
|
analyze_posts_sentiment, extract_keywords,
|
|
calculate_engagement_metrics, find_best_posting_times
|
|
)
|
|
from search.query import search_all_data, advanced_search, get_top_posts
|
|
|
|
# Page config
|
|
st.set_page_config(
|
|
page_title="Reddit Scraper Dashboard",
|
|
page_icon="🤖",
|
|
layout="wide",
|
|
initial_sidebar_state="expanded"
|
|
)
|
|
|
|
# Custom CSS
|
|
st.markdown("""
|
|
<style>
|
|
.main-header {
|
|
font-size: 2.5rem;
|
|
font-weight: 700;
|
|
background: linear-gradient(90deg, #FF4500, #FF6B6B);
|
|
-webkit-background-clip: text;
|
|
-webkit-text-fill-color: transparent;
|
|
margin-bottom: 1rem;
|
|
}
|
|
.metric-card {
|
|
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
|
|
padding: 1rem;
|
|
border-radius: 10px;
|
|
color: white;
|
|
}
|
|
.stTabs [data-baseweb="tab-list"] {
|
|
gap: 24px;
|
|
}
|
|
.stTabs [data-baseweb="tab"] {
|
|
height: 50px;
|
|
padding: 10px 20px;
|
|
background-color: #262730;
|
|
border-radius: 5px;
|
|
}
|
|
</style>
|
|
""", unsafe_allow_html=True)
|
|
|
|
def load_subreddit_data(subreddit_path):
|
|
"""Load all data for a subreddit."""
|
|
data = {}
|
|
|
|
posts_file = subreddit_path / 'posts.csv'
|
|
if posts_file.exists():
|
|
data['posts'] = pd.read_csv(posts_file)
|
|
|
|
comments_file = subreddit_path / 'comments.csv'
|
|
if comments_file.exists():
|
|
data['comments'] = pd.read_csv(comments_file)
|
|
|
|
return data
|
|
|
|
def get_available_subreddits():
|
|
"""Get list of scraped subreddits."""
|
|
data_dir = Path(__file__).parent.parent / 'data'
|
|
subs = []
|
|
|
|
if data_dir.exists():
|
|
for sub_dir in data_dir.iterdir():
|
|
if sub_dir.is_dir() and (sub_dir / 'posts.csv').exists():
|
|
subs.append(sub_dir.name)
|
|
|
|
return sorted(subs)
|
|
|
|
def main():
|
|
# Header
|
|
st.markdown('<h1 class="main-header">🤖 Reddit Scraper Dashboard</h1>', unsafe_allow_html=True)
|
|
|
|
# Sidebar
|
|
st.sidebar.title("📊 Navigation")
|
|
|
|
# Get available subreddits
|
|
subreddits = get_available_subreddits()
|
|
|
|
if not subreddits:
|
|
st.warning("No scraped data found! Run the scraper first:")
|
|
st.code("python main.py <subreddit> --mode full --limit 100")
|
|
return
|
|
|
|
# Subreddit selector
|
|
selected_sub = st.sidebar.selectbox(
|
|
"Select Subreddit",
|
|
subreddits,
|
|
format_func=lambda x: f"📁 {x}"
|
|
)
|
|
|
|
# Load data
|
|
data_dir = Path(__file__).parent.parent / 'data'
|
|
sub_path = data_dir / selected_sub
|
|
data = load_subreddit_data(sub_path)
|
|
|
|
if 'posts' not in data:
|
|
st.error("No posts data found!")
|
|
return
|
|
|
|
posts_df = data['posts']
|
|
comments_df = data.get('comments', pd.DataFrame())
|
|
|
|
# Main content tabs
|
|
tab1, tab2, tab3, tab4, tab5, tab6, tab7 = st.tabs([
|
|
"📊 Overview", "📈 Analytics", "🔍 Search", "💬 Comments", "⚙️ Scraper", "📋 Job History", "🔌 Integrations"
|
|
])
|
|
|
|
with tab1:
|
|
st.header(f"📊 Overview: {selected_sub}")
|
|
|
|
# Metrics row
|
|
col1, col2, col3, col4, col5 = st.columns(5)
|
|
|
|
with col1:
|
|
st.metric("Total Posts", len(posts_df))
|
|
with col2:
|
|
st.metric("Total Comments", len(comments_df))
|
|
with col3:
|
|
total_score = posts_df['score'].sum() if 'score' in posts_df else 0
|
|
st.metric("Total Score", f"{total_score:,}")
|
|
with col4:
|
|
avg_score = posts_df['score'].mean() if 'score' in posts_df else 0
|
|
st.metric("Avg Score", f"{avg_score:.1f}")
|
|
with col5:
|
|
media_count = posts_df['has_media'].sum() if 'has_media' in posts_df else 0
|
|
st.metric("Media Posts", int(media_count))
|
|
|
|
st.divider()
|
|
|
|
# Post type distribution
|
|
col1, col2 = st.columns(2)
|
|
|
|
with col1:
|
|
st.subheader("📝 Post Types")
|
|
if 'post_type' in posts_df:
|
|
type_counts = posts_df['post_type'].value_counts()
|
|
st.bar_chart(type_counts)
|
|
|
|
with col2:
|
|
st.subheader("📅 Posts Over Time")
|
|
if 'created_utc' in posts_df:
|
|
posts_df['date'] = pd.to_datetime(posts_df['created_utc']).dt.date
|
|
daily = posts_df.groupby('date').size()
|
|
st.line_chart(daily)
|
|
|
|
st.divider()
|
|
|
|
# Top posts
|
|
st.subheader("🔥 Top Posts by Score")
|
|
if 'score' in posts_df:
|
|
top_posts = posts_df.nlargest(10, 'score')[['title', 'score', 'num_comments', 'post_type', 'created_utc']]
|
|
st.dataframe(top_posts)
|
|
|
|
with tab2:
|
|
st.header("📈 Analytics")
|
|
|
|
# Sentiment Analysis
|
|
st.subheader("😀 Sentiment Analysis")
|
|
|
|
if st.button("Run Sentiment Analysis"):
|
|
with st.spinner("Analyzing sentiment..."):
|
|
posts_list = posts_df.to_dict('records')
|
|
analyzed_posts, sentiment_counts = analyze_posts_sentiment(posts_list)
|
|
|
|
col1, col2, col3 = st.columns(3)
|
|
col1.metric("Positive", sentiment_counts['positive'], delta=None)
|
|
col2.metric("Neutral", sentiment_counts['neutral'], delta=None)
|
|
col3.metric("Negative", sentiment_counts['negative'], delta=None)
|
|
|
|
# Pie chart
|
|
sentiment_df = pd.DataFrame({
|
|
'Sentiment': ['Positive', 'Neutral', 'Negative'],
|
|
'Count': [sentiment_counts['positive'], sentiment_counts['neutral'], sentiment_counts['negative']]
|
|
})
|
|
st.bar_chart(sentiment_df.set_index('Sentiment'))
|
|
|
|
st.divider()
|
|
|
|
# Keywords
|
|
st.subheader("☁️ Top Keywords")
|
|
texts = posts_df['title'].tolist()
|
|
if 'selftext' in posts_df:
|
|
texts.extend(posts_df['selftext'].dropna().tolist())
|
|
|
|
keywords = extract_keywords(texts, top_n=30)
|
|
|
|
if keywords:
|
|
kw_df = pd.DataFrame(keywords, columns=['Word', 'Count'])
|
|
st.bar_chart(kw_df.set_index('Word').head(20))
|
|
|
|
st.divider()
|
|
|
|
# Best posting times
|
|
st.subheader("⏰ Best Posting Times")
|
|
|
|
if 'created_utc' in posts_df:
|
|
timing_data = find_best_posting_times(posts_df.to_dict('records'))
|
|
|
|
if timing_data['best_hours']:
|
|
st.write("**Best Hours to Post:**")
|
|
for hour, avg_score in timing_data['best_hours']:
|
|
st.write(f"• {hour}:00 - Avg Score: {avg_score:.1f}")
|
|
|
|
if timing_data['best_days']:
|
|
st.write("**Best Days to Post:**")
|
|
for day, avg_score in timing_data['best_days']:
|
|
st.write(f"• {day} - Avg Score: {avg_score:.1f}")
|
|
|
|
with tab3:
|
|
st.header("🔍 Search Posts")
|
|
|
|
# Search form
|
|
col1, col2 = st.columns([3, 1])
|
|
|
|
with col1:
|
|
search_query = st.text_input("Search query", placeholder="Enter keywords...")
|
|
|
|
with col2:
|
|
min_score = st.number_input("Min Score", min_value=0, value=0)
|
|
|
|
col3, col4, col5 = st.columns(3)
|
|
|
|
with col3:
|
|
if 'post_type' in posts_df:
|
|
post_types = ['All'] + posts_df['post_type'].dropna().unique().tolist()
|
|
selected_type = st.selectbox("Post Type", post_types)
|
|
|
|
with col4:
|
|
if 'author' in posts_df:
|
|
authors = ['All'] + posts_df['author'].dropna().unique().tolist()[:50]
|
|
selected_author = st.selectbox("Author", authors)
|
|
|
|
with col5:
|
|
sort_by = st.selectbox("Sort by", ['score', 'num_comments', 'created_utc'])
|
|
|
|
# Search button
|
|
if st.button("🔍 Search"):
|
|
filtered = posts_df.copy()
|
|
|
|
if search_query:
|
|
mask = filtered['title'].str.contains(search_query, case=False, na=False)
|
|
if 'selftext' in filtered:
|
|
mask |= filtered['selftext'].str.contains(search_query, case=False, na=False)
|
|
filtered = filtered[mask]
|
|
|
|
if min_score > 0:
|
|
filtered = filtered[filtered['score'] >= min_score]
|
|
|
|
if selected_type != 'All' and 'post_type' in filtered:
|
|
filtered = filtered[filtered['post_type'] == selected_type]
|
|
|
|
if selected_author != 'All' and 'author' in filtered:
|
|
filtered = filtered[filtered['author'] == selected_author]
|
|
|
|
filtered = filtered.sort_values(sort_by, ascending=False)
|
|
|
|
st.write(f"Found {len(filtered)} results")
|
|
st.dataframe(filtered[['title', 'score', 'num_comments', 'post_type', 'author', 'created_utc']].head(50))
|
|
|
|
with tab4:
|
|
st.header("💬 Comments Analysis")
|
|
|
|
if len(comments_df) == 0:
|
|
st.warning("No comments data found for this subreddit")
|
|
else:
|
|
col1, col2, col3 = st.columns(3)
|
|
|
|
with col1:
|
|
st.metric("Total Comments", len(comments_df))
|
|
with col2:
|
|
avg_score = comments_df['score'].mean() if 'score' in comments_df else 0
|
|
st.metric("Avg Score", f"{avg_score:.1f}")
|
|
with col3:
|
|
unique_authors = comments_df['author'].nunique() if 'author' in comments_df else 0
|
|
st.metric("Unique Commenters", unique_authors)
|
|
|
|
st.divider()
|
|
|
|
# Top comments
|
|
st.subheader("🔥 Top Comments by Score")
|
|
if 'score' in comments_df:
|
|
top_comments = comments_df.nlargest(10, 'score')[['body', 'score', 'author', 'created_utc']]
|
|
for _, row in top_comments.iterrows():
|
|
with st.expander(f"⬆️ {row['score']} - by u/{row['author']}"):
|
|
st.write(row['body'][:500])
|
|
|
|
st.divider()
|
|
|
|
# Top commenters
|
|
st.subheader("👥 Top Commenters")
|
|
if 'author' in comments_df:
|
|
top_authors = comments_df['author'].value_counts().head(10)
|
|
st.bar_chart(top_authors)
|
|
|
|
with tab5:
|
|
st.header("⚙️ Scraper Controls")
|
|
|
|
st.subheader("🚀 Start New Scrape")
|
|
|
|
col1, col2 = st.columns(2)
|
|
|
|
with col1:
|
|
new_sub = st.text_input("Subreddit/User name", placeholder="e.g. python")
|
|
is_user = st.checkbox("Is a User (not subreddit)")
|
|
|
|
with col2:
|
|
limit = st.number_input("Post Limit", min_value=10, max_value=5000, value=100)
|
|
mode = st.selectbox("Mode", ['full', 'history'])
|
|
|
|
no_media = st.checkbox("Skip media download")
|
|
no_comments = st.checkbox("Skip comments")
|
|
|
|
if st.button("🚀 Start Scraping"):
|
|
if not new_sub:
|
|
st.error("Please enter a subreddit/user name!")
|
|
else:
|
|
# Build command
|
|
cmd = ["python", "main.py", new_sub, "--mode", mode, "--limit", str(limit)]
|
|
if is_user:
|
|
cmd.append("--user")
|
|
if no_media:
|
|
cmd.append("--no-media")
|
|
if no_comments:
|
|
cmd.append("--no-comments")
|
|
|
|
st.info(f"Running: {' '.join(cmd)}")
|
|
|
|
# Run the scraper
|
|
import subprocess
|
|
import time
|
|
|
|
# Create status displays
|
|
status_container = st.empty()
|
|
|
|
col_m1, col_m2, col_m3, col_m4 = st.columns(4)
|
|
posts_metric = col_m1.empty()
|
|
comments_metric = col_m2.empty()
|
|
images_metric = col_m3.empty()
|
|
videos_metric = col_m4.empty()
|
|
|
|
progress_bar = st.progress(0)
|
|
output_container = st.empty()
|
|
|
|
# Initialize metrics
|
|
posts_metric.metric("📊 Posts", "0")
|
|
comments_metric.metric("💬 Comments", "0")
|
|
images_metric.metric("🖼️ Images", "0")
|
|
videos_metric.metric("🎬 Videos", "0")
|
|
|
|
try:
|
|
import os
|
|
env = os.environ.copy()
|
|
env['PYTHONIOENCODING'] = 'utf-8'
|
|
env['PYTHONUNBUFFERED'] = '1' # Force unbuffered output
|
|
|
|
# Use -u flag for unbuffered Python output
|
|
cmd_with_unbuffered = ["python", "-u"] + cmd[1:] # Replace python with python -u
|
|
|
|
process = subprocess.Popen(
|
|
cmd_with_unbuffered,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
bufsize=0, # Unbuffered
|
|
cwd=str(Path(__file__).parent.parent),
|
|
env=env
|
|
)
|
|
|
|
output_lines = []
|
|
start_time = time.time()
|
|
posts_count = 0
|
|
comments_count = 0
|
|
images_count = 0
|
|
videos_count = 0
|
|
|
|
for raw_line in iter(process.stdout.readline, b''):
|
|
if not raw_line:
|
|
break
|
|
# Decode binary to text
|
|
line = raw_line.decode('utf-8', errors='replace')
|
|
output_lines.append(line)
|
|
|
|
# Parse metrics from output
|
|
if "Progress:" in line:
|
|
try:
|
|
parts = line.split("/")
|
|
posts_count = int(parts[0].split()[-1])
|
|
total = int(parts[1].split()[0])
|
|
progress_bar.progress(min(posts_count / total, 1.0))
|
|
posts_metric.metric("📊 Posts", f"{posts_count}/{total}")
|
|
except:
|
|
pass
|
|
|
|
if "Comments:" in line:
|
|
try:
|
|
comments_count = int(line.split("Comments:")[-1].strip().split()[0])
|
|
comments_metric.metric("💬 Comments", str(comments_count))
|
|
except:
|
|
pass
|
|
|
|
if "Images:" in line:
|
|
try:
|
|
images_count = int(line.split("Images:")[-1].strip().split()[0])
|
|
images_metric.metric("🖼️ Images", str(images_count))
|
|
except:
|
|
pass
|
|
|
|
if "Videos:" in line:
|
|
try:
|
|
videos_count = int(line.split("Videos:")[-1].strip().split()[0])
|
|
videos_metric.metric("🎬 Videos", str(videos_count))
|
|
except:
|
|
pass
|
|
|
|
# Update status
|
|
elapsed = time.time() - start_time
|
|
rate = posts_count / elapsed if elapsed > 0 else 0
|
|
status_container.info(f"⏱️ Running for {elapsed:.0f}s | Rate: {rate:.1f} posts/sec")
|
|
|
|
# Keep last 15 lines for display
|
|
display_text = ''.join(output_lines[-15:])
|
|
output_container.code(display_text, language="text")
|
|
|
|
process.wait()
|
|
elapsed = time.time() - start_time
|
|
progress_bar.progress(1.0)
|
|
|
|
if process.returncode == 0:
|
|
status_container.success(f"✅ Completed in {elapsed:.0f}s!")
|
|
st.balloons()
|
|
else:
|
|
status_container.error(f"❌ Failed with code {process.returncode}")
|
|
# Show full error output
|
|
st.code(''.join(output_lines[-30:]), language="text")
|
|
|
|
except Exception as e:
|
|
st.error(f"Error running scraper: {e}")
|
|
import traceback
|
|
st.code(traceback.format_exc())
|
|
|
|
st.divider()
|
|
|
|
# Export options
|
|
st.subheader("📤 Export Data")
|
|
|
|
export_format = st.selectbox("Format", ['CSV', 'JSON', 'Excel'])
|
|
|
|
if st.button("📥 Download Posts"):
|
|
if export_format == 'CSV':
|
|
csv = posts_df.to_csv(index=False)
|
|
st.download_button(
|
|
"Download CSV",
|
|
csv,
|
|
f"{selected_sub}_posts.csv",
|
|
"text/csv"
|
|
)
|
|
elif export_format == 'JSON':
|
|
json_data = posts_df.to_json(orient='records', indent=2)
|
|
st.download_button(
|
|
"Download JSON",
|
|
json_data,
|
|
f"{selected_sub}_posts.json",
|
|
"application/json"
|
|
)
|
|
|
|
st.divider()
|
|
|
|
# Media Export
|
|
st.subheader("🖼️ Media Export")
|
|
|
|
media_dir = Path(f"data/{selected_sub}/media")
|
|
if media_dir.exists():
|
|
images_dir = media_dir / "images"
|
|
videos_dir = media_dir / "videos"
|
|
|
|
images = list(images_dir.glob("*")) if images_dir.exists() else []
|
|
videos = list(videos_dir.glob("*")) if videos_dir.exists() else []
|
|
|
|
col1, col2, col3 = st.columns(3)
|
|
with col1:
|
|
st.metric("📷 Images", len(images))
|
|
with col2:
|
|
st.metric("🎬 Videos", len(videos))
|
|
with col3:
|
|
total_size = sum(f.stat().st_size for f in images + videos) / (1024 * 1024)
|
|
st.metric("💾 Total Size", f"{total_size:.1f} MB")
|
|
|
|
if images or videos:
|
|
if st.button("📦 Download All Media (ZIP)"):
|
|
import zipfile
|
|
import io
|
|
|
|
zip_buffer = io.BytesIO()
|
|
with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zf:
|
|
for img in images:
|
|
zf.write(img, f"images/{img.name}")
|
|
for vid in videos:
|
|
zf.write(vid, f"videos/{vid.name}")
|
|
|
|
st.download_button(
|
|
"💾 Download ZIP",
|
|
zip_buffer.getvalue(),
|
|
f"{selected_sub}_media.zip",
|
|
"application/zip"
|
|
)
|
|
st.success(f"✅ ZIP ready: {len(images)} images, {len(videos)} videos")
|
|
|
|
# Preview recent images
|
|
if images:
|
|
st.write("**Recent Images:**")
|
|
preview_cols = st.columns(min(5, len(images)))
|
|
for i, img in enumerate(images[:5]):
|
|
with preview_cols[i]:
|
|
try:
|
|
st.image(str(img), width=100)
|
|
except:
|
|
st.text(img.name[:15])
|
|
else:
|
|
st.info(f"No media found for {selected_sub}. Run with `--mode full` to download media.")
|
|
|
|
with tab6:
|
|
st.header("📋 Job History")
|
|
|
|
try:
|
|
from export.database import get_job_history, get_job_stats
|
|
|
|
# Job stats
|
|
stats = get_job_stats()
|
|
|
|
col1, col2, col3, col4 = st.columns(4)
|
|
with col1:
|
|
st.metric("Total Jobs", stats.get('total_jobs', 0))
|
|
with col2:
|
|
st.metric("Completed", stats.get('completed', 0))
|
|
with col3:
|
|
st.metric("Failed", stats.get('failed', 0))
|
|
with col4:
|
|
avg_dur = stats.get('avg_duration')
|
|
st.metric("Avg Duration", f"{avg_dur:.1f}s" if avg_dur else "-")
|
|
|
|
st.divider()
|
|
|
|
# Job history table
|
|
st.subheader("Recent Jobs")
|
|
|
|
col1, col2 = st.columns(2)
|
|
with col1:
|
|
filter_status = st.selectbox("Filter by Status", ['All', 'completed', 'failed', 'running'])
|
|
with col2:
|
|
limit = st.number_input("Show last N jobs", min_value=10, max_value=100, value=20)
|
|
|
|
status_filter = None if filter_status == 'All' else filter_status
|
|
jobs = get_job_history(limit=limit, status=status_filter)
|
|
|
|
if jobs:
|
|
jobs_df = pd.DataFrame(jobs)
|
|
# Format for display
|
|
display_cols = ['job_id', 'target', 'mode', 'status', 'posts_scraped',
|
|
'comments_scraped', 'duration_seconds', 'started_at', 'dry_run']
|
|
display_cols = [c for c in display_cols if c in jobs_df.columns]
|
|
st.dataframe(jobs_df[display_cols])
|
|
|
|
# Success rate chart
|
|
st.subheader("Success Rate")
|
|
if 'status' in jobs_df.columns:
|
|
status_counts = jobs_df['status'].value_counts()
|
|
st.bar_chart(status_counts)
|
|
else:
|
|
st.info("No job history found. Run some scrapes first!")
|
|
|
|
except Exception as e:
|
|
st.error(f"Failed to load job history: {e}")
|
|
st.info("Make sure the database is initialized.")
|
|
|
|
with tab7:
|
|
st.header("🔌 Integrations & Settings")
|
|
|
|
# REST API Section
|
|
st.subheader("🚀 REST API")
|
|
|
|
col1, col2, col3 = st.columns(3)
|
|
with col1:
|
|
api_port = st.number_input("API Port", value=8000, min_value=1000, max_value=65535)
|
|
|
|
with col2:
|
|
if st.button("🚀 Start API Server"):
|
|
st.info("Starting API server in background...")
|
|
import subprocess
|
|
try:
|
|
# Start API in background (non-blocking)
|
|
subprocess.Popen(
|
|
["python", "main.py", "--api"],
|
|
cwd=str(Path(__file__).parent.parent),
|
|
creationflags=subprocess.CREATE_NEW_CONSOLE if hasattr(subprocess, 'CREATE_NEW_CONSOLE') else 0
|
|
)
|
|
st.success(f"✅ API server starting on port {api_port}!")
|
|
st.markdown(f"**Open:** [http://localhost:{api_port}/docs](http://localhost:{api_port}/docs)")
|
|
except Exception as e:
|
|
st.error(f"❌ Failed to start API: {e}")
|
|
|
|
with col3:
|
|
# Check if API is running
|
|
import requests
|
|
try:
|
|
resp = requests.get(f"http://localhost:{api_port}/health", timeout=1)
|
|
if resp.status_code == 200:
|
|
st.success("🟢 API is running")
|
|
else:
|
|
st.warning("🟡 API responded but not healthy")
|
|
except:
|
|
st.info("🔴 API not running")
|
|
|
|
st.markdown("""
|
|
**Available Endpoints:**
|
|
| Endpoint | Description |
|
|
|----------|-------------|
|
|
| `/posts` | List posts with filters |
|
|
| `/comments` | List comments |
|
|
| `/subreddits` | All scraped subreddits |
|
|
| `/jobs` | Job history |
|
|
| `/query?sql=...` | Raw SQL queries |
|
|
| `/docs` | Interactive Swagger UI |
|
|
""")
|
|
|
|
st.divider()
|
|
|
|
# External Tools
|
|
st.subheader("📊 External Tools Integration")
|
|
|
|
tool_tabs = st.tabs(["📈 Metabase", "📊 Grafana", "🔗 DreamFactory", "🧦 DuckDB"])
|
|
|
|
with tool_tabs[0]:
|
|
st.markdown("""
|
|
**Metabase Setup:**
|
|
1. Start API: `python main.py --api`
|
|
2. In Metabase: New Question → Native Query
|
|
3. Use HTTP datasource with `http://localhost:8000`
|
|
4. Query: `/posts?subreddit=python&limit=100`
|
|
|
|
**Or use raw SQL:**
|
|
```
|
|
/query?sql=SELECT title, score FROM posts ORDER BY score DESC
|
|
```
|
|
""")
|
|
|
|
with tool_tabs[1]:
|
|
st.markdown("""
|
|
**Grafana Setup:**
|
|
1. Install "JSON API" or "Infinity" plugin
|
|
2. Add datasource: `http://localhost:8000`
|
|
3. Use `/grafana/query` for time-series
|
|
|
|
**Example Panel Query:**
|
|
```sql
|
|
SELECT date(created_utc) as time, COUNT(*) as posts
|
|
FROM posts GROUP BY date(created_utc)
|
|
```
|
|
""")
|
|
|
|
with tool_tabs[2]:
|
|
st.markdown("""
|
|
**DreamFactory Setup:**
|
|
1. Point to SQLite file: `data/reddit_scraper.db`
|
|
2. Or use REST API: `http://localhost:8000`
|
|
3. Auto-generates API for all tables
|
|
""")
|
|
|
|
with tool_tabs[3]:
|
|
st.markdown("""
|
|
**DuckDB (Analytics):**
|
|
1. Export to Parquet first (see below)
|
|
2. Query directly:
|
|
```python
|
|
import duckdb
|
|
duckdb.query("SELECT * FROM 'data/parquet/*.parquet'").df()
|
|
```
|
|
""")
|
|
|
|
st.divider()
|
|
|
|
# Parquet Export
|
|
st.subheader("📦 Parquet Export")
|
|
|
|
col1, col2 = st.columns(2)
|
|
with col1:
|
|
export_sub = st.selectbox("Select subreddit to export", subreddits, key="parquet_export")
|
|
with col2:
|
|
if st.button("📦 Export to Parquet"):
|
|
target_name = export_sub.replace('r_', '').replace('u_', '')
|
|
with st.spinner(f"Exporting {target_name} to Parquet..."):
|
|
import subprocess
|
|
result = subprocess.run(
|
|
["python", "main.py", "--export-parquet", target_name],
|
|
capture_output=True,
|
|
text=True,
|
|
cwd=str(Path(__file__).parent.parent)
|
|
)
|
|
if result.returncode == 0:
|
|
st.success(f"✅ Exported {target_name} to Parquet!")
|
|
st.code(result.stdout[-500:] if len(result.stdout) > 500 else result.stdout)
|
|
else:
|
|
st.error(f"❌ Export failed: {result.stderr}")
|
|
|
|
# List existing parquet files
|
|
parquet_dir = Path("data/parquet")
|
|
if parquet_dir.exists():
|
|
parquet_files = list(parquet_dir.glob("*.parquet"))
|
|
if parquet_files:
|
|
st.write("**Existing Parquet files:**")
|
|
for f in parquet_files[:10]:
|
|
size_mb = f.stat().st_size / (1024 * 1024)
|
|
st.text(f" • {f.name} ({size_mb:.2f} MB)")
|
|
|
|
st.divider()
|
|
|
|
# Database Maintenance
|
|
st.subheader("🛠️ Database Maintenance")
|
|
|
|
col1, col2, col3 = st.columns(3)
|
|
|
|
with col1:
|
|
if st.button("💾 Backup Database"):
|
|
with st.spinner("Creating backup..."):
|
|
import subprocess
|
|
result = subprocess.run(
|
|
["python", "main.py", "--backup"],
|
|
capture_output=True,
|
|
text=True,
|
|
cwd=str(Path(__file__).parent.parent)
|
|
)
|
|
if result.returncode == 0:
|
|
st.success("✅ Database backed up!")
|
|
st.code(result.stdout[-300:] if len(result.stdout) > 300 else result.stdout)
|
|
else:
|
|
st.error(f"❌ Backup failed: {result.stderr}")
|
|
|
|
with col2:
|
|
if st.button("🧹 Vacuum/Optimize"):
|
|
with st.spinner("Optimizing database..."):
|
|
import subprocess
|
|
result = subprocess.run(
|
|
["python", "main.py", "--vacuum"],
|
|
capture_output=True,
|
|
text=True,
|
|
cwd=str(Path(__file__).parent.parent)
|
|
)
|
|
if result.returncode == 0:
|
|
st.success("✅ Database optimized!")
|
|
st.code(result.stdout[-300:] if len(result.stdout) > 300 else result.stdout)
|
|
else:
|
|
st.error(f"❌ Vacuum failed: {result.stderr}")
|
|
|
|
with col3:
|
|
try:
|
|
from export.database import get_database_info
|
|
db_info = get_database_info()
|
|
st.metric("DB Size", f"{db_info.get('size_mb', 0):.2f} MB")
|
|
except:
|
|
st.metric("DB Size", "N/A")
|
|
|
|
# Show backup files
|
|
backup_dir = Path("data/backups")
|
|
if backup_dir.exists():
|
|
backups = sorted(backup_dir.glob("*.db"), reverse=True)[:5]
|
|
if backups:
|
|
st.write("**Recent Backups:**")
|
|
for b in backups:
|
|
size_mb = b.stat().st_size / (1024 * 1024)
|
|
st.text(f" • {b.name} ({size_mb:.2f} MB)")
|
|
|
|
st.divider()
|
|
|
|
# Plugin Configuration
|
|
st.subheader("🔌 Plugins")
|
|
|
|
try:
|
|
from plugins import load_plugins
|
|
plugins = load_plugins()
|
|
|
|
if plugins:
|
|
st.write("**Available Plugins:**")
|
|
for plugin in plugins:
|
|
status = "✅" if plugin.enabled else "❌"
|
|
st.markdown(f"{status} **{plugin.name}** - {plugin.description}")
|
|
|
|
st.info("💡 Enable plugins when scraping: `python main.py <target> --plugins`")
|
|
else:
|
|
st.warning("No plugins found in plugins/ directory")
|
|
except Exception as e:
|
|
st.error(f"Plugin loading error: {e}")
|
|
|
|
st.divider()
|
|
|
|
# Quick Commands Reference
|
|
st.subheader("📋 Quick Commands")
|
|
st.code("""
|
|
# Start REST API
|
|
python main.py --api
|
|
|
|
# Export to Parquet
|
|
python main.py --export-parquet <subreddit>
|
|
|
|
# Backup database
|
|
python main.py --backup
|
|
|
|
# Scrape with plugins
|
|
python main.py <target> --plugins
|
|
|
|
# Dry run (test without saving)
|
|
python main.py <target> --dry-run
|
|
""", language="bash")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|