Update app.py

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-
 import json
 import os
 import requests
@@ -25,19 +24,34 @@ if 'server_started' not in st.session_state:
 if not st.session_state['server_started']:
   start_server()              
 
-st.title('Passport Recognition Demo')
+st.sidebar.title("Server Status")
+st.sidebar.write("🟢 Server is running" if st.session_state['server_started'] else "🔴 Server is not running")
+
+st.title('Passport Recognition with passportEYE, paddleocr and a LLM')
+
+upload_option = st.radio("Select Image Source", ("Upload Image", "Load Example Image"), index=0)
 
-image_path = st.file_uploader("Upload Passport Image", type=["jpg", "jpeg", "png"])
+if upload_option == "Load Example Image":
+    image_path = "sample.jpg"
+else:
+    image_path = st.file_uploader("Upload Passport Image", type=["jpg", "jpeg", "png"], key="image_uploader")
 
-if image_path is not None:
-    st.image(image_path, caption="Uploaded Image.", use_column_width=True)
-    st.write("")
-    st.write("Classifying...")
+if image_path is not None and st.button("Read Passport"):
 
-    with open("temp_image.jpg", "wb") as f:
-        f.write(image_path.read())
+    with st.spinner("Running..."):
+        if upload_option == "Upload Image":
+            with open("temp_image.jpg", "wb") as f:
+                f.write(image_path.read())
+        else:
+            passport_info = recognize_passport(image_path)
+            
+    if upload_option == "Upload Image":
+        passport_info = recognize_passport("temp_image.jpg")
 
-    passport_info = recognize_passport("temp_image.jpg")
+    col1, col2 = st.columns(2)
+    with col1:
+        st.markdown(f'## Passport Recognition Results')
+        st.json(json.dumps(passport_info, indent=2))
 
-    st.markdown(f'## Passport Recognition Results')
-    st.json(json.dumps(passport_info, indent=2))
+    with col2:
+        st.image(image_path, caption="Uploaded Image.", use_column_width=True)