Update app.py

app.py

@@ -7,35 +7,24 @@ import fitz  # PyMuPDF
 import time
 import os
 
-# Configuration
-GEMINI_API_KEY = st.secrets["GEMINI_API_KEY"]
-GEMINI_MODEL = "gemini-2.0-flash"  # Verify with Gemini API documentation
+# Document categories
 DOCUMENT_TYPES = ["Land Records", "Caste Certificates", "Property Registrations"]
 
-# Define a class to mimic Streamlit's UploadedFile object for the example image
-class ExampleFile:
-    def __init__(self, file_path):
-        self.name = os.path.basename(file_path)
-        self.type = "image/jpeg"
-        with open(file_path, "rb") as f:
-            self._buffer = BytesIO(f.read())
-    
-    def getvalue(self):
-        return self._buffer.getvalue()
-
-# Initialize session state
+# Set up session state
 def initialize_session_state():
     if "processed_doc" not in st.session_state:
         st.session_state.processed_doc = None
     if "current_file" not in st.session_state:
         st.session_state.current_file = None
+    if "gemini_api_key" not in st.session_state:
+        st.session_state.gemini_api_key = None
 
 # Reset session state
 def reset_session_state():
     st.session_state.processed_doc = None
     st.session_state.current_file = None
 
-# Encode file to base64 for Gemini API
+# Encode file to base64
 def encode_file(file):
     try:
         file_content = file.getvalue()
@@ -46,23 +35,27 @@ def encode_file(file):
 
 # Query Gemini API
 def query_gemini(prompt, image_b64):
+    if not st.session_state.gemini_api_key:
+        st.error("API Key is missing. Please enter your Gemini API Key.")
+        return None
+    
     try:
         headers = {
-            "Authorization": f"Bearer {GEMINI_API_KEY}",
+            "Authorization": f"Bearer {st.session_state.gemini_api_key}",
             "Content-Type": "application/json"
         }
         payload = {
-            "model": GEMINI_MODEL,
+            "model": "gemini-1.5-pro",  # Change based on the latest API model
             "prompt": prompt,
             "image": image_b64
         }
         response = requests.post(
-            "https://api.gemini.com/v1/analyze",  # Replace with actual Gemini API endpoint
+            "https://generativelanguage.googleapis.com/v1/models/gemini-1.5-pro:generateContent",  # Corrected URL
             headers=headers,
             json=payload
         )
         response.raise_for_status()
-        return response.json().get("result", "")
+        return response.json().get("result", "No response received.")
     except Exception as e:
         st.error(f"Gemini API error: {str(e)}")
         return None
@@ -71,7 +64,6 @@ def query_gemini(prompt, image_b64):
 def process_document(file):
     try:
         with st.spinner("Analyzing document..."):
-            # Encode file to base64
             image_b64 = encode_file(file)
             if not image_b64:
                 return
@@ -84,12 +76,6 @@ def process_document(file):
                 st.session_state.doc_preview = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
             elif file.type.startswith('image/'):
                 st.session_state.doc_preview = Image.open(BytesIO(file.getvalue()))
-            elif file.type == "text/plain":
-                text = file.getvalue().decode('utf-8')
-                img = Image.new('RGB', (800, 600), color=(73, 109, 137))
-                d = ImageDraw.Draw(img)
-                d.text((10, 10), text, fill=(255, 255, 0))
-                st.session_state.doc_preview = img
             else:
                 st.error("Unsupported file format")
                 return
@@ -109,7 +95,7 @@ def process_document(file):
             details = query_gemini(extract_prompt, image_b64)
 
             # Verify authenticity
-            verify_prompt = "Analyze this document for signs of tampering or forgery. Check for: inconsistent fonts, alignment issues, suspicious modifications. Provide verification conclusion."
+            verify_prompt = "Analyze this document for signs of tampering or forgery. Provide verification results."
             verification = query_gemini(verify_prompt, image_b64)
 
             st.session_state.processed_doc = {
@@ -134,24 +120,20 @@ def main():
     # Sidebar Controls
     with st.sidebar:
         st.header("Document Controls")
-        
+
+        # API Key input
+        st.session_state.gemini_api_key = st.text_input("Enter Gemini API Key", type="password")
+
         uploaded_file = st.file_uploader(
             "Upload Document",
-            type=["pdf", "jpg", "jpeg", "png", "txt"],
+            type=["pdf", "jpg", "jpeg", "png"],
             key="uploaded_file"
         )
         
-        if st.button("Use Example Image"):
-            example_file_path = "Caste-Certificate.jpg"
-            st.session_state.current_file = ExampleFile(example_file_path)
-        elif uploaded_file:
-            st.session_state.current_file = uploaded_file
-        else:
-            st.session_state.current_file = None
-        
         if st.button("Process Document"):
-            if st.session_state.current_file:
-                process_document(st.session_state.current_file)
+            if uploaded_file:
+                st.session_state.current_file = uploaded_file
+                process_document(uploaded_file)
             else:
                 st.error("Please select a document to process.")
         
@@ -181,7 +163,7 @@ def main():
             st.subheader("Verification Analysis")
             st.markdown(st.session_state.processed_doc['verification'])
     else:
-        st.info("Please select a document and click 'Process Document' to begin verification analysis.")
+        st.info("Please upload a document and click 'Process Document' to start verification.")
 
 if __name__ == "__main__":
-    main()
+    main()