Update app.py

app.py

@@ -1,13 +1,19 @@
-"""
-Marathi Translation API with AI4Bharat IndicTrans2
+# --- CLEANUP CODE ADDED HERE ---
+# This will run once when the Space starts up.
+import os
+print("🧹 Running cleanup command...")
+os.system("huggingface-cli delete-cache && rm -rf ~/.cache/pip")
+print("✅ Cleanup complete.")
+# -----------------------------
+
+
+"""Marathi Translation API with AI4Bharat IndicTrans2
 High-quality translation service for English <-> Marathi
 """
-
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from pathlib import Path
-import os
 import json
 import torch
 
@@ -28,6 +34,7 @@ LANG_CODE_MAP = {
     "mr": "mar_Deva"
 }
 
+
 def load_translations_dict():
     """Load the custom translations dictionary"""
     global translations_dict
@@ -46,6 +53,7 @@ def load_translations_dict():
     
     return translations_dict
 
+
 def load_model():
     """Load the IndicTrans2 translation model"""
     global model, tokenizer
@@ -80,6 +88,7 @@ def load_model():
         print(f"✗ Failed to load model: {e}")
         raise
 
+
 def translate_with_indictrans2(text, source_lang, target_lang):
     """
     Translate using IndicTrans2 model
@@ -125,6 +134,7 @@ def translate_with_indictrans2(text, source_lang, target_lang):
     
     return translation
 
+
 def translate_with_dict(text, source, target):
     """
     Translate using dictionary first, fallback to IndicTrans2
@@ -151,6 +161,7 @@ def translate_with_dict(text, source, target):
         print(f"Translation error: {e}")
         return text, False
 
+
 @app.route('/')
 def home():
     """Home endpoint with API information"""
@@ -166,6 +177,7 @@ def home():
         }
     })
 
+
 @app.route('/health')
 def health():
     """Health check endpoint"""
@@ -184,6 +196,7 @@ def health():
             "error": str(e)
         }), 503
 
+
 @app.route('/languages', methods=['GET'])
 def languages():
     """Get supported languages"""
@@ -200,6 +213,7 @@ def languages():
         }
     ])
 
+
 @app.route('/translate', methods=['POST'])
 def translate():
     """
@@ -266,21 +280,24 @@ def translate():
         }
         
         return jsonify(response), 200
-        
+    
     except Exception as e:
         print(f"Error in translate endpoint: {e}")
         import traceback
         traceback.print_exc()
         return jsonify({"error": str(e)}), 500
 
+
 @app.errorhandler(404)
 def not_found(e):
     return jsonify({"error": "Endpoint not found"}), 404
 
+
 @app.errorhandler(500)
 def server_error(e):
     return jsonify({"error": "Internal server error"}), 500
 
+
 if __name__ == '__main__':
     # Pre-load model
     print("=" * 60)
@@ -290,6 +307,7 @@ if __name__ == '__main__':
     try:
         load_translations_dict()
         load_model()
+        
         print("\nStarting server...")
         
         # Get port from environment or use default (7860 for HF Spaces)
@@ -301,4 +319,4 @@ if __name__ == '__main__':
         print(f"\n✗ Failed to start server: {e}")
         import traceback
         traceback.print_exc()
-        exit(1)
+        exit(1)