PDF_Linguist/Digital_PDF.py at main · NitinReddy-A/PDF_Linguist · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
import streamlit as st
import fitz
from io import BytesIO
import convertapi
from DigitalPDF_Final import app
import tempfile
import os

def display_pdf_with_scrollbar(input_pdf_bytes):
    """
    Display the PDF with a scrollbar for page navigation.
    """
    try:
        # Open the PDF with PyMuPDF
        pdf_document = fitz.open(stream=input_pdf_bytes, filetype="pdf")
        total_pages = pdf_document.page_count

        # Add a slider to navigate pages
        st.markdown("### PDF Preview:")
        page_num = st.slider("Select Page", min_value=0, max_value=total_pages, value=1)
        page = pdf_document[page_num - 1]
        pix = page.get_pixmap()  # Render page to an image
        img_data = BytesIO(pix.tobytes("png"))
        st.image(img_data, caption=f"Page {page_num} of {total_pages}", use_column_width=True)

        pdf_document.close()
    except Exception as e:
        st.error(f"Unable to generate preview: {e}")


def process_pdfk(input_pdf_bytes):
    """
    Process the uploaded PDF to extract text, apply translations, and generate output.
    """
    try:
        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
            temp_pdf.write(input_pdf_bytes)
            temp_pdf_path = temp_pdf.name


        # Text analysis and translation
        pdf_path = temp_pdf_path
        output_json_path = r"extracted_text_with_coordinates.json"
        app.extract_text_with_coordinates(pdf_path, output_json_path)
        app.add_text_and_character_count(pdf_path, output_json_path)
        app.translate_and_insert_newlinesk(pdf_path, output_json_path, dest_language='kn')
        font_path = r"NotoSansKannada-VariableFont_wdth,wght.ttf"

        output_pdf_path = r"TranslatedOutput.pdf"
        app.create_pdf_with_images(pdf_path, output_pdf_path)

        app.create_translated_pdf(output_json_path, output_pdf_path, font_path)

        # Return processed PDF as bytes
        with open(output_pdf_path, "rb") as f:
            return f.read()
    finally:
        os.unlink(temp_pdf_path)

def process_pdfh(input_pdf_bytes):
    """
    Process the uploaded PDF to extract text, apply translations, and generate output.
    """
    try:
        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
            temp_pdf.write(input_pdf_bytes)
            temp_pdf_path = temp_pdf.name


        # Text analysis and translation
        pdf_path = temp_pdf_path
        output_json_path = r"extracted_text_with_coordinates.json"
        app.extract_text_with_coordinates(pdf_path, output_json_path)
        app.add_text_and_character_count(pdf_path, output_json_path)
        app.translate_and_insert_newlinesh(pdf_path, output_json_path, dest_language='hi')
        font_path = r"NotoSansKannada-VariableFont_wdth,wght.ttf"

        output_pdf_path = r"TranslatedOutput.pdf"
        app.create_pdf_with_images(pdf_path, output_pdf_path)

        app.create_translated_pdf(output_json_path, output_pdf_path, font_path)

        # Return processed PDF as bytes
        with open(output_pdf_path, "rb") as f:
            return f.read()
    finally:
        os.unlink(temp_pdf_path)


def process_pdfr(input_pdf_bytes):
    """
    Process the uploaded PDF to extract text, apply translations, and generate output.
    """
    try:
        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
            temp_pdf.write(input_pdf_bytes)
            temp_pdf_path = temp_pdf.name

        # Text analysis and translation
        pdf_path = temp_pdf_path
        output_json_path = r"extracted_text_with_coordinates.json"
        app.extract_text_with_coordinates(pdf_path, output_json_path)
        app.add_text_and_character_count(pdf_path, output_json_path)
        app.translate_and_insert_newlinesr(pdf_path, output_json_path, dest_language='ru')
        font_path = r"NotoSansKannada-VariableFont_wdth,wght.ttf"

        output_pdf_path = r"TranslatedOutput.pdf"
        app.create_pdf_with_images(pdf_path, output_pdf_path)

        app.create_translated_pdf(output_json_path, output_pdf_path, font_path)

        # Return processed PDF as bytes
        with open(output_pdf_path, "rb") as f:
            return f.read()
    finally:
        os.unlink(temp_pdf_path)

# Streamlit UI
st.title("PDF Linguist")
st.markdown("Upload a Digital PDF and download the translated result.")

uploaded_pdf = st.file_uploader("Upload a PDF file", type=["pdf"])
if uploaded_pdf is not None:
    st.success(f"Uploaded file: {uploaded_pdf.name}")
    input_pdf_bytes = uploaded_pdf.read()

    # Display PDF preview with scrollbar
    display_pdf_with_scrollbar(input_pdf_bytes)

    st.markdown("### Translate PDF:")
    col1, col2, col3 = st.columns([1, 1, 1])
    with col1:
        if st.button("Kannada"):
            st.markdown("### Processing...")
            try:
                processed_pdf_bytes = process_pdfk(input_pdf_bytes)
                st.success("Processing complete!")
                st.markdown("### Download the Translated PDF")
                st.download_button(
                    label="Download",
                    data=processed_pdf_bytes,
                    file_name=f"Translated_{uploaded_pdf.name}",
                    mime="application/pdf",
                )
            except Exception as e:
                st.error(f"Error during processing: {e}")

    with col2:
        if st.button("Hindi"):
            st.markdown("### Processing...")
            try:
                processed_pdf_bytes = process_pdfh(input_pdf_bytes)
                st.success("Processing complete!")
                st.markdown("### Download the Translated PDF")
                st.download_button(
                    label="Download",
                    data=processed_pdf_bytes,
                    file_name=f"Translated_{uploaded_pdf.name}",
                    mime="application/pdf",
                )
            except Exception as e:
                st.error(f"Error during processing: {e}")

    with col3:
        if st.button("Russian"):
            st.markdown("### Processing...")
            try:
                processed_pdf_bytes = process_pdfr(input_pdf_bytes)
                st.success("Processing complete!")
                st.markdown("### Download the Translated PDF")
                st.download_button(
                    label="Download",
                    data=processed_pdf_bytes,
                    file_name=f"Translated_{uploaded_pdf.name}",
                    mime="application/pdf",
                )
            except Exception as e:
                st.error(f"Error during processing: {e}")