Update app.py

app.py

@@ -37,7 +37,7 @@ def main():
     st.title("🗼 Lutece-Vision-Base Demo")
     st.markdown("Please keep in mind that inference might be slower since this Huggingface space is running on CPU only.")
 
-        # Sidebar with SujetAI watermark
+    # Sidebar with SujetAI watermark
     st.sidebar.image("sujetAI.svg", use_column_width=True)
     st.sidebar.markdown("---")
     st.sidebar.markdown("Sujet AI, a Paris-based AI startup, is on a noble mission to democratize investment opportunities by leveraging built-in models and cutting-edge technologies. Committed to open-sourcing its technology, Sujet AI aims to contribute to the research and development communities, ultimately serving the greater good of humanity.")
@@ -47,30 +47,56 @@ def main():
     # Load model and processor
     model, processor = load_model_and_processor()
 
-    # File uploader for document
-    uploaded_file = st.file_uploader("📄 Upload a financial document", type=["png", "jpg", "jpeg"])
+    # Two-column layout
+    col1, col2 = st.columns(2)
 
-    if uploaded_file is not None:
-        image = Image.open(uploaded_file).convert('RGB')
+    with col1:
+        st.subheader("📄 Financial Document")
+        # Option to use example image or upload new one
+        use_example = st.checkbox("Use example image", value=True)
         
-        # Two-column layout
-        col1, col2 = st.columns(2)
+        if use_example:
+            image = Image.open("test_image.png").convert('RGB')
+            st.image(image, caption="Example Document", use_column_width=True)
+        else:
+            uploaded_file = st.file_uploader("Upload a financial document", type=["png", "jpg", "jpeg"])
+            if uploaded_file is not None:
+                image = Image.open(uploaded_file).convert('RGB')
+                st.image(image, caption="Uploaded Document", use_column_width=True)
+            else:
+                image = None
+
+    with col2:
+        st.subheader("❓ Ask a Question")
+        # Predefined questions
+        example_questions = [
+            "What's the current expenses amount?",
+            "When was this document produced?",
+            "Who is this document addressed to?",
+            "What is the amount that's circled?",
+            "What's the project's identifier?"
+        ]
         
-        with col1:
-            # Display image with controlled size
-            st.image(image, caption="Uploaded Document", use_column_width=True)
+        selected_question = st.selectbox("Select a question or type your own:", 
+                                         [""] + example_questions,
+                                         index=0)
         
-        with col2:
-            # Question input
-            question = st.text_input("❓ Ask a question about the document", "")
-            submit_button = st.button("🔍 Generate Answer")
+        if selected_question:
+            question = selected_question
+        else:
+            question = st.text_input("Type your question here:")
 
-            # Answer section spanning both columns
-            if submit_button and question:
-                with st.spinner("Generating answer..."):
-                    answer = generate_answer(model, processor, image, question)
-                st.success(f"## 💡 {answer}")
+        submit_button = st.button("🔍 Generate Answer")
 
+        # Answer section
+        if submit_button and question and image is not None:
+            with st.spinner("Generating answer..."):
+                answer = generate_answer(model, processor, image, question)
+            st.success(f"## 💡 {answer}")
+        elif submit_button and image is None:
+            st.warning("Please upload an image or use the example image before asking a question.")
+        elif submit_button and not question:
+            st.warning("Please enter a question or select one from the examples.")
 
 if __name__ == "__main__":
     main()