Update app.py

app.py

@@ -54,14 +54,24 @@ def run_GOT(pdf_file):
             image_path = os.path.join(UPLOAD_FOLDER, f"{unique_id}_page_{i+1}.png")
             image.save(image_path)
             
-            res = model.chat_crop(tokenizer, image_path, ocr_type='ocr')
+            result_path = os.path.join(RESULTS_FOLDER, f"{unique_id}_page_{i+1}.html")
+            
+            res = model.chat_crop(tokenizer, image_path, ocr_type='format', render=True, save_render_file=result_path)
+            
+            # Read the rendered HTML content
+            with open(result_path, 'r') as f:
+                html_content = f.read()
+            
             results.append({
                 "page_number": i + 1,
-                "text": res
+                "text": res,
+                "html": html_content
             })
             
             if os.path.exists(image_path):
                 os.remove(image_path)
+            if os.path.exists(result_path):
+                os.remove(result_path)
     except Exception as e:
         return f"Error: {str(e)}", None
     finally: