Update app.py

app.py

@@ -19,18 +19,17 @@ if not api_key:
 
 genai.configure(api_key=api_key)
 
-# Use gemini-1.5-flash for faster and more accessible text analysis
+# Use gemini-1.5-flash for faster text analysis
 try:
     model = genai.GenerativeModel('gemini-1.5-flash')
 except Exception as e:
-    # Fallback: List available models if the specified model is not found
     print(f"Error initializing model: {str(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.")
 
-# Prompt for Gemini to analyze text with specified output format
+# Prompt for Gemini to analyze text
 PROMPT = """
 You are an AI content reviewer. Analyze the provided text for the following:
 1. *Grammar Issues*: Identify and suggest corrections for grammatical errors.
@@ -74,17 +73,18 @@ async def fetch_url_content(url):
         response = requests.get(url, timeout=10)
         response.raise_for_status()
         soup = BeautifulSoup(response.text, 'html.parser')
-        # Extract text from common content tags
         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."
     except Exception as e:
         return f"Error fetching URL: {str(e)}"
 
 async def review_blog(text_input, url_input):
-    # Start loading effect immediately
+    # Initialize output variables
     button_text = "Processing..."
+    report = ""
+    download_path = None
 
-    # Determine input type based on which field is populated
+    # Determine input type
     if text_input and not url_input:
         input_type = "Text"
         input_text = text_input
@@ -99,17 +99,22 @@ async def review_blog(text_input, url_input):
         return "Review Blog", "Error: No input provided.", gr.update(visible=False)
 
     try:
-        # Use asyncio.wait_for for timeout in Python 3.10
+        # Wrap the entire process in a timeout
         async def process_with_timeout():
+            nonlocal button_text, report, download_path
+
             # Handle URL input
             if input_type == "URL":
                 button_text = "Fetching content..."
-                input_text = await fetch_url_content(input_text)
-                if input_text.startswith("Error"):
-                    return "Review Blog", input_text, gr.update(visible=False)
+                content = await fetch_url_content(input_text)
+                if content.startswith("Error"):
+                    return "Review Blog", content, gr.update(visible=False)
+                input_text_content = content
+            else:
+                input_text_content = input_text
 
             # Tokenize input for analysis
-            sentences = sent_tokenize(input_text)
+            sentences = sent_tokenize(input_text_content)
             analysis_text = "\n".join(sentences)
 
             # Update button for API call
@@ -117,11 +122,9 @@ async def review_blog(text_input, url_input):
             try:
                 response = await asyncio.to_thread(model.generate_content, PROMPT + "\n\nText to analyze:\n" + analysis_text)
                 report = response.text.strip()
-                # Ensure the response is markdown by removing any code fences
-                report = re.sub(r'^markdown\n|$', '', report, flags=re.MULTILINE)
+                report = re.sub(r'^```markdown\n|```$', '', report, flags=re.MULTILINE)
             except Exception as e:
                 report = f"Error analyzing content with Gemini: {str(e)}. Please check your API key, network connection, or model availability."
-                # Fallback: List available models for debugging
                 print("Available models:")
                 for m in genai.list_models():
                     print(m.name)
@@ -131,18 +134,19 @@ async def review_blog(text_input, url_input):
             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
-                # Add note to scroll to report
+                    download_path = temp_file.name
                 report = f"**Report generated, please scroll down to view.**\n\n{report}"
-                return "Review Blog", report, gr.update(visible=True, value=temp_file_path)
+                return "Review Blog", report, gr.update(visible=True, value=download_path)
             except Exception as e:
                 return "Review Blog", f"Error creating temporary file: {str(e)}", gr.update(visible=False)
 
-        # Run the process with a 30-second timeout
+        # Execute with timeout
         return await asyncio.wait_for(process_with_timeout(), timeout=30)
 
     except asyncio.TimeoutError:
-        return "Timeout", "Error: Process timed out after 30 seconds.", gr.update(visible=False)
+        return "Review Blog", "Error: Process timed out after 30 seconds.", gr.update(visible=False)
+    except Exception as e:
+        return "Review Blog", f"Unexpected error: {str(e)}", gr.update(visible=False)
 
 # Custom CSS for hover effect, loading state, and Inter font
 custom_css = """
@@ -217,4 +221,4 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as demo:
         outputs=[status_button, report_output, download_btn]
     )
 
-demo.launch()
+demo.launch()