Update app.py

app.py

@@ -6,6 +6,7 @@ from transformers import NougatProcessor, VisionEncoderDecoderModel
 from PIL import Image
 import fitz  # PyMuPDF
 from typing import List
+import os
 
 # --- Model ve İşlemci Yükleme ---
 # Modelin yalnızca bir kez yüklenmesini sağlamak için global olarak tanımlıyoruz.
@@ -34,14 +35,14 @@ 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 [1]
+        # Görüntüyü modelin beklediği formata dönüştürme
         pixel_values = processor(images=image, return_tensors="pt").pixel_values
         
-        # Metin üretimi [1]
+        # Metin üretimi
         outputs = model.generate(
             pixel_values.to(device),
             min_length=1,
-            max_new_tokens=4096,  # Sayfa içeriğine göre ayarlanabilir
+            max_new_tokens=4096,
             bad_words_ids=[[processor.tokenizer.unk_token_id]],
         )
         
@@ -49,6 +50,7 @@ def process_single_image(image: Image.Image) -> str:
         sequence = processor.batch_decode(outputs, skip_special_tokens=True)
         sequence = processor.post_process_generation(sequence, fix_markdown=False)
         
+        # batch_decode bir liste döndürür, bu yüzden ilk elemanı alıyoruz.
         return "".join(sequence)
     except Exception as e:
         return f"Görüntü işlenirken bir hata oluştu: {e}"
@@ -58,8 +60,9 @@ def process_pdf_file(pdf_file) -> str:
     if not MODEL_LOADED or pdf_file is None:
         return "Model yüklenemedi veya PDF dosyası yüklenmedi."
 
-    # HATA DÜZELTMESİ: Değişken boş bir liste olarak başlatıldı.
+    # DÜZELTME: Değişken, boş bir liste olarak doğru şekilde başlatıldı.
     full_markdown_content =
+    doc = None  # doc değişkenini try bloğundan önce tanımla
     try:
         # Gelen dosya nesnesinden PDF'i oku
         doc = fitz.open(stream=pdf_file.read(), filetype="pdf")
@@ -79,7 +82,7 @@ def process_pdf_file(pdf_file) -> str:
     except Exception as e:
         return f"PDF işlenirken bir hata oluştu: {e}"
     finally:
-        if 'doc' in locals() and doc:
+        if doc:
             doc.close()
 
 
@@ -96,15 +99,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Tabs():
         # PDF İşleme Sekmesi
         with gr.TabItem("PDF Dosyasını İşle"):
-            with gr.Row():
-                pdf_input = gr.File(label="PDF Dosyası Yükle", file_types=[".pdf"])
+            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")
             pdf_output = gr.Markdown(label="Dönüştürülen Metin (Markdown)")
 
         # Tek Görüntü İşleme Sekmesi
         with gr.TabItem("Tek Görüntü İşle"):
-            with gr.Row():
-                image_input = gr.Image(label="Belge Sayfası Görüntüsü Yükle", type="pil")
+            image_input = gr.Image(label="Belge Sayfası Görüntüsü Yükle", type="pil")
             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)")
     
@@ -124,14 +125,33 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     
     gr.Markdown("### Örnek Kullanım")
-    gr.Examples(
-        examples=["nougat_paper_example.png"],
-        inputs=image_input,
-        outputs=image_output,
-        fn=process_single_image,
-        cache_examples=True,
-        label="Örnek Görüntü"
-    )
+    # Ö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(
+            examples=[example_image_path],
+            inputs=image_input,
+            outputs=image_output,
+            fn=process_single_image,
+            cache_examples=True,
+            label="Örnek Görüntü"
+        )
 
 if __name__ == "__main__":
+    # Örnek resim dosyasını indirme (eğer yoksa)
+    if not os.path.exists("nougat_paper_example.png"):
+        try:
+            import requests
+            from PIL import Image
+            from io import BytesIO
+            
+            url = "https://huggingface.co/datasets/hf-internal-testing/fixtures_docvqa/resolve/main/nougat_paper.png"
+            response = requests.get(url)
+            img = Image.open(BytesIO(response.content))
+            img.save("nougat_paper_example.png")
+            print("Örnek resim 'nougat_paper_example.png' indirildi.")
+        except Exception as e:
+            print(f"Örnek resim indirilemedi: {e}")
+
     demo.launch(debug=True)