Update app.py

app.py

@@ -671,19 +671,43 @@ def process_invoice(pdf_file):
 
     return "\n".join(output)
 
-def gradio_interface(pdf_file):
+def gradio_interface(pdf_file, session_state=None):
     """Gradio interface to process uploaded PDF and display structured results."""
+    if session_state is None:
+        session_state = gr.State({"logged_in": True})
+
+    if not session_state["logged_in"]:
+        return "You have been logged out. Please log in again.", session_state
+
     if pdf_file is None:
-        return "Please upload a PDF file."
+        return "Please upload a PDF file.", session_state
+
     result = process_invoice(pdf_file)
-    return result
+    return result, session_state
+
+def logout(session_state):
+    """Handle logout by updating session state and returning login page message."""
+    session_state["logged_in"] = False
+    return "You have been logged out. Please log in again.", session_state
 
 with gr.Blocks(css=".prose a[href*='share']:has(svg) {display:none !important;}") as iface:
+    session_state = gr.State({"logged_in": True})
     gr.Markdown("# Invoice Fraud Detection")
     with gr.Row():
         file_input = gr.File(label="Upload Invoice PDF")
     result_output = gr.Markdown(label="Fraud Detection Results")
-    file_input.change(fn=gradio_interface, inputs=file_input, outputs=result_output)
+    logout_button = gr.Button("Log Out")
     
+    file_input.change(
+        fn=gradio_interface,
+        inputs=[file_input, session_state],
+        outputs=[result_output, session_state]
+    )
+    logout_button.click(
+        fn=logout,
+        inputs=session_state,
+        outputs=[result_output, session_state]
+    )
+
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()