Update app.py

app.py

@@ -481,14 +481,34 @@ def process_image_hf(image: Image.Image, prompt: str, use_api: bool) -> str:
 
 async def process_hf_ocr(image: Image.Image, output_file: str, use_api: bool) -> str:
     """ Performs OCR using the process_image_hf function framework. """
-    ocr_prompt = "Perform OCR on this image. Extract all text content." # More specific prompt
+    # Simple prompt for OCR task
+    ocr_prompt = "Extract text content from this image."
     result = process_image_hf(image, ocr_prompt, use_api=use_api) # Pass use_api flag
+
+    # Save the result if it looks like text (basic check)
     if result and not result.startswith("Error") and not result.startswith("["):
         try:
-            async with aiofiles.open(output_file, "w", encoding='utf-8') as f: await f.write(result)
+            async with aiofiles.open(output_file, "w", encoding='utf-8') as f:
+                await f.write(result)
             logger.info(f"HF OCR result saved to {output_file}")
-        except IOError as e: logger.error(f"Failed to save HF OCR output to {output_file}: {e}"); result += f"\n[Error saving file: {e}]"
-    elif os.path.exists(output_file): try: os.remove(output_file) except OSError: pass
+        except IOError as e:
+             logger.error(f"Failed to save HF OCR output to {output_file}: {e}")
+             result += f"\n[Error saving file: {e}]" # Append error to result if save fails
+
+    # --- CORRECTED BLOCK ---
+    elif os.path.exists(output_file):
+        # Remove file if processing failed or was just a placeholder message
+        try:
+            os.remove(output_file)
+        except OSError:
+            # Log error or just ignore if removal fails
+            logger.warning(f"Could not remove potentially empty/failed OCR file: {output_file}")
+            pass # Ignore removal error
+        except Exception as e_rem: # Catch any other error during removal
+             logger.warning(f"Error removing OCR file {output_file}: {e_rem}")
+             pass
+    # --- END CORRECTION ---
+
     return result
 
 # --- Character Functions (Keep from previous) -----------