Update app.py

app.py

@@ -59,20 +59,9 @@ def input_file_setup(uploaded_file):
 st.set_page_config(page_title="Invoice Information Extractor", page_icon="🔮")
 st.title("Invoice Information Extractor 📃")
 
-st.write("Upload your invoice (image or PDF) and ask a question related to the invoice. Here are some example prompts:")
+st.write("Upload your invoice (image or PDF)")
 
 # Display sample prompts
-st.subheader("Sample Prompts:")
-sample_prompts = [
-    "What is the total amount?",
-    'What are the Customter details?' , 
-    'What is the information about the Products?'
-]
-sp.pills(sample_prompts,
-         options=Iterable[sample_prompts],
-         clearable=True,
-            index=None
-        )
 
 uploaded_file = st.file_uploader("Upload an Invoice Image 🖼️ or PDF 📄", type=["jpg", "jpeg", "png", "pdf"])
 input_query = st.text_input("Ask a question about the invoice", key="input")
@@ -92,6 +81,13 @@ You are an expert in understanding invoices. We will show you an invoice and you
 
 if submit:
     file_data = input_file_setup(uploaded_file)
+    selected = sp.pills(" Ask a question related to the invoice. Here are some example prompts:", 
+                        ["What is the total amount?",
+                         'What are the Customter details?' , 
+                         'What is the information about the Products?'],
+                        clearable=True,index=None)
+    if selected:
+        input_query = selected
     response = get_gemini_response(input_prompt, file_data, input_query)
     st.subheader("Response🤖:")
     st.write(response)