`Update radio button options in PDF processor interface`

app.py

@@ -28,7 +28,7 @@ demo = gr.Interface(
     fn=process_pdf,
     inputs=[
         gr.File(label="Upload PDF"),  # File upload input
-        gr.Radio(["Option 1", "Option 2"], label="Choose an option")  # Radio buttons for options
+        gr.Radio(["Noc_timesheet_resdiential","Noc_timesheet_rotational", "Noc_invoice"], label="Choose an option")  # Radio buttons for options
     ],
     outputs="text",  # Text output
     title="PDF Processor",

dataSchema.py

@@ -0,0 +1,201 @@
+from pydantic import BaseModel, Field
+from typing import Optional,List
+
+class Noc_Residential_TimeSheetInformation(BaseModel):
+    """Details of a timesheet entry."""
+    full_name: str = Field(..., description="Full name of the person.")
+    position_title: str = Field(..., description="Position title of the person.")
+    work_location: str = Field(..., description="Work location of the person.")
+    contractor: str = Field(..., description="Contractor's name.")
+    noc_id: str = Field(..., description="NOC ID of the person.")
+    month_year: str = Field(..., description="Month and year in MM/YYYY format.")
+    
+    service_days_onshore: int = Field(0, description="Number of service days onshore.")
+    standby_days_onshore: int = Field(0, description="Number of standby days onshore in Doha.")
+    service_days_offshore: int = Field(0, description="Number of service days offshore.")
+    standby_extended_hitch_days_offshore: int = Field(0, description="Number of standby and extended hitch days offshore.")
+    extended_hitch_days_onshore: int = Field(0, description="Number of extended hitch days onshore for rotational personnel.")
+    service_days_weekend_public_holiday: int = Field(0, description="Number of service days during weekends or public holidays.")
+    per_diem_days: int = Field(0, description="Number of Per Diem days for onshore/offshore rotational personnel.")
+    training_days: int = Field(0, description="Number of training days.")
+    travel_days: int = Field(0, description="Number of travel days.")
+
+class Noc_Rotational_TimeSheetInformation(BaseModel):
+    """Details of a timesheet entry."""
+    full_name: str = Field(..., description="Full name of the person.")
+    position_title: str = Field(..., description="Position title of the person.")
+    work_location: str = Field(..., description="Work location of the person.")
+    contractor: str = Field(..., description="Contractor's name.")
+    PO_number:  Optional[str] = Field(None,description="Purchase order")
+    noc_id: str = Field(..., description="NOC ID of the person.")
+    month_year: str = Field(..., description="Month and year in MM/YYYY format.")
+    
+    service_days_onshore: int = Field(0, description="Number of service days onshore.")
+    standby_days_onshore: int = Field(0, description="Number of standby days onshore in Doha.")
+    service_days_offshore: int = Field(0, description="Number of service days offshore.")
+    service_days_weekend_public_holiday: int = Field(0, description="Number of service days during weekends or public holidays.")
+    standby_extended_hitch_days_offshore: int = Field(0, description="Number of standby and extended hitch days offshore.")
+    extended_hitch_days_onshore: int = Field(0, description="Number of extended hitch days onshore for rotational personnel.")
+    overtime_Hourly_Rate_ONSHORE:int = Field(0,description="number of over time hours onshore (Over 8 hours) ")
+    overtime_Hourly_Rate_OFFSHORE:int = Field(0,description="number of over time hours offshore (Over 12 hours) ")
+    per_diem_days: int = Field(0, description="Number of Per Diem days for onshore/offshore rotational personnel.")
+    training_days: int = Field(0, description="Number of training days.")
+    travel_days: int = Field(0, description="Number of travel days.")   
+
+class Noc_Item_Information(BaseModel):
+    """Details of each item in the document."""
+    item_number : int = Field(...,description="the number of the item")
+    service_description_code: str = Field(..., description="Service description code for the item.")
+    completion_date: str = Field(..., description="Completion date of the service in DD/MM/YYYY format.")
+    quantity: float = Field(..., description="Quantity of the service or item provided.")
+    unit: str = Field(..., description="Unit of measurement, e.g., Days.")
+    unit_price: float = Field(..., description="Unit price.")
+    net_amount: float = Field(..., description="Total net amount for this item, calculated as quantity * unit price.")                             
+    service_details: str = Field(..., description="Detailed description of the service provided.")
+
+class Noc_Document_Information(BaseModel):
+    """Details of the entire document."""
+    position_title: str = Field(..., description="Position title of the person.")
+    location: Optional[str] = Field(None, description="Location where the service is rendered.")
+    mobilization_date: str = Field(..., description="Mobilization date in DD/MM/YYYY format.")
+    end_date: str = Field(..., description="End date of the contract in DD/MM/YYYY format.")
+    notice_period: Optional[str] = Field(None, description="Notice period for resignation, if applicable")
+    items: List[Noc_Item_Information] = Field(..., description="List of items or services provided in the document.")
+    
+class Noc_items(BaseModel):
+    items: List[Noc_Item_Information] = Field(None, description="List of items or services provided in the document.")
+    
+class Noc_total(BaseModel):
+    """totals of the invoice"""
+    total_amount_wo_taxes : float = Field(...,description="total amount without taxes")
+    total_net_amount_of_order: float = Field(..., description="total net amount of order.")
+    total_amount_of_order: float = Field(..., description="total amount of order.")
+
+class Noc_PurchaseOrderInformation(BaseModel):
+    """Details of a purchase order entry."""
+    purchase_order_number: str = Field(..., description="The unique identifier for the purchase order.")
+    date: str = Field(..., description="Date of the purchase order in DD/MM/YYYY format.")
+    company_name: str = Field(..., description="Name of the company issuing the purchase order.")
+    address: str = Field(..., description="Address of the company issuing the purchase order.")
+    tel: Optional[str] = Field(None, description="Telephone number of the company.")
+    email: Optional[str] = Field(None, description="Email address of the company.")
+    final_shipping_address: Optional[str] = Field(None, description="Final shipping address for the order.")
+    buyer_contact_name: str = Field(..., description="Full name of the buyer contact.")
+    buyer_contact_company: str = Field(..., description="Company name of the buyer contact.")
+    buyer_contact_tel: Optional[str] = Field(None, description="Telephone number of the buyer contact.")
+    buyer_contact_email: Optional[str] = Field(None, description="Email address of the buyer contact.")
+    our_reference: Optional[str] = Field(None, description="under Our reference title.")
+    your_reference: Optional[str] = Field(None, description="under Your reference title.")
+    incoterms: Optional[str] = Field(None, description="Incoterms applicable to the order.")
+    total_value_of_order: str = Field(..., description="Total value of the purchase order.")
+    signature_released_by: str = Field(None, description="Name of the person who released the purchase order.")
+    signature_date: Optional[str] = Field(None, description="Date the order was signed.")
+
+class Noc_Clauses(BaseModel):
+    Clauses: str = Field(..., description="the contract clauses.")
+
+
+Noc_Res_timesheet_prompt = """
+   Based on the provided timesheet details, extract the following information:
+   - Full name of the person
+   - Position title of the person
+   - Work location
+   - Contractor's name
+   - NOC ID
+   - Month and year (in MM/YYYY format)
+   And from the bottom table : 
+   - Number of service days onshore
+   - Number of standby days onshore in Doha
+   - Number of service days offshore
+   - Number of service days during weekends or public holidays
+   - Number of standby and extended hitch days offshore
+   - Number of extended hitch days onshore for rotational personnel
+   - Number of Per Diem days for onshore/offshore rotational personnel
+   - Number of training days
+   - Number of travel days
+"""
+
+Noc_Rot_timesheet_prompt = """
+   Based on the provided timesheet details, extract the following information:
+   - Full name of the person
+   - Position title of the person
+   - Work location
+   - Contractor's name
+   - PO number which is the Purchase order
+   - NOC ID
+   - Month and year (in MM/YYYY format)
+   And from the bottom table : 
+   - Number of service days onshore
+   - Number of standby days onshore in Doha
+   - Number of service days offshore
+   - Number of service days during weekends or public holidays
+   - Number of standby and extended hitch days offshore
+   - Number of extended hitch days onshore for rotational personnel
+   - ONSHORE Overtime Hourly Rate (Over 8 hours)
+   - OFFSHORE Overtime Hourly Rate (Over 12 hours)
+   - Number of Per Diem days for onshore/offshore rotational personnel
+   - Number of training days
+   - Number of travel days
+"""
+
+invoice_first_page_prompt="""
+Extract the following details from the provided purchase order document:
+- Purchase Order Number: The unique identifier for the purchase order.
+- Date: The date the purchase order was issued (format: DD/MM/YYYY).
+- Company Name: The name of the company issuing the purchase order.
+- Address: The address of the company issuing the purchase order.
+- Telephone Number: The company's telephone number (if provided).
+- Email: The company's email address (if provided).
+- Final Shipping Address: The destination shipping address (if specified).
+- Buyer Contact Name: The full name of the buyer's contact person.
+- Buyer Contact Company: The company name of the buyer contact.
+- Buyer Contact Telephone Number: The buyer contact's telephone number (if provided).
+- Buyer Contact Email: The buyer contact's email address (if provided).
+- Our Reference: Reference specified under the "Our Reference" section (if present).
+- Your Reference: Reference specified under the "Your Reference" section (if present).
+- Incoterms: Any applicable incoterms mentioned in the document (e.g., FOB, CIF).
+- Total Value of the Order: The total monetary value of the purchase order (include currency).
+- Signature Released By: The name of the person who authorized or released the purchase order.
+- Signature Date: The date when the order was signed (format: DD/MM/YYYY).
+"""
+
+invoice_item_page1_prompt = """ 
+Given the document, extract the following information:
+- Position Title: The role or title mentioned in the document.
+- Location: The place where the service is being provided.
+- Mobilization Date: The date work begins in DD-MM-YYYY format.
+- End Date: The date the work ends in DD-MM-YYYY format.
+- Notice Period: The required notice period for resignation or termination.
+- Items: For each item in the document, provide:
+  - Service Description Code: A code identifying the service.
+  - Completion Date: The date the service was completed in DD-MM-YYYY format.
+  - Quantity: The amount of the item/service provided.
+  - Unit: The unit of measurement (e.g., Days, Hours).
+  - Unit Price: The price per unit.
+  - Net Amount: The total value for the item.
+  - Service Details:A description of the service, which follows the corresponding row for the item.
+
+"""
+
+invoice_item_pages_prompt = """ 
+Given the document, extract the following information:
+- Items:
+  - Service Description Code: A code identifying the service.
+  - Completion Date: The date the service was completed in DD-MM-YYYY format.
+  - Quantity: The amount of the item/service provided.
+  - Unit: The unit of measurement (e.g., Days, Hours).
+  - Unit Price: The price per unit.
+  - Net Amount: The total value for the item.
+  - Service Details:A description of the service, which follows the corresponding row for the item.
+
+"""
+
+invoice_total_page_prompt = """
+extract from the document:
+- Total Amount without taxes.
+- Total net amount of order.
+- Total amount of order.
+"""
+
+invoice_clauses_page_prompt = """ 
+extract from the document the clauses  """