ahmadtalha commited on
Commit
147203c
1 Parent(s): 60bb5b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -81,9 +81,20 @@ def process_input(image=None, file=None, audio=None, text=""):
81
  if file is not None:
82
  if file.name.lower().endswith(('.png', '.jpg', '.jpeg')):
83
  pil_image = Image.open(file)
84
- final_text += "\n" + ocr_processor.process_image(pil_image)
 
 
 
 
 
85
  elif file.name.lower().endswith('.pdf'):
86
- final_text += "\n" + ocr_processor.process_pdf(file.name)
 
 
 
 
 
 
87
  else:
88
  final_text += "\nUnsupported file type."
89
  if audio is not None:
 
81
  if file is not None:
82
  if file.name.lower().endswith(('.png', '.jpg', '.jpeg')):
83
  pil_image = Image.open(file)
84
+ ocr_prediction = ocr_processor.process_image(pil_image)
85
+ # gettig text from ocr object
86
+ for idx in range(len((list(ocr_prediction)[0][1]))):
87
+ final_text += " "
88
+ final_text += list((list(ocr_prediction)[0][1])[idx])[1][1]
89
+ print(final_text)
90
  elif file.name.lower().endswith('.pdf'):
91
+ ocr_prediction = ocr_processor.process_pdf(file.name)
92
+ # gettig text from ocr object
93
+ for idx in range(len((list(ocr_prediction)[0][1]))):
94
+ final_text += " "
95
+ final_text += list((list(ocr_prediction)[0][1])[idx])[1][1]
96
+ print(final_text)
97
+ # final_text += "\n" + ocr_processor.process_pdf(file.name)
98
  else:
99
  final_text += "\nUnsupported file type."
100
  if audio is not None: