Update app.py

app.py

@@ -5,17 +5,42 @@ from transformers import pipeline
 # ✅ Load OCR Model (Text Extraction)
 ocr_pipeline = pipeline("image-to-text", model="microsoft/trocr-large-handwritten")
 
-# ✅ Load Translation Model
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-hi")  # English to Hindi
+# ✅ Load Translation Model (English to Hindi)
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-hi")
 
 def process_image(image):
-    # Extract text
-    extracted_text = ocr_pipeline(image)[0]['generated_text']
+    try:
+        # Extract text from image
+        ocr_result = ocr_pipeline(image)
 
-    # Translate text
-    translated_text = translator(extracted_text)[0]['translation_text']
+        # Debug: Check OCR output
+        print("OCR Raw Output:", ocr_result)
 
-    return extracted_text, translated_text
+        # Ensure the OCR output is valid
+        if not ocr_result or "generated_text" not in ocr_result[0]:
+            return "Error: Could not extract text. Try another image.", ""
+
+        extracted_text = ocr_result[0]['generated_text']
+        print("Extracted Text:", extracted_text)  # Debugging Output
+
+        # Handle empty extracted text
+        if not extracted_text.strip():
+            return "Error: No text detected in image.", ""
+
+        # Translate the extracted text
+        translated_result = translator(extracted_text)
+
+        # Debug: Check Translation output
+        print("Translation Raw Output:", translated_result)
+
+        translated_text = translated_result[0]['translation_text']
+        print("Translated Text:", translated_text)  # Debugging Output
+
+        return extracted_text, translated_text
+
+    except Exception as e:
+        print("Error:", str(e))  # Print error in the console
+        return f"Error: {str(e)}", ""
 
 # ✅ Gradio Interface
 iface = gr.Interface(