added language choice

app.py

@@ -8,6 +8,8 @@ from impresso_pipelines.ocrqa import OCRQAPipeline
 
 pipeline = OCRQAPipeline()
 
+LANGUAGES = ["en", "de", "fr"]
+
 # Example OCR text (German text with typical OCR errors)
 EXAMPLE_TEXT = """Vieles Seltsame geschieht auf Erden :
 Nichts Seltsameres sieht der Mond
@@ -22,9 +24,10 @@ Und so jagt er zuletzt auf die Brücke,
 Die über dem Abgrund, d:m nächtlich schwarzen
 Jählings abbricht."""
 
-def process_ocr_qa(text):
+def process_ocr_qa(text, lang_choice):
     try:
-        result = pipeline(text, diagnostics=True)
+        lang = None if lang_choice == "Auto-detect" else lang_choice
+        result = pipeline(text, language=lang, diagnostics=True)
 
         # Format the output for better readability
         if isinstance(result, dict):
@@ -99,6 +102,11 @@ with gr.Blocks(title="OCR QA Demo") as demo:
                 lines=8,
                 placeholder="Enter your OCR text here..."
             )
+            lang_dropdown = gr.Dropdown(
+                choices=["Auto-detect"] + LANGUAGES, 
+                value="de", 
+                label="Language"
+            )
             submit_btn = gr.Button("🔍 Analyze OCR Quality", variant="primary")
         
         with gr.Column():
@@ -123,7 +131,7 @@ with gr.Blocks(title="OCR QA Demo") as demo:
     
     submit_btn.click(
         fn=process_ocr_qa,
-        inputs=[text_input],
+        inputs=[text_input, lang_dropdown],
         outputs=output
     )