Spaces:
Running
Running
| from pydantic import BaseModel, Field | |
| from typing import Optional,List | |
| from langchain_core.output_parsers import JsonOutputParser # type: ignore | |
| 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.") | |
| 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.") | |
| approval_status: str = Field("not approved",description="Indicates whether the NOC representative's name and date are present on the document (e.g., 'approved', 'not approved').") | |
| approved_on : str = Field(...,description="DD/MM/YY of the stamp") | |
| approved_by : str = Field(...,description="Name of the person who approved the document") | |
| 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.") | |
| approval_status: str = Field("not approved",description="Indicates whether the NOC representative's name and date are present on the document (e.g., 'approved', 'not approved').") | |
| approved_on : str = Field(...,description="DD/MM/YY of the stamp") | |
| approved_by : str = Field(...,description="Name of the person who approved the document") | |
| class Noc_TimeSheetInformation(BaseModel): | |
| """Details of a timesheet entry.""" | |
| position_title: str = Field(..., description="Position title of the person.") | |
| work_location: str = Field(..., description="Work location ") | |
| agency: str = Field(..., description="the agency") | |
| noc_id: str = Field(..., description="NOC ID of the person.") | |
| full_name: str = Field(..., description="Full name of the person.") | |
| approval_status: str = Field("not approved",description="(e.g., 'approved', 'not approved').") | |
| approved_on : str = Field(...,description="DD/MM/YY of the stamp") | |
| approved_by : str = Field(...,description="Name of the person who approved the document") | |
| 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.") | |
| signed: bool = Field(..., description="Whether the document has been signed or not.") | |
| 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.") | |
| class service(BaseModel): | |
| service: str = Field(..., description="the service name.") | |
| from_date: str = Field(None,description="starting date in DD/MM/YYYY format..") | |
| to_date: str = Field(None,description="ending date in DD/MM/YYYY format.") | |
| currency : str =Field(...,description="currency of the rate.") | |
| fx: str = Field(None,description="foreign exchange.") | |
| Number_of_days_hours: str = Field(...,description="number of hours or days for the service.") | |
| rate: str = Field(..., description="the rate of the service.") | |
| total: str = Field(...,description="total which is the rate* No of days/hours .") | |
| class bank_details(BaseModel): | |
| bank_name: str = Field(..., description="Name of the bank.") | |
| swift_bic_code: str = Field(None, description=" SWIFT/BIC CODE.") | |
| iban_number: str = Field(...,description="IBAN Number.") | |
| beneficiary_name:str = Field(...,description="full name") | |
| account_currency: str = Field(...,description="Account currency.") | |
| expected_amount: str = Field(...,description="the amount.") | |
| class Noc_Invoice(BaseModel): | |
| """Details of an invoice.""" | |
| invoice_date: str = Field(..., description="Date of the invoice in DD/MM/YYYY format") | |
| invoice_number: str = Field(..., description="Unique identifier for the invoice.") | |
| full_name: str = Field(..., description="Full name of the person.") | |
| invoice_to:str = Field(...,description="email to send the invoice forward to.") | |
| company_name: str = Field(...,description="company name") | |
| address: str = Field(..., description="Address of the company.") | |
| services : List[service] = Field(None,description="list of services in the table.") | |
| sub_total: float = Field(...,description="the sub total.") | |
| vat: float = Field(None, description="the vat.") | |
| Withholding_tax : float = Field(None,description="Withholding Tax.") | |
| total_due:float= Field(...,description="the total due.") | |
| first_bank : List[bank_details]=Field(...,description="first bank informations.") | |
| second_details : List[bank_details]=Field(...,description="second bank details") | |
| Noc_timesheet_prompt = """ | |
| Based on the provided timesheet details, extract the following information: | |
| - Position title of the person. | |
| - Work location . | |
| - the agency. | |
| - NOC ID of the person. | |
| - Name of the person. | |
| - approval status | |
| - date of the approval | |
| - approved by | |
| - 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 over time hours onshore (Over 8 hours) | |
| - Number of over time hours offshore (Over 12 hours) | |
| - Number of Per Diem days for onshore/offshore rotational personnel | |
| - Number of training days | |
| - Number of travel days | |
| """ | |
| Noc_invoice_prompt =""" | |
| Based on the provided timesheet details, extract the following information: | |
| - Invoice date | |
| - Invoice number | |
| - Full name of the person | |
| - Email to send the invoice forward to. | |
| - Company name | |
| - Address of the company | |
| - List of services,for each existing service provide: | |
| - the service name. | |
| - starting date in DD/MM/YYYY format. | |
| - ending date in DD/MM/YYYY format. | |
| - currency of the rate. | |
| - foreign exchange. | |
| - number of hours or days for the service. | |
| - the rate of the service. | |
| - total which is the rate* No of days/hours . | |
| - Sub total | |
| - VAT or Withholding Tax. | |
| - Total due | |
| - First bank informations, for each existing bank provide, do not mess up the iban: | |
| - Name of the bank. | |
| - SWIFT/BIC CODE. | |
| - IBAN Number. | |
| - full name. | |
| - Account currency. | |
| - the amount. | |
| - Second bank informations, for each existing bank provide,do not mess up the iban: | |
| - Name of the bank. | |
| - SWIFT/BIC CODE. | |
| - IBAN Number. | |
| - full name. | |
| - Account currency. | |
| - the amount. | |
| """ | |
| 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 over time hours onshore (Over 8 hours) | |
| - Number of over time hours offshore (Over 12 hours) | |
| - Number of Per Diem days for onshore/offshore rotational personnel | |
| - Number of training days | |
| - Number of travel days | |
| - Indicates whether the NOC representative's name and date are present on the document (e.g., 'approved', 'not approved'). | |
| - DD/MM/YY of the stamp | |
| - Name of the person who approved the document | |
| """ | |
| 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 | |
| - Indicates whether the NOC representative's name and date are present on the document (e.g., 'approved', 'not approved'). | |
| - DD/MM/YY of the stamp | |
| - Name of the person who approved the document | |
| """ | |
| 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). | |
| - signed: Whether the document has been signed or not. | |
| - 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 """ | |
| # CHOOSING PARSER DEPENDING ON THE TYPE OF DOCUMENT | |
| Noc_Res_timeSheet_parser = JsonOutputParser(pydantic_object=Noc_Residential_TimeSheetInformation) | |
| Noc_Rot_timeSheet_parser = JsonOutputParser(pydantic_object=Noc_Rotational_TimeSheetInformation) | |
| Noc_PurchaseOrder_information_parser = JsonOutputParser(pydantic_object=Noc_PurchaseOrderInformation) | |
| Noc_PurchaseOrder_item1_parser = JsonOutputParser(pydantic_object=Noc_Document_Information) | |
| Noc_PurchaseOrder_items_parser = JsonOutputParser(pydantic_object=Noc_items) | |
| Noc_PurchaseOrder_total_parser = JsonOutputParser(pydantic_object=Noc_total) | |
| Noc_PurchaseOrder_clauses_parser = JsonOutputParser(pydantic_object=Noc_Clauses) | |
| Noc_invoice_parser_v1 = JsonOutputParser(pydantic_object=Noc_Invoice) | |
| Noc_timesheet_parser_v1 = JsonOutputParser(pydantic_object=Noc_TimeSheetInformation) |