Added default FIR copy feature

app.py

@@ -31,28 +31,46 @@ def input_image_setup(uploaded_file):
         ]
         return image_parts
     else:
-        raise FileNotFoundError("No file uploaded")
+        return None  # Return None if no file uploaded
 
-##initialize our streamlit app
+# initialize our streamlit app
 
-st.set_page_config(page_title="Gemini Image Demo")
+st.set_page_config(page_title="Crime Lens API")
 
 st.header("Criminal Case Filing System OCR")
 
-uploaded_file = st.file_uploader("Choose an image...", type=["pdf", "jpg", "jpeg", "png"])
+# Add option to use default FIR copy
+use_default_copy = st.checkbox("Use Default FIR Copy")
+
+uploaded_file = None
 image = ""
-if uploaded_file is not None:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image.", use_column_width=True)
+if not use_default_copy:
+    uploaded_file = st.file_uploader("Choose an image...", type=["pdf", "jpg", "jpeg", "png"])
+
+if use_default_copy or uploaded_file is not None:
+    if use_default_copy:
+
+        image_path = "fir-copy.jpg"
+        image = Image.open(image_path)
+        st.image(image, caption="Using Default FIR Copy", use_column_width=True)
+    else:
+        # User uploaded their own file
+        image = Image.open(uploaded_file)
+        st.image(image, caption="Uploaded Image.", use_column_width=True)
 
 submit = st.button("Extract data")
 
 ## If ask button is clicked
 if submit:
     input_prompt = """
-                      Extract the data of FIR No, Date and Time of FIR, Occurrence of offence, Type of information, Place of occurrence, Complaint / Victim's name
-                  """
-    image_data = input_image_setup(uploaded_file)
+            Extract the data of FIR No, Date and Time of FIR, Occurrence of offence, Type of information, Place of occurrence, Complaint / Victim's name
+        """
+    image_data = input_image_setup(uploaded_file)  # Use uploaded file if available, otherwise None
+
+    if image_data is None:  # Check if user selected default copy
+        # Use default image path if no file uploaded
+        image_data = [{'mime_type': 'image/jpeg', 'data': open('fir-copy.jpg', 'rb').read()}]
+
     response = get_gemini_response(input_prompt, image_data, input_prompt)
     st.subheader("The Response is")
     print("Respons")
@@ -89,16 +107,3 @@ if submit:
         file_name="field_data.json",
         mime="application/json"
     )
-
-     # Assuming your privacy ops portal URL is localhost:3000
-    portal_url = "http://localhost:3000/new-case?"
-    print(portal_url)  # Print portal URL for debugging
-
-    cleaned_field_data = {key.replace(' ', ''): value.replace(' ', '_') for key, value in field_data.items()}
-
-    # Construct the URL with cleaned_field_data
-    url_params = "&".join([f"{key}={value}" for key, value in cleaned_field_data.items()])
-    full_url = portal_url + url_params
-    st.write("Click on the below link to open the PrivacyOps Portal with the extracted data:")
-    st.write(full_url)
-