Update app.py

app.py

@@ -13,7 +13,7 @@ from models import ParsedDARReport, HarmonisedPara
 # Firebase setup
 FIREBASE_CREDENTIALS = os.environ.get("FIREBASE_CREDENTIALS")
 if FIREBASE_CREDENTIALS:
-    # Load credentials from environment variable (preferred)
+    # Load credentials from environment variable (preferred for security)
     cred = credentials.Certificate(json.loads(FIREBASE_CREDENTIALS))
 else:
     # Fallback to reading from firebase.json file
@@ -25,6 +25,14 @@ firebase_admin.initialize_app(cred)
 db = firestore.client()
 request_counts = db.collection('request_counts')
 
+def get_request_count():
+    """Retrieve the current request count for today."""
+    today = datetime.utcnow().strftime('%Y-%m-%d')
+    doc_ref = request_counts.document(today)
+    doc = doc_ref.get()
+    count = doc.to_dict().get('count', 0) if doc.exists else 0
+    return count
+
 def check_request_limit():
     """Check if the request limit for the day has been reached."""
     today = datetime.utcnow().strftime('%Y-%m-%d')
@@ -77,33 +85,33 @@ def process_dar_pdf(pdf_file):
     # Check request limit before processing
     can_process, error_msg = check_request_limit()
     if not can_process:
-        return error_msg, None, None
+        return error_msg, None, None, f"Requests today: {get_request_count()}/400"
 
     gemini_api_key = os.environ.get("GEMINI_API_KEY")
     if not pdf_file:
-        return "Please upload a PDF file.", None, None
+        return "Please upload a PDF file.", None, None, f"Requests today: {get_request_count()}/400"
     if not gemini_api_key:
-        return "Error: GEMINI_API_KEY secret not found in Space settings.", None, None
+        return "Error: GEMINI_API_KEY secret not found in Space settings.", None, None, f"Requests today: {get_request_count()}/400"
 
     # Step 1: Process PDF to text
     full_text = preprocess_pdf_text(pdf_file.name)
     if full_text.startswith("Error"):
-        return f"Failed to process PDF: {full_text}", None, None
+        return f"Failed to process PDF: {full_text}", None, None, f"Requests today: {get_request_count()}/400"
 
     # Step 2: Extract structured data
     parsed_report = get_structured_data_with_gemini(gemini_api_key, full_text)
     if parsed_report.parsing_errors or not parsed_report.audit_paras:
         error_msg = parsed_report.parsing_errors or "Could not find any audit paras."
-        return error_msg, None, None
+        return error_msg, None, None, f"Requests today: {get_request_count()}/400"
 
     # Step 3: Get harmonised titles
     original_headings = [p.audit_para_heading for p in parsed_report.audit_paras if p.audit_para_heading]
     if not original_headings:
-        return "Found paras but no headings to harmonise.", None, None
+        return "Found paras but no headings to harmonise.", None, None, f"Requests today: {get_request_count()}/400"
 
     harmonised_results = get_harmonised_titles(gemini_api_key, full_text, original_headings)
     if not harmonised_results:
-        return "Failed to generate harmonised titles.", None, None
+        return "Failed to generate harmonised titles.", None, None, f"Requests today: {get_request_count()}/400"
 
     # Step 4: Combine and prepare outputs
     harmonised_map = {item.original_heading: item.harmonised_heading for item in harmonised_results}
@@ -133,14 +141,14 @@ def process_dar_pdf(pdf_file):
     with open(excel_file_name, "wb") as f:
         f.write(output_excel.getbuffer())
 
-    return "Processing complete.", html_output, gr.File(value=excel_file_name)
+    return "Processing complete.", html_output, gr.File(value=excel_file_name), f"Requests today: {get_request_count()}/400"
 
 # --- Gradio Interface Definition ---
 with gr.Blocks(theme=gr.themes.Soft(), title="DAR Harmonisation Tool") as demo:
 
     # --- Login UI (visible initially) ---
     with gr.Column(visible=True) as login_ui:
-        gr.Markdown("# DAR Harmonisation Tool Login")
+        gr.Markdown("# Audit Officer Login")
         gr.Markdown("Please enter the credentials to access the tool.")
         with gr.Row():
             username_input = gr.Textbox(label="Username", placeholder="Enter your username")
@@ -153,8 +161,9 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DAR Harmonisation Tool") as demo:
         gr.Markdown("# DAR Draft Audit Report Harmonisation Tool")
         gr.Markdown("## Initiative by Mumbai Audit 1 Commissionerate")
         gr.Markdown(
-            "Upload a Departmental Audit Report (DAR) in PDF format. The tool will process it and generate harmonised titles for Audit paras in accordance with GST law."
+            "Upload a Observation letter to Taxpayer or  Departmental Audit Report (DAR) in PDF format. The tool will process it and generate harmonised titles for Audit paras in accordance with GST law."
         )
+        request_count_output = gr.Textbox(label="Requests Made Today", interactive=False, value="Requests today: 0/400")
         with gr.Row():
             with gr.Column(scale=1):
                 pdf_input = gr.File(label="Upload DAR PDF", file_types=[".pdf"])
@@ -168,7 +177,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DAR Harmonisation Tool") as demo:
         submit_btn.click(
             fn=process_dar_pdf,
             inputs=[pdf_input],
-            outputs=[status_output, html_output, excel_output]
+            outputs=[status_output, html_output, excel_output, request_count_output]
         )
 
     # --- Login Functionality ---
@@ -183,26 +192,31 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DAR Harmonisation Tool") as demo:
         is_valid_user = (username == auth_username and password == auth_password)
 
         if is_valid_user:
+            # Login successful: hide login UI, show main app, display request count
+            request_count = get_request_count()
             return {
                 login_ui: gr.update(visible=False),
                 main_app_ui: gr.update(visible=True),
-                login_error_msg: gr.update(visible=False)
+                login_error_msg: gr.update(visible=False),
+                request_count_output: gr.update(value=f"Requests today: {request_count}/400")
             }
         else:
+            # Login failed: keep login UI visible, show error message
             return {
                 login_ui: gr.update(visible=True),
                 main_app_ui: gr.update(visible=False),
-                login_error_msg: gr.update(value="<p style='color:red;'>Invalid username or password.</p>", visible=True)
+                login_error_msg: gr.update(value="<p style='color:red;'>Invalid username or password.</p>", visible=True),
+                request_count_output: gr.update(value="Requests today: 0/400")
             }
 
     login_button.click(
         login,
         inputs=[username_input, password_input],
-        outputs=[login_ui, main_app_ui, login_error_msg]
+        outputs=[login_ui, main_app_ui, login_error_msg, request_count_output]
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)# import gradio as gr
+    demo.launch(debug=True)
 # import pandas as pd
 # from io import BytesIO
 # import os