Add: Claude-only restriction for table summaries

ui.py

@@ -4,8 +4,9 @@ from ocr_engine import extract_text_from_image
 from pdf_reader import extract_text_chunks_from_pdf
 from summarizer import summarize_text
 
-def process_input(pdf, image, manual_text, mode, model_name, start_page, end_page, lang_mode):
-
+def process_input(pdf, image, manual_text, mode, model_name, start_page, end_page, lang_mode, is_table):
+    if is_table and model_name != "anthropic/claude-3-haiku":
+        return "Tablo içeriği için yalnızca Claude önerilir.","",None
     if pdf is not None:
         text_chunks = extract_text_chunks_from_pdf(pdf, start=int(start_page), end=int(end_page))
         if any("[ERROR]" in chunk for chunk in text_chunks):
@@ -24,7 +25,7 @@ def process_input(pdf, image, manual_text, mode, model_name, start_page, end_pag
     summaries = []
 
     for chunk in text_chunks:
-        summary = summarize_text(chunk, mode, model_name, lang_mode)
+        summary = summarize_text(chunk, mode, model_name, lang_mode, is_table)
         summaries.append(summary)
 
     full_summary = "\n\n".join(summaries)
@@ -104,10 +105,12 @@ with gr.Blocks() as demo:
         label="Dil Algılama / Çeviri Modu",
         value="Otomatik"
     )
+    
+    is_table = gr.Checkbox(label="Tablo içeriyor (Claude tablo gibi özetlesin)", value=False)
 
     submit_btn.click(
         fn=process_input,
-        inputs=[pdf_input, image_input, manual_input, mode_selector, model_selector, start_page, end_page, lang_mode],
+        inputs=[pdf_input, image_input, manual_input, mode_selector, model_selector, start_page, end_page, lang_mode, is_table],
         outputs=[text_output, summary_output, summary_file]
     )