Update app.py

app.py

@@ -76,9 +76,21 @@ def run_GOT(pdf_file):
 
             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()
+
+            # Encode the HTML content to base64
+            encoded_html = base64.b64encode(html_content.encode('utf-8')).decode('utf-8')
+            iframe_src = f"data:text/html;base64,{encoded_html}"
+            iframe = f'<iframe src="{iframe_src}" width="100%" height="600px"></iframe>'
+            download_link = f'<a href="data:text/html;base64,{encoded_html}" download="result_{unique_id}_page_{i+1}.html">Download Full Result</a>'
+
             results.append({
                 "page_number": i + 1,
-                "text": res  # Directly use the output from model.chat_crop
+                "text": res,  # Directly use the output from model.chat_crop
+                "html": iframe,
+                "download_link": download_link
             })
 
             if os.path.exists(image_path):