Spaces:
Sleeping
Sleeping
| # app.py | |
| import streamlit as st | |
| import base64 | |
| from annotations import analyze_pdf | |
| def display_pdf(pdf_bytes): | |
| """Displays the PDF in the browser using an embed tag.""" | |
| if pdf_bytes and len(pdf_bytes) > 0: | |
| # Encode the PDF to base64 | |
| base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8') | |
| pdf_display = f""" | |
| <embed src="data:application/pdf;base64,{base64_pdf}" width="100%" height="800px" type="application/pdf"> | |
| """ | |
| st.components.v1.html(pdf_display, height=800, width=700, scrolling=True) | |
| else: | |
| st.info("No annotated PDF to display.") | |
| def main(): | |
| st.set_page_config( | |
| page_title="PDF Analyzer", | |
| page_icon="📄", | |
| layout="wide", | |
| ) | |
| st.title("📄 PDF Analyzer") | |
| st.markdown(""" | |
| Upload a PDF to analyze its language, highlight errors, and view detailed error reports. | |
| """) | |
| uploaded_file = st.file_uploader("Upload your PDF file", type=["pdf"]) | |
| if uploaded_file is not None: | |
| with st.spinner("Analyzing PDF..."): | |
| language_results, annotated_pdf = analyze_pdf(uploaded_file) | |
| if "error" in language_results: | |
| st.error("An error occurred during analysis:") | |
| st.code(language_results["error"]) | |
| else: | |
| st.success("Analysis complete!") | |
| # Debugging: Show the size of annotated_pdf | |
| if annotated_pdf and len(annotated_pdf) > 0: | |
| st.write(f"Annotated PDF size: {len(annotated_pdf)} bytes") | |
| else: | |
| st.write("Annotated PDF is empty.") | |
| # Display the annotated PDF | |
| st.subheader("📄 Annotated PDF") | |
| display_pdf(annotated_pdf) | |
| # Sidebar for error details | |
| st.sidebar.header("📝 Error Details") | |
| if language_results.get("total_issues", 0) > 0: | |
| for idx, issue in enumerate(language_results["issues"], 1): | |
| with st.sidebar.expander(f"Issue {idx}"): | |
| st.markdown(f"**Message:** {issue['message']}") | |
| st.markdown(f"**Category:** {issue['category']}") | |
| st.markdown(f"**Suggestions:** {', '.join(issue['suggestions']) if issue['suggestions'] else 'No suggestions'}") | |
| st.markdown(f"**Sentence:** {issue['context']}") | |
| else: | |
| st.sidebar.success("No language issues found!") | |
| # Option to download the annotated PDF | |
| if annotated_pdf and len(annotated_pdf) > 0: | |
| st.download_button( | |
| label="📥 Download Annotated PDF", | |
| data=annotated_pdf, | |
| file_name="annotated.pdf", | |
| mime="application/pdf", | |
| ) | |
| else: | |
| st.info("No annotated PDF available for download.") | |
| if __name__ == "__main__": | |
| main() |