import streamlit as st import asyncio from pathlib import Path from main import red_flag_analyzer import fitz # ============================================================================ # DATA FLOW: # 1. User uploads .txt or .doc file via Streamlit UI # 2. File is read and text is extracted # 3. Text is passed to red_flag_analyzer() from main.py # 4. red_flag_analyzer() sends text to AI model # 5. AI analyzes and returns RedFlagReport with identified red flags # 6. Results are displayed in Streamlit UI with severity color-coding # ============================================================================ if "show_disclaimer" not in st.session_state: st.session_state.show_disclaimer = None # ============================================================================ # PAGE CONFIGURATION # ============================================================================ st.set_page_config( page_title="Red Flag Analyzer", page_icon="🚩", layout="wide", initial_sidebar_state="expanded" ) st.title("🚩 Red Flag Analyzer") st.markdown("Upload a document to identify potential risks and unfair clauses") # ============================================================================ # SIDEBAR - FILE UPLOAD # ============================================================================ st.sidebar.header("📄 Upload Document") st.sidebar.markdown("Supported formats: `.txt`, `.docx`, `.pdf`") # File uploader widget uploaded_file = st.sidebar.file_uploader( "Choose a file", type=["txt", "docx", "pdf"], help="Upload your document for analysis" ) # ============================================================================ # FUNCTION: Extract text from uploaded file # ============================================================================ def extract_text_from_file(uploaded_file): """ Extract text from uploaded file based on file type. Args: uploaded_file: Streamlit UploadedFile object Returns: str: Extracted text from the file """ file_extension = Path(uploaded_file.name).suffix.lower() #gets the extension of the file try: #checking which extension does it belong if file_extension == ".txt": # For .txt files, decode directly text = uploaded_file.getvalue().decode("utf-8") return text if file_extension == ".docx": from docx import Document from io import BytesIO # Extract text from document doc = Document(BytesIO(uploaded_file.getbuffer())) text = "\n".join([para.text for para in doc.paragraphs]) return text if file_extension == ".pdf": # Ensure the stream buffer is at the beginning uploaded_file.seek(0) # Extract text from PDF using PyMuPDF (fitz) pdf_document = fitz.open(stream=uploaded_file.read(), filetype="pdf") text = "" # 'page' is already the loaded page object, no need to re-load it for page in pdf_document: text += page.get_text() pdf_document.close() # Clean up memory handles return text except Exception as e: st.error(f"Error reading file: {str(e)}") return None # ============================================================================ # MAIN APP LOGIC # ============================================================================ if uploaded_file is not None: # Step 1: Extract text from file st.sidebar.success(f"✅ File uploaded: {uploaded_file.name}") with st.spinner("📖 Reading document..."): document_text = extract_text_from_file(uploaded_file) if document_text: # Step 2: Display document preview st.subheader("📋 Document Preview") with st.expander("View full document", expanded=False): st.text_area( "Document Content", value=document_text, height=200, disabled=True ) st.markdown(f"**Document Size:** {len(document_text)} characters") st.markdown("---") if st.button("🔍 Analyze for Red Flags", type="primary"): st.session_state.show_disclaimer=True if st.session_state.show_disclaimer: if st.session_state.show_disclaimer==True: st.warning( "This tool is designed to assist you in identifying potential " "risks and red flags in documents. However, it should not be considered legal advice. " "**Please consult with a qualified legal professional before making any final decisions.** " "This analysis is for awareness purposes only.", icon="⚠ī¸") if st.checkbox("I understand and wish to proceed with the analysis"): st.session_state.show_disclaimer=False with st.spinner("🤖 AI is analyzing your document..."): try: # Call the red_flag_analyzer function result = red_flag_analyzer(document_text) #from main.py # Step 4: Display results st.success("✅ Analysis complete!") st.markdown("---") # Display document summary st.subheader("📝 Document Summary") st.info(result.document_summary) # Display red flags st.subheader(f"🚩 Red Flags Found: {len(result.red_flags)}") #if result returns any red flag if result.red_flags: # Color mapping for severity levels severity_colors = { "High": "🔴", "Medium": "🟡", "Low": "đŸŸĸ" } # Display each red flag in a card-like format for idx, flag in enumerate(result.red_flags, 1): # Create a container for each red flag with st.container(border=True): col1, col2 = st.columns([3, 1]) with col1: st.markdown( f"### {severity_colors.get(flag.severity, '❓')} " f"Red Flag #{idx}" ) with col2: st.markdown( f"**Severity:** `{flag.severity}`" ) st.markdown("**Problem Text:**") st.code(flag.item, language="text") st.markdown("**Why it's a problem:**") st.write(flag.reason) st.markdown("---") else: st.success("✨ No red flags found! This document looks good.") except Exception as e: st.error(f"❌ Error during analysis: {str(e)}") st.info("Make sure your `.env` file is configured with the API key.") else: # Show empty state when no file is uploaded st.info("👈 Upload a document in the sidebar to get started!") # Display example with st.expander("ℹī¸ Example - What kind of red flags will be detected?"): st.markdown(""" The analyzer looks for: - **Unfair clauses** (one-sided terms, hidden conditions) - **Hidden fees** (unexpected costs, surprise charges) - **High-risk commitments** (unlimited liability, perpetual obligations) - **Ambiguous rules** (vague language, unclear definitions) - **Legal risks** (arbitration clauses, waived rights) """) # ============================================================================ # FOOTER # ============================================================================ st.markdown("---") st.markdown( "💡 **Tip:** Use this tool to review contracts, terms of service, and agreements. " "Always consult with legal professionals for final decisions." )