Added rest of the forms - Need to implement batch processing for them

app.py

@@ -34,7 +34,7 @@ with gr.Blocks() as demo:
         
 
 
-        with gr.Tab("Child Fee Reimbursement Form"):
+        with gr.Tab("Reimbursement Forms"):
             with gr.Row():
                 with gr.Column(scale=2):
                     img_input = gr.Image(
@@ -48,7 +48,8 @@ with gr.Blocks() as demo:
 
                 with gr.Column(scale=2):
                     # Dropdown for form names
-                    gr.Markdown("## Child Fee Reimbursement")
+                    gr.Markdown("## Reimbursement Forms")
+                    form_name = gr.Dropdown(choices=["Child Fee Reimbursement Form", "Internet Charges Form", "Mobile Reimbursement Form"], interactive=True, multiselect=False)
                     
                     # 2x2 grid for info fields
                     with gr.Row():
@@ -61,9 +62,11 @@ with gr.Blocks() as demo:
                     preview_output = gr.File(label="Download Filled Form")
 
 
+
+
                     upload_btn.click(
                         fn=extract_child_fee_info,
-                        inputs=[img_input, emp_name, emp_code, department],
+                        inputs=[img_input, emp_name, emp_code, department,form_name],
                         outputs=preview_output
                     )
 

pipeline.py

@@ -43,7 +43,7 @@ reciept_system_prompt = (
 
 fee_bill_system_prompt = (
     "You are an expert at extracting data from fee bills. "
-    "Read the provided image of a child fee bill and return a JSON object that matches the following Pydantic model:\n"
+    "Read the provided image of a fee bill and return a JSON object that matches the following Pydantic model:\n"
     "from typing import List, Optional\n"
     "class FeeItem(BaseModel):\n"
     "    bill_date: Optional[str] = None  # Bill Date Field, leave null if not found\n"
@@ -145,7 +145,7 @@ def extract_info(pil_img):
         return f"```json\n{json.dumps({'error': str(e), 'raw_output': raw_output}, indent=2)}\n```"
 
 
-def extract_child_fee_info(img_input, emp_name, emp_code, department):
+def extract_child_fee_info(img_input, emp_name, emp_code, department, form_name):
     print(emp_name, emp_code, department)
     processed_image = preprocess_image(img_input)
     img_bytes = pil_to_bytes(processed_image)
@@ -178,12 +178,25 @@ def extract_child_fee_info(img_input, emp_name, emp_code, department):
         if items and "bill_month" in items[0]:
             bill_month = items[0]["bill_month"]
 
+
         os.makedirs("outputs", exist_ok=True)
-        output_pdf_path = f"outputs/filled_child_fee_form_{uuid.uuid4().hex}.pdf"
+
+        if form_name == "Child Fee Reimbursement Form":
+           
+            output_pdf_path = f"outputs/filled_child_fee_reimbursement_form_{uuid.uuid4().hex}.pdf"
+
+        elif form_name == "Internet Charges Form":
+
+            output_pdf_path = f"outputs/filled_internet_charges_reimbursement_form_{uuid.uuid4().hex}.pdf"
+
+        elif form_name == "Mobile Reimbursement Form":
+            
+            output_pdf_path = f"outputs/filled_mobile_reimbursement_form_{uuid.uuid4().hex}.pdf"
+        
 
 
         filled_pdf_path = fill_child_fee_pdf(
-            template_pdf_path="CHILD FEE REIMBURSEMENT FORM.pdf",
+            template_pdf_path="templates/REIMBURSEMENT FORM.pdf",
             output_pdf_path=output_pdf_path,
             emp_name=emp_name,
             emp_code=emp_code,
@@ -281,12 +294,10 @@ def extract_medical_info(pil_img, emp_name, emp_code, department, designation, c
     }});
 </script>
 </body>'''
-        # Ensure we are replacing the </body> tag correctly
         if "</body>" in html_content:
             html_content = html_content.replace("</body>", script_to_inject, 1)
         else:
-            # Fallback if no </body> tag, append at the end (less ideal)
-            html_content += script_to_inject.replace("</body>","") # remove the body tag we added if appending
+            html_content += script_to_inject.replace("</body>","") 
 
 
         output_dir = "outputs"

templates/medical_form.html

@@ -24,8 +24,8 @@
         .logo-placeholder {
             width: 90px;
             height: 90px;
-            background: #f2f2f2;
-            border: 1px solid #ddd;
+            background: white;
+            border: 1px solid white;
             margin-right: 18px;
         }
         .header-title {
@@ -162,7 +162,7 @@
     <div class="form-container">
         <div class="form-header">
             <div class="logo-placeholder">
-                <!-- Logo goes here -->
+                <img src="NetSol_logo.svg.png" alt="NetSol Logo" style="width: 100%; height: 100%; object-fit: contain;">
             </div>
             <div class="header-title">
                 <h2>Medical Reimbursement Form</h2>
@@ -225,7 +225,7 @@
             <table style="margin-top: 0;">
                 <tr class="total-row">
                     <td colspan="4" style="border: none;"></td>
-                    <td style="border-top: none; border-left: 1px solid #000;">Total -
+                    <td style="border-top: none; border-left: 0px solid #000;">Total -
                         <input style="width: 80px; border: none; border-bottom: 1px solid #000; font-size: 16px; background: none; text-align: center;" type="number" step="0.01" min="0" name="total" readonly id="total_amount">
                     </td>
                 </tr>