Update app.py

app.py

@@ -1,22 +1,61 @@
 import streamlit as st
+import PyPDF2
 import io
 import os
-import pdfplumber
+
 
 unicodeatoz = ["ब", "द", "अ", "म", "भ", "ा", "न", "ज", "ष्", "व", "प", "ि", "फ", "ल", "य", "उ", "त्र", "च", "क", "त", "ग", "ख", "ध", "ह", "थ", "श"]
 unicodeAtoZ = ["ब्", "ध", "ऋ", "म्", "भ्", "ँ", "न्", "ज्", "क्ष्", "व्", "प्", "ी", "ः", "ल्", "इ", "ए", "त्त", "च्", "क्", "त्", "ग्", "ख्", "ध्", "ह्", "थ्", "श्"]
 unicode0to9 = ["ण्", "ज्ञ", "द्द", "घ", "द्ध", "छ", "ट", "ठ", "ड", "ढ"]
 symbolsDict = {
-    "~": "ञ्", "`": "ञ", "!": "१", "@": "२", "#": "३", "$": "४", "%": "५", "^": "६", "&": "७", "*": "८",
-    "(": "९", ")": "०", "-": "(", "_": ")", "+": "ं", "[": "ृ", "{": "र्", "]": "े", "}": "ै", "\\": "्",
-    "|": "्र", ";": "स", ":": "स्", "'": "ु", "\"": "ू", ",": ",", "<": "?", ".": "।", ">": "श्र", "/": "र",
-    "?": "रु", "=": ".", "ˆ": "फ्", "Î": "ङ्ख", "å": "द्व", "÷": "/"
+    "~": "ञ्",
+    "`": "ञ",
+    "!": "१",
+    "@": "२",
+    "#": "३",
+    "$": "४",
+    "%": "५",
+    "^": "६",
+    "&": "७",
+    "*": "८",
+    "(": "९",
+    ")": "०",
+    "-": "(",
+    "_": ")",
+    "+": "ं",
+    "[": "ृ",
+    "{": "र्",
+    "]": "े",
+    "}": "ै",
+    "\\": "्",
+    "|": "्र",
+    ";": "स",
+    ":": "स्",
+    "'": "ु",
+    "\"": "ू",
+    ",": ",",
+    "<": "?",
+    ".": "।",
+    ">": "श्र",
+    "/": "र",
+    "?": "रु",
+    "=": ".",
+    "ˆ": "फ्",
+    "Î": "ङ्ख",
+    "å": "द्व",
+    "÷": "/"
 }
 
 def normalizePreeti(preetitxt):
     normalized = ''
     previoussymbol = ''
-    preetitxt = preetitxt.replace('qm', 's|').replace('f]', 'ो').replace('km', 'फ').replace('0f', 'ण').replace('If', 'क्ष').replace('if', 'ष').replace('cf', 'आ')
+    preetitxt = preetitxt.replace('qm', 's|')
+    preetitxt = preetitxt.replace('f]', 'ो')
+    preetitxt = preetitxt.replace('km', 'फ')
+    preetitxt = preetitxt.replace('0f', 'ण')
+    preetitxt = preetitxt.replace('If', 'क्ष')
+    preetitxt = preetitxt.replace('if', 'ष')
+    preetitxt = preetitxt.replace('cf', 'आ')
     index = -1
     while index + 1 < len(preetitxt):
         index += 1
@@ -45,47 +84,40 @@ def normalizePreeti(preetitxt):
 def convert(preeti):
     converted = ''
     normalizedpreeti = normalizePreeti(preeti)
-    for character in normalizedpreeti:
+    for index, character in enumerate(normalizedpreeti):
         try:
-            if 97 <= ord(character) <= 122:
+            if ord(character) >= 97 and ord(character) <= 122:
                 converted += unicodeatoz[ord(character) - 97]
-            elif 65 <= ord(character) <= 90:
+            elif ord(character) >= 65 and ord(character) <= 90:
                 converted += unicodeAtoZ[ord(character) - 65]
-            elif 48 <= ord(character) <= 57:
+            elif ord(character) >= 48 and ord(character) <= 57:
                 converted += unicode0to9[ord(character) - 48]
             else:
                 converted += symbolsDict[character]
         except KeyError:
             converted += character
+
     return converted
 
 def extract_text_from_pdf(pdf_file):
     text = ''
-    pdf = pdfplumber.open(pdf_file)
-    for page in pdf.pages:
-        extracted_text = page.extract_text()
-        if extracted_text:
-            text += extracted_text
-    return handle_vertical_text(text)
-
-def handle_vertical_text(text):
-    lines = text.split('\n')
-    vertical_lines = []
-    horizontal_line = ''
-    for line in lines:
-        if len(line) == 1:  # Possible vertical arrangement (single character per line)
-            horizontal_line += line
-        else:
-            if horizontal_line:  # If we've built a horizontal line, add it.
-                vertical_lines.append(horizontal_line)
-                horizontal_line = ''
-            vertical_lines.append(line)  # Add the full line if it's not vertical.
-    if horizontal_line:
-        vertical_lines.append(horizontal_line)
-    return ' '.join(vertical_lines)
+    with open(pdf_file, 'rb') as file:
+        reader = PyPDF2.PdfReader(file)
+        for page in reader.pages:
+            text += page.extract_text()
+    return text
+
+def process_file(inputfile):
+    ext = os.path.splitext(inputfile)[1].lower()
+    if ext == '.pdf':
+        preeti = extract_text_from_pdf(inputfile)
+    else:
+        with open(inputfile, "r") as fp:
+            preeti = fp.read()
+    return convert(preeti)
 
 def main():
-    st.title("PDF/TXT to Unicode Converter (Nepali RAG)")
+    st.title("PDF/TXT to Unicode Converter")
 
     uploaded_file = st.file_uploader("Choose a PDF or TXT file", type=["pdf", "txt"])
 
@@ -93,7 +125,10 @@ def main():
         file_extension = os.path.splitext(uploaded_file.name)[1].lower()
 
         if file_extension == ".pdf":
-            text = extract_text_from_pdf(io.BytesIO(uploaded_file.read()))
+            pdf_reader = PyPDF2.PdfReader(io.BytesIO(uploaded_file.read()))
+            text = ""
+            for page in pdf_reader.pages:
+                text += page.extract_text()
         else:  # .txt file
             text = uploaded_file.getvalue().decode("utf-8")
 
@@ -113,8 +148,5 @@ def main():
             mime="text/plain"
         )
 
-    # Write footer
-    st.markdown("Made with ❤️ by Sumit Yadav(https://sumityadav.com.np)")
-
 if __name__ == "__main__":
     main()