Update app.py

app.py

@@ -6,7 +6,7 @@ import pytesseract
 from PIL import Image
 import io
 import os
-from huggingface_hub import HfApi, create_repo
+from huggingface_hub import HfApi, create_repo, RepositoryNotFoundError
 import re
 from datetime import datetime
 import urllib.parse
@@ -36,6 +36,7 @@ def ensure_hf_dataset():
     """Create or get Hugging Face dataset repository."""
     try:
         repo_id = create_repo(repo_id=REPO_NAME, token=HF_TOKEN, repo_type="dataset", exist_ok=True)
+        logger.info(f"Using dataset repo: {repo_id}")
         return repo_id
     except Exception as e:
         logger.error(f"Error creating dataset repo: {str(e)}")
@@ -61,6 +62,7 @@ def upload_image_to_hf(image, filename):
             token=HF_TOKEN
         )
         os.remove(temp_path)
+        logger.info(f"Uploaded image to: {file_url}")
         return file_url
     except Exception as e:
         logger.error(f"Error uploading image: {str(e)}")
@@ -187,4 +189,5 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    # In Hugging Face Spaces, share=False is sufficient as Spaces handles the server
+    iface.launch(share=False)