Update app.py

app.py

@@ -45,8 +45,8 @@ os.makedirs(CACHE_DIR, exist_ok=True)
 # Maximum dimensions for images
 MAX_IMAGE_DIMENSION = 500
 
-# Timeout for OCR operation (increased)
-OCR_TIMEOUT = 20  # Increased from 10 to 20 seconds
+# Timeout for OCR operation
+OCR_TIMEOUT = 20
 
 # Crawling settings
 CRAWL_DEPTH = 1
@@ -73,7 +73,7 @@ def preprocess_image(image):
         image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
         logger.info(f"Resized image to: {new_width}x{new_height}")
     else:
-        target_dpi = 50  # Reduced from 72 for faster OCR
+        target_dpi = 50
         image = image.resize((int(width * target_dpi / 72), int(height * target_dpi / 72)), Image.Resampling.LANCZOS)
     image = image.convert("L")
     return image
@@ -271,34 +271,43 @@ except Exception as e:
 async def home():
     return render_template("index.html")
 
+async def process_web_translate():
+    start_time = time.time()
+    text = request.form.get("text")
+    file = request.files.get("file")
+
+    logger.info(f"Received text: {text}, file: {file.filename if file else None}")
+
+    if not text and not file:
+        return render_template("index.html", error="Please enter text or upload a file.")
+
+    if file and allowed_file(file.filename):
+        logger.info("Starting OCR extraction for uploaded file")
+        extracted_text = extract_text(file)
+        logger.info(f"OCR result: {extracted_text}")
+        if extracted_text.startswith("Error") or extracted_text.startswith("Please"):
+            return render_template("index.html", error=extracted_text, text=text)
+        text_to_translate = extracted_text
+    else:
+        text_to_translate = text
+
+    if not text_to_translate:
+        return render_template("index.html", error="No valid text to translate.", text=text)
+
+    logger.info("Starting translation")
+    translated = await translate_text(text_to_translate, model, tokenizer)
+    logger.info(f"Translation result: {translated}")
+    if translated.startswith("Error"):
+        return render_template("index.html", error=translated, text=text, extracted_text=text_to_translate)
+
+    logger.info(f"Total request took {time.time() - start_time:.2f} seconds")
+    return render_template("index.html", extracted_text=text_to_translate, translated_text=translated, text=text)
+
 @app.route("/web_translate", methods=["POST"])
 async def web_translate():
+    text = None  # Initialize text to avoid UnboundLocalError
     try:
-        async with asyncio.timeout(60):  # Timeout for the entire route
-            start_time = time.time()
-            text = request.form.get("text")
-            file = request.files.get("file")
-
-            if not text and not file:
-                return render_template("index.html", error="Please enter text or upload a file.")
-
-            if file and allowed_file(file.filename):
-                extracted_text = extract_text(file)
-                if extracted_text.startswith("Error") or extracted_text.startswith("Please"):
-                    return render_template("index.html", error=extracted_text, text=text)
-                text_to_translate = extracted_text
-            else:
-                text_to_translate = text
-
-            if not text_to_translate:
-                return render_template("index.html", error="No valid text to translate.", text=text)
-
-            translated = await translate_text(text_to_translate, model, tokenizer)
-            if translated.startswith("Error"):
-                return render_template("index.html", error=translated, text=text, extracted_text=text_to_translate)
-
-            logger.info(f"Total request took {time.time() - start_time:.2f} seconds")
-            return render_template("index.html", extracted_text=text_to_translate, translated_text=translated, text=text)
+        return await asyncio.wait_for(process_web_translate(), timeout=60)  # Timeout after 60 seconds
     except asyncio.TimeoutError:
         logger.error("Request timed out after 60 seconds")
         return render_template("index.html", error="Request timed out. Please try again with a smaller input.", text=text)
@@ -306,6 +315,42 @@ async def web_translate():
         logger.error(f"Error in web_translate: {str(e)}")
         return render_template("index.html", error=f"Error processing request: {str(e)}", text=text)
 
+async def process_crawl_and_translate():
+    start_time = time.time()
+    url = request.form.get("url")
+
+    if not url:
+        return render_template("index.html", error="Please enter a website URL.")
+
+    parsed_url = urlparse(url)
+    if not parsed_url.scheme or not parsed_url.netloc:
+        return render_template("index.html", error="Invalid URL format.", url=url)
+
+    logger.info(f"Starting crawl for URL: {url}")
+    extracted_text = crawl_website(url)
+    
+    if not extracted_text or extracted_text.startswith("No Bangla text"):
+        return render_template("index.html", error=extracted_text or "No text found to translate.", url=url)
+
+    translated = await translate_text(extracted_text, model, tokenizer)
+    if translated.startswith("Error"):
+        return render_template("index.html", error=translated, url=url, extracted_text=extracted_text)
+
+    logger.info(f"Total crawl and translate request took {time.time() - start_time:.2f} seconds")
+    return render_template("index.html", extracted_text=extracted_text, translated_text=translated, url=url)
+
+@app.route("/crawl_and_translate", methods=["POST"])
+async def crawl_and_translate():
+    url = None  # Initialize url to avoid UnboundLocalError
+    try:
+        return await asyncio.wait_for(process_crawl_and_translate(), timeout=60)  # Timeout after 60 seconds
+    except asyncio.TimeoutError:
+        logger.error("Crawl and translate request timed out after 60 seconds")
+        return render_template("index.html", error="Request timed out. Please try again with a smaller website.", url=url)
+    except Exception as e:
+        logger.error(f"Error in crawl_and_translate: {str(e)}")
+        return render_template("index.html", error=f"Error processing request: {str(e)}", url=url)
+
 @app.route("/translate", methods=["POST"])
 async def translate():
     try:
@@ -322,38 +367,5 @@ async def translate():
         logger.error(f"Error in translate: {str(e)}")
         return jsonify({"error": str(e)}), 500
 
-@app.route("/crawl_and_translate", methods=["POST"])
-async def crawl_and_translate():
-    try:
-        async with asyncio.timeout(60):  # Timeout for the entire route
-            start_time = time.time()
-            url = request.form.get("url")
-
-            if not url:
-                return render_template("index.html", error="Please enter a website URL.")
-
-            parsed_url = urlparse(url)
-            if not parsed_url.scheme or not parsed_url.netloc:
-                return render_template("index.html", error="Invalid URL format.", url=url)
-
-            logger.info(f"Starting crawl for URL: {url}")
-            extracted_text = crawl_website(url)
-            
-            if not extracted_text or extracted_text.startswith("No Bangla text"):
-                return render_template("index.html", error=extracted_text or "No text found to translate.", url=url)
-
-            translated = await translate_text(extracted_text, model, tokenizer)
-            if translated.startswith("Error"):
-                return render_template("index.html", error=translated, url=url, extracted_text=extracted_text)
-
-            logger.info(f"Total crawl and translate request took {time.time() - start_time:.2f} seconds")
-            return render_template("index.html", extracted_text=extracted_text, translated_text=translated, url=url)
-    except asyncio.TimeoutError:
-        logger.error("Crawl and translate request timed out after 60 seconds")
-        return render_template("index.html", error="Request timed out. Please try again with a smaller website.", url=url)
-    except Exception as e:
-        logger.error(f"Error in crawl_and_translate: {str(e)}")
-        return render_template("index.html", error=f"Error processing request: {str(e)}", url=url)
-
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=False)