Update app.py

app.py

@@ -45,6 +45,10 @@ class HindiSignboardTranslator:
             str: Translated English text
         """
         try:
+            # Handle empty or None input
+            if not hindi_text:
+                return "No text detected"
+            
             # Tokenize and translate
             inputs = self.tokenizer(hindi_text, return_tensors="pt", padding=True)
             outputs = self.model.generate(**inputs)
@@ -52,7 +56,7 @@ class HindiSignboardTranslator:
             return english_text
         except Exception as e:
             print(f"Translation Error: {e}")
-            return None
+            return "Translation failed"
     
     def translate_signboard(self, image):
         """
@@ -62,6 +66,15 @@ class HindiSignboardTranslator:
         Returns:
             dict: Translation results
         """
+        # Validate input
+        if image is None:
+            return {
+                "status": "error",
+                "message": "No image provided",
+                "original_text": "",
+                "translated_text": ""
+            }
+        
         # Extract text via OCR
         hindi_text = self.extract_text(image)
         
@@ -79,7 +92,7 @@ class HindiSignboardTranslator:
         return {
             "status": "success",
             "original_text": hindi_text,
-            "translated_text": english_text or "Translation failed"
+            "translated_text": english_text
         }
 
 # Initialize the translator
@@ -106,10 +119,7 @@ iface = gr.Interface(
     ],
     title="Hindi Signboard Translator",
     description="Upload a Hindi signboard image to extract and translate its text.",
-    examples=[
-        ["example_signboard1.jpg"],
-        ["example_signboard2.jpg"]
-    ]
+    # Removed example images
 )
 
 # Launch the app