Update app.py

app.py

@@ -9,28 +9,25 @@ import tempfile
 import asyncio
 import time
 
-# Download NLTK data
+# Download NLTK tokenizer data
 download('punkt')
-download('punkt_tab')
 
-# Configure Gemini API using environment variable
+# 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. Please set it in your environment.")
+    raise ValueError("GEMINI_API_KEY not found in environment variables.")
 
 genai.configure(api_key=api_key)
 
-# Use gemini-1.5-flash for faster and more accessible text analysis
 try:
     model = genai.GenerativeModel('gemini-1.5-flash')
 except Exception as e:
-    print(f"Error initializing model: {str(e)}")
+    print(f"Error initializing model: {e}")
     print("Available models:")
     for m in genai.list_models():
         print(m.name)
-    raise ValueError("Failed to initialize gemini-1.5-flash. Check available models above and update the model name.")
+    raise
 
-# Prompt for Gemini to analyze text with specified output format
 PROMPT = """
 You are an AI content reviewer. Analyze the provided text for the following:
 1. Grammar Issues: Identify and suggest corrections for grammatical errors.
@@ -47,11 +44,6 @@ Return the results in the following markdown format:
    - SUGGESTION: [Suggested correction]
    - ISSUE: [Description of the issue]
 
-2. [Heading of next 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
@@ -65,8 +57,6 @@ Return the results in the following markdown format:
 
 ## Sensitive Topics
 - [List instances of sensitive topics or "None detected"]
-
-For each issue, provide the exact text, a suggested correction or action, and a concise explanation. Be precise and ensure the output strictly follows the specified format.
 """
 
 async def fetch_url_content(url):
@@ -75,202 +65,155 @@ async def fetch_url_content(url):
         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 if content else "No readable content found on the page."
+        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_type = "Text"
         input_text = text_input
     elif url_input and not text_input:
-        input_type = "URL"
-        input_text = url_input
-    else:
-        return "Review Blog", "Error: Please provide input in either the Text or URL tab, but not both.", gr.update(visible=False)
-
-    if not input_text:
-        return "Review Blog", "Error: No input provided.", gr.update(visible=False)
-
-    if input_type == "URL":
-        progress(0, desc="Fetching content...")
-        input_text = await fetch_url_content(input_text)
+        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, desc="Generating report...")
-    start_time = time.time()
-    for i in range(1, 10):
-        await asyncio.sleep(1)
-        progress(i / 10, desc=f"Generating report... ({int(i * 10)}%)")
-        if time.time() - start_time > 30:
-            return "Review Blog", "Error: API request timed out after 30 seconds.", gr.update(visible=False)
-
+    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:
-        report = f"Error analyzing content with Gemini: {str(e)}"
-        print("Available models:")
-        for m in genai.list_models():
-            print(m.name)
-        return "Review Blog", report, gr.update(visible=False)
+        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)
-            temp_file_path = temp_file.name
-        progress(1.0, desc="Report generated!")
-        return "Review Blog", report, gr.update(visible=True, value=temp_file_path)
+            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 creating temporary file: {str(e)}", gr.update(visible=False)
+        return "Review Blog", f"Error saving report: {str(e)}", gr.update(visible=False)
 
-# Enhanced custom CSS for a more attractive and user-friendly UI
+# ------------------ CSS ------------------ #
 custom_css = """
-@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap');
+@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(135deg, #f0f4f8 0%, #d9e2ec 100%);
-    padding: 30px;
-    border-radius: 15px;
-    box-shadow: 0 6px 25px rgba(0, 0, 0, 0.15);
-    color: #2d3748;
+    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 {
-    color: #2b6cb0;
-    text-align: center;
     font-weight: 700;
-    margin-bottom: 15px;
-    text-shadow: 1px 1px 3px rgba(0, 0, 0, 0.1);
-}
-.review-btn {
-    background: linear-gradient(45deg, #4a90e2, #63b3ed);
-    color: white;
-    font-weight: 600;
-    border: none;
-    border-radius: 12px;
-    padding: 14px 28px;
-    transition: all 0.3s ease;
-    box-shadow: 0 4px 12px rgba(74, 144, 226, 0.3);
-    cursor: pointer;
-}
-.review-btn:hover {
-    background: linear-gradient(45deg, #63b3ed, #4a90e2);
-    transform: translateY(-3px);
-    box-shadow: 0 6px 15px rgba(74, 144, 226, 0.4);
-}
-.review-btn:disabled {
-    opacity: 0.7;
-    cursor: not-allowed;
-    transform: none;
-    box-shadow: none;
-}
-.review-btn:disabled::after {
-    content: ' ⏳';
-}
-.tab-nav button {
-    font-family: 'Inter', sans-serif;
-    font-weight: 600;
-    background: #edf2f7;
-    color: #2d3748;
-    border-radius: 10px 10px 0 0;
-    padding: 12px 24px;
-    transition: all 0.2s ease;
-    border: 1px solid #e2e8f0;
-}
-.tab-nav button:hover {
-    background: #4a90e2;
-    color: white;
-    border-color: #4a90e2;
+    color: #2b6cb0;
 }
+
 input, textarea {
-    font-family: 'Inter', sans-serif;
-    border: 2px solid #e2e8f0;
-    border-radius: 10px;
-    padding: 12px;
-    background: #ffffff;
-    box-shadow: 0 2px 6px rgba(0, 0, 0, 0.05);
-    color: #2d3748;
+    border: 2px solid #cbd5e0;
+    border-radius: 12px;
+    padding: 15px;
+    background-color: #ffffff;
     font-size: 16px;
+    color: #2d3748;
+    transition: border 0.3s ease;
 }
-.gr-textbox textarea {
-    resize: vertical;
-    min-height: 150px;
+
+input:focus, textarea:focus {
+    border-color: #3182ce;
+    outline: none;
 }
-.gr-progress {
-    background: #edf2f7;
+
+.review-btn {
+    background: linear-gradient(45deg, #3182ce, #63b3ed);
+    padding: 14px 28px;
     border-radius: 12px;
-    overflow: hidden;
-    height: 30px;
+    font-size: 16px;
+    color: #fff;
+    font-weight: 600;
+    border: none;
+    cursor: pointer;
+    transition: all 0.3s ease;
 }
-.gr-progress > div {
-    background: linear-gradient(90deg, #48bb78, #38a169);
-    height: 100%;
-    transition: width 0.5s ease;
+
+.review-btn:hover {
+    transform: scale(1.04);
+    background: linear-gradient(45deg, #63b3ed, #3182ce);
 }
+
 .markdown-container {
     background: #ffffff;
-    border-radius: 12px;
     padding: 25px;
-    box-shadow: 0 4px 15px rgba(0, 0, 0, 0.1);
-    margin-top: 20px;
-    color: #2d3748;
+    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: #48bb78;
+    background-color: #38a169;
     color: white;
+    padding: 12px 20px;
     border-radius: 10px;
-    padding: 12px 24px;
     font-weight: 600;
-    transition: all 0.3s ease;
-    cursor: pointer;
+    transition: background 0.3s ease;
 }
+
 .download-btn:hover {
-    background: #38a169;
-    transform: translateY(-3px);
+    background-color: #2f855a;
 }
 """
 
-# Gradio UI with enhanced layout
-with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as demo:
-    gr.Markdown(
-        """
-        # 📝 AI Blog Reviewer
-        Easily analyze your blog for grammar, legal issues, crude language, and sensitive topics.
-        """
-    )
+# ------------------ UI ------------------ #
+with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     with gr.Row():
-        with gr.Column(scale=2):
+        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"):
+                with gr.TabItem("✍️ Text Input"):
                     text_input = gr.Textbox(
-                        lines=10,
                         label="Paste Your Blog Content",
-                        placeholder="Enter your blog text here...",
+                        placeholder="Write or paste your blog text here...",
+                        lines=12,
                         elem_classes=["input-text"]
                     )
-                with gr.TabItem("URL Input"):
+                with gr.TabItem("🔗 URL Input"):
                     url_input = gr.Textbox(
-                        lines=1,
                         label="Enter Blog URL",
                         placeholder="https://example.com/blog",
+                        lines=1,
                         elem_classes=["input-url"]
                     )
-        with gr.Column(scale=1):
-            gr.Markdown(
-                """
-                ### How It Works
-                1. **Input**: Paste text or provide a URL.
-                2. **Analysis**: AI checks grammar, legal issues, and more.
-                3. **Report**: Download a detailed markdown report.
-                """
-            )
-            status_button = gr.Button("Review Blog", elem_classes=["review-btn"])
-    
+        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():
@@ -283,4 +226,4 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as demo:
         outputs=[status_button, report_output, download_btn]
     )
 
-demo.launch()
+demo.launch()