app.py

import gradio as gr
import requests
from bs4 import BeautifulSoup
from nltk import download, sent_tokenize
import google.generativeai as genai
import os
import re
import tempfile
import asyncio
import time

# Download NLTK tokenizer data
download('punkt')

# Set Gemini API key
api_key = os.environ.get("GEMINI_API_KEY")
if not api_key:
    raise ValueError("GEMINI_API_KEY not found in environment variables.")

genai.configure(api_key=api_key)

try:
    model = genai.GenerativeModel('gemini-1.5-flash')
except Exception as e:
    print(f"Error initializing model: {e}")
    print("Available models:")
    for m in genai.list_models():
        print(m.name)
    raise

PROMPT = """
You are an AI content reviewer. Analyze the provided text for the following:
1. Grammar Issues: Identify and suggest corrections for grammatical errors.
2. Legal Policy Violations: Flag content that may violate common legal policies (e.g., copyright infringement, defamation, incitement to violence).
3. Crude/Abusive Language: Detect crude, offensive, or abusive language.
4. Sensitive Topics: Identify content related to sensitive topics such as racism, gender bias, or other forms of discrimination.

Return the results in the following markdown format:
# Blog Review Report

## Grammar Corrections
1. [Heading of issue]
   - CONTENT: [Exact line or part of text with the issue]
   - SUGGESTION: [Suggested correction]
   - ISSUE: [Description of the issue]

[Continue numbering for additional issues or state "None detected"]

## Legal Policy Violations
- CONTENT: [Exact line or part of text with the issue]
  SUGGESTION: [Suggested action or correction]
  ISSUE: [Description of the legal violation]
[Or state "None detected"]

## Crude/Abusive Language
- [List instances of crude or abusive language or "None detected"]

## Sensitive Topics
- [List instances of sensitive topics or "None detected"]
"""

async def fetch_url_content(url):
    try:
        response = requests.get(url, timeout=10)
        response.raise_for_status()
        soup = BeautifulSoup(response.text, 'html.parser')
        content = ' '.join([p.get_text(strip=True) for p in soup.find_all(['p', 'article', 'div'])])
        return content or "No readable content found on the page."
    except Exception as e:
        return f"Error fetching URL: {str(e)}"

async def review_blog(text_input, url_input, progress=gr.Progress()):
    if text_input and not url_input:
        input_text = text_input
    elif url_input and not text_input:
        input_text = await fetch_url_content(url_input)
        if input_text.startswith("Error"):
            return "Review Blog", input_text, gr.update(visible=False)
    else:
        return "Review Blog", "Error: Provide either text or URL, not both.", gr.update(visible=False)

    sentences = sent_tokenize(input_text)
    analysis_text = "\n".join(sentences)

    progress(0.2, desc="Analyzing...")
    try:
        response = await asyncio.to_thread(model.generate_content, PROMPT + "\n\nText to analyze:\n" + analysis_text)
        report = response.text.strip()
        report = re.sub(r'^markdown\n|$', '', report, flags=re.MULTILINE)
    except Exception as e:
        return "Review Blog", f"Error with Gemini API: {str(e)}", gr.update(visible=False)

    try:
        with tempfile.NamedTemporaryFile(mode='w', suffix='.md', delete=False, encoding='utf-8') as temp_file:
            temp_file.write(report)
            file_path = temp_file.name
        progress(1.0, desc="Report Ready!")
        return "Review Blog", report, gr.update(visible=True, value=file_path)
    except Exception as e:
        return "Review Blog", f"Error saving report: {str(e)}", gr.update(visible=False)

# ------------------ CSS ------------------ #
custom_css = """
@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;600;700&display=swap');

.gradio-container {
    font-family: 'Inter', sans-serif !important;
    background: linear-gradient(120deg, #e0f7fa 0%, #f9f9f9 100%);
    padding: 40px;
    border-radius: 20px;
    box-shadow: 0 10px 40px rgba(0, 0, 0, 0.07);
}

h1, h3 {
    font-weight: 700;
    color: #2b6cb0;
}

input, textarea {
    border: 2px solid #cbd5e0;
    border-radius: 12px;
    padding: 15px;
    background-color: #ffffff;
    font-size: 16px;
    color: #2d3748;
    transition: border 0.3s ease;
}

input:focus, textarea:focus {
    border-color: #3182ce;
    outline: none;
}

.review-btn {
    background: linear-gradient(45deg, #3182ce, #63b3ed);
    padding: 14px 28px;
    border-radius: 12px;
    font-size: 16px;
    color: #fff;
    font-weight: 600;
    border: none;
    cursor: pointer;
    transition: all 0.3s ease;
}

.review-btn:hover {
    transform: scale(1.04);
    background: linear-gradient(45deg, #63b3ed, #3182ce);
}

.markdown-container {
    background: #ffffff;
    padding: 25px;
    border-radius: 16px;
    box-shadow: 0 8px 24px rgba(0, 0, 0, 0.08);
    margin-top: 25px;
    font-size: 15px;
    line-height: 1.6;
}

.download-btn {
    background-color: #38a169;
    color: white;
    padding: 12px 20px;
    border-radius: 10px;
    font-weight: 600;
    transition: background 0.3s ease;
}

.download-btn:hover {
    background-color: #2f855a;
}
"""

# ------------------ UI ------------------ #
with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
    with gr.Row():
        gr.HTML("""
        <div style="text-align: center; margin-bottom: 10px;">
            <h1 style="color:#2b6cb0; font-size: 2.5rem;">✨ AI Blog Reviewer</h1>
            <p style="font-size: 1.1rem; color: #4a5568;">
                Analyze your blog for grammar, legal violations, abusive content, and more.
            </p>
        </div>
        """)

    with gr.Row(equal_height=True):
        with gr.Column(scale=3):
            with gr.Tabs():
                with gr.TabItem("✍️ Text Input"):
                    text_input = gr.Textbox(
                        label="Paste Your Blog Content",
                        placeholder="Write or paste your blog text here...",
                        lines=12,
                        elem_classes=["input-text"]
                    )
                with gr.TabItem("🔗 URL Input"):
                    url_input = gr.Textbox(
                        label="Enter Blog URL",
                        placeholder="https://example.com/blog",
                        lines=1,
                        elem_classes=["input-url"]
                    )
        with gr.Column(scale=2):
            gr.HTML("""
                <div style="background: #fff; padding: 20px 25px; border-radius: 15px; box-shadow: 0 4px 14px rgba(0,0,0,0.08);">
                    <h3 style="color:#2b6cb0;">🚀 How It Works</h3>
                    <ul style="font-size: 15px; color: #2d3748; line-height: 1.7;">
                        <li>Paste blog text or a URL</li>
                        <li>Gemini analyzes for quality and issues</li>
                        <li>Download the review as Markdown</li>
                    </ul>
                </div>
            """)
            status_button = gr.Button("🧠 Review Blog", elem_classes=["review-btn"])

    gr.Markdown("### 📄 Review Report")
    with gr.Row():
        with gr.Column():
            report_output = gr.Markdown(elem_classes=["markdown-container"])
            download_btn = gr.File(label="Download Report", visible=False, elem_classes=["download-btn"])

    status_button.click(
        fn=review_blog,
        inputs=[text_input, url_input],
        outputs=[status_button, report_output, download_btn]
    )

demo.launch()