Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 from transformers import NougatProcessor, VisionEncoderDecoderModel
 from PIL import Image
 import fitz  # PyMuPDF
-from typing import List
+from typing import List, Iterator, Tuple
 import os
 import requests
 from io import BytesIO
@@ -37,57 +37,68 @@ def process_single_image(image: Image.Image) -> str:
         return "Model yüklenemedi veya geçersiz görüntü."
         
     try:
-        # Görüntüyü modelin beklediği formata dönüştürme
         pixel_values = processor(images=image, return_tensors="pt").pixel_values
         
-        # Metin üretimi
         outputs = model.generate(
             pixel_values.to(device),
             min_length=1,
-            max_new_tokens=4096, # Hugging Face ücretsiz katmanlarında daha düşük bir değere ayarlamanız gerekebilir
+            max_new_tokens=4096,
             bad_words_ids=[[processor.tokenizer.unk_token_id]],
         )
         
-        # Çıktıyı okunabilir metne dönüştürme ve son işleme
         sequence = processor.batch_decode(outputs, skip_special_tokens=True)
         sequence = processor.post_process_generation(sequence[0], fix_markdown=False)
         
         return sequence
     except Exception as e:
+        # Hatanın konsola yazdırılması
+        print(f"Görüntü işleme hatası: {e}")
         return f"Görüntü işlenirken bir hata oluştu: {e}"
 
-def process_pdf_file(pdf_file) -> str:
-    """Yüklenen bir PDF dosyasını işler, her sayfasını dönüştürür ve birleştirir."""
-    if not MODEL_LOADED or pdf_file is None:
-        return "Model yüklenemedi veya PDF dosyası yüklenmedi."
+def process_pdf_file(pdf_file) -> Iterator[Tuple[str, str]]:
+    """Yüklenen bir PDF dosyasını işler, her sayfa için durum güncellemesi yapar."""
+    if not MODEL_LOADED:
+        yield "Hata: Model yüklenemedi.", ""
+        return
+    if pdf_file is None:
+        yield "Hata: PDF dosyası yüklenmedi.", ""
+        return
 
-    full_markdown_content = []
-    doc = None  # doc değişkenini try bloğundan önce tanımla
+    doc = None
     try:
-        # DÜZELTME: Gradio'nun File bileşeni bir tempfile nesnesi döndürür.
-        # Bu nesnenin .name özelliği dosya yolunu içerir.
-        # PyMuPDF'in open fonksiyonuna doğrudan bu dosya yolunu veriyoruz.
+        yield "PDF dosyası açılıyor...", ""
         doc = fitz.open(pdf_file.name)
+        total_pages = len(doc)
         
-        for page_num in range(len(doc)):
-            page = doc.load_page(page_num)
+        if total_pages == 0:
+            yield "Hata: PDF dosyasında işlenecek sayfa bulunamadı.", ""
+            return
+
+        full_markdown_content = []
+        for page_num in range(total_pages):
+            status_message = f"Sayfa {page_num + 1} / {total_pages} işleniyor..."
+            # Her sayfadan önce durum güncellemesi ve o ana kadarki içeriği gönder
+            yield status_message, "\n\n---\n\n".join(full_markdown_content)
             
-            # Sayfayı yüksek çözünürlüklü bir görüntüye dönüştürme
+            page = doc.load_page(page_num)
             pix = page.get_pixmap(dpi=150)
             image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples).convert("RGB")
             
-            # Her sayfayı tekil görüntü olarak işleme
             page_markdown = process_single_image(image)
             full_markdown_content.append(f"## Sayfa {page_num + 1}\n\n{page_markdown}")
 
-        return "\n\n---\n\n".join(full_markdown_content)
+        # Sonuç
+        final_output = "\n\n---\n\n".join(full_markdown_content)
+        yield "İşlem tamamlandı!", final_output
+
     except Exception as e:
-        return f"PDF işlenirken bir hata oluştu: {e}"
+        error_msg = f"PDF işlenirken bir hata oluştu: {e}"
+        print(error_msg) # Hatanın sunucu loglarına yazdırılması
+        yield error_msg, "" # Hatanın arayüzde gösterilmesi
     finally:
         if doc:
             doc.close()
 
-
 # --- Gradio Arayüzü ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
@@ -103,6 +114,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.TabItem("PDF Dosyasını İşle"):
             pdf_input = gr.File(label="PDF Dosyası Yükle", file_types=[".pdf"])
             pdf_process_button = gr.Button("PDF'i Dönüştür", variant="primary")
+            # YENİ: Durum mesajları için bir metin kutusu eklendi
+            pdf_status = gr.Textbox(label="İşlem Durumu", value="İşlem bekleniyor...", interactive=False)
             pdf_output = gr.Markdown(label="Dönüştürülen Metin (Markdown)")
 
         # Tek Görüntü İşleme Sekmesi
@@ -111,11 +124,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             image_process_button = gr.Button("Görüntüyü Dönüştür", variant="primary")
             image_output = gr.Markdown(label="Dönüştürülen Metin (Markdown)")
     
-    # Buton tıklama olaylarını ilgili fonksiyonlara bağlama
+    # Buton tıklama olayı güncellendi
     pdf_process_button.click(
         fn=process_pdf_file,
         inputs=[pdf_input],
-        outputs=[pdf_output],
+        # YENİ: Çıktılar hem durum kutusunu hem de sonuç kutusunu güncelliyor
+        outputs=[pdf_status, pdf_output],
         api_name="process_pdf"
     )
     
@@ -127,8 +141,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     
     gr.Markdown("### Örnek Kullanım")
-    # Örnek bir resim dosyası oluşturun veya yükleyin ve adını buraya yazın.
-    # Örneğin, projenizin kök dizininde 'nougat_paper_example.png' adında bir dosya olmalı.
     example_image_path = "nougat_paper_example.png"
     if os.path.exists(example_image_path):
         gr.Examples(
@@ -141,7 +153,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         )
 
 if __name__ == "__main__":
-    # Örnek resim dosyasını indirme (eğer yoksa)
     if not os.path.exists("nougat_paper_example.png"):
         try:
             url = "https://huggingface.co/datasets/hf-internal-testing/fixtures_docvqa/resolve/main/nougat_paper.png"
@@ -152,4 +163,4 @@ if __name__ == "__main__":
         except Exception as e:
             print(f"Örnek resim indirilemedi: {e}")
 
-    demo.launch(debug=True)
+    demo.launch(debug=True)