Update app.py

app.py

@@ -130,7 +130,7 @@ def bot_streaming(prompt_option, max_new_tokens=4096):
     try:
         # Define predetermined prompts
         prompts = {
-            "Timesheet Details (Full Extraction)": (
+            "NOC Timesheet": (
                 """Extract structured information from the provided timesheet. The extracted details should include:
 
 1. Personnel Details:
@@ -192,7 +192,7 @@ Provide totals for all categories where applicable.
 
 Ensure all extracted data is presented in a clean, structured format. Omit any irrelevant or unrecognizable content. Use the exact terminology and units (e.g., 'days,' 'hours') as found in the document."""
             ),
-            "Timesheet Details (Basic Extraction)": (
+            "NOC Basic": (
                 "Based on the provided timesheet details, extract the following information:\n"
                 "   - Full name of the person\n"
                 "   - Position title of the person\n"
@@ -201,7 +201,7 @@ Ensure all extracted data is presented in a clean, structured format. Omit any i
                 "   - NOC ID\n"
                 "   - Month and year (in MM/YYYY format)"
             ),
-            "Structured Data Extraction": (
+            "NOC Structured test": (
                 "You are an advanced data extraction assistant. Your task is to parse structured input text and extract key data points into clearly defined categories. Focus only on the requested details, ensuring accuracy and proper grouping. Below is the format for extracting the data:\n\n"
                 "---\n"
                 "Project Information\n\n"
@@ -227,6 +227,393 @@ Ensure all extracted data is presented in a clean, structured format. Omit any i
                 "Date of Approval:\n\n"
                 "---\n\n"
                 "Ensure the extracted data strictly follows the format above and is organized by category. Ignore unrelated text. Respond only with the formatted output."
+            ),
+             "Aramco Full structured ": (
+                """You are a document parsing assistant designed to extract structured data from various document types, including invoices, timesheets, purchase orders, and travel bookings. Your goal is to return highly accurate, properly formatted JSON for each document type.
+                General Rules:
+                1. Always return ONLY valid JSON—no explanations, comments, or additional text.
+                2. Use null for any fields that are not present or cannot be extracted.
+                3. Ensure all JSON keys are enclosed in double quotes and properly formatted.
+                4. Validate financial, time tracking, and contract details carefully before output.
+                
+                Extraction Instructions:
+                
+                1. Invoice:
+                   - Parse and extract financial and invoice-specific details.
+                   - JSON structure:
+                     ```json
+                     {
+                         "invoice": {
+                             "date": null,
+                             "dueDate": null,
+                             "accountNumber": null,
+                             "invoiceNumber": null,
+                             "customerContact": null,
+                             "kintecContact": null,
+                             "accountsContact": null,
+                             "periodEnd": null,
+                             "contractNo": null,
+                             "specialistsName": null,
+                             "rpoNumber": null,
+                             "assignmentProject": null,
+                             "workLocation": null,
+                             "expenses": null,
+                             "regularHours": null,
+                             "overtime": null,
+                             "mobilisationAllowance": null,
+                             "dailyHousing": null,
+                             "opPipTechnical": null,
+                             "code": null,
+                             "vatBasis": null,
+                             "vatRate": null,
+                             "vatAmount": null,
+                             "totalExclVat": null,
+                             "totalInclVat": null
+                         }
+                     }
+                     ```
+                
+                2. Timesheet:
+                   - Extract time tracking, work details, and approvals.
+                   - JSON structure:
+                     ```json
+                     {
+                         "timesheet": {
+                             "Year": null,
+                             "RPO_Number": null,
+                             "PMC_Name": null,
+                             "Project_Location": null,
+                             "Project_and_Package": null,
+                             "Month": null,
+                             "Timesheet_Details": [
+                                 {
+                                     "Week": null,
+                                     "Regular_Hours": null,
+                                     "Overtime_Hours": null,
+                                     "Total_Hours": null,
+                                     "Comments": null
+                                 },
+                                 {
+                                     "Week": null,
+                                     "Regular_Hours": null,
+                                     "Overtime_Hours": null,
+                                     "Total_Hours": null,
+                                     "Comments": null
+                                 }
+                             ],
+                             "Monthly_Totals": {
+                                 "Regular_Hours": null,
+                                 "Overtime_Hours": null,
+                                 "Total_Hours": null
+                             },
+                             "reviewedBy": {
+                                 "name": null,
+                                 "position": null,
+                                 "date": null
+                             },
+                             "approvedBy": {
+                                 "name": null,
+                                 "position": null,
+                                 "date": null
+                             }
+                         }
+                     }
+                     ```
+                
+                3. Purchase Order:
+                   - Extract contract and pricing details with precision.
+                   - JSON structure:
+                     ```json
+                     {
+                         "purchaseOrder": {
+                             "contractNo": null,
+                             "relPoNo": null,
+                             "version": null,
+                             "title": null,
+                             "startDate": null,
+                             "endDate": null,
+                             "costCenter": null,
+                             "purchasingGroup": null,
+                             "contractor": null,
+                             "location": null,
+                             "workDescription": null,
+                             "pricing": {
+                                 "regularRate": null,
+                                 "overtimeRate": null,
+                                 "totalBudget": null
+                             }
+                         }
+                     }
+                     ```
+                
+                4. Travel Booking:
+                   - Parse travel-specific and employee information.
+                   - JSON structure:
+                     ```json
+                     {
+                         "travelBooking": {
+                             "requestId": null,
+                             "approvalStatus": null,
+                             "employee": {
+                                 "name": null,
+                                 "id": null,
+                                 "email": null,
+                                 "firstName": null,
+                                 "lastName": null,
+                                 "gradeCodeGroup": null
+                             },
+                             "defaultManager": {
+                                 "name": null,
+                                 "email": null
+                             },
+                             "sender": {
+                                 "name": null,
+                                 "email": null
+                             },
+                             "travel": {
+                                 "startDate": null,
+                                 "endDate": null,
+                                 "requestPolicy": null,
+                                 "requestType": null,
+                                 "employeeType": null,
+                                 "travelActivity": null,
+                                 "tripType": null
+                             },
+                             "cost": {
+                                 "companyCode": null,
+                                 "costObject": null,
+                                 "costObjectId": null
+                             },
+                             "transport": {
+                                 "type": null,
+                                 "comments": null
+                             },
+                             "changeRequired": null,
+                             "comments": null
+                         }
+                     }
+                     ```
+                
+                Use these structures for parsing documents and ensure compliance with the rules and instructions provided for each type.
+                """
+            ),
+             "Aramco Timesheet only ": (
+                """ Extract time tracking, work details, and approvals.
+                   - JSON structure:
+                     ```json
+                     {
+                         "timesheet": {
+                             "Year": null,
+                             "RPO_Number": null,
+                             "PMC_Name": null,
+                             "Project_Location": null,
+                             "Project_and_Package": null,
+                             "Month": null,
+                             "Timesheet_Details": [
+                                 {
+                                     "Week": null,
+                                     "Regular_Hours": null,
+                                     "Overtime_Hours": null,
+                                     "Total_Hours": null,
+                                     "Comments": null
+                                 },
+                                 {
+                                     "Week": null,
+                                     "Regular_Hours": null,
+                                     "Overtime_Hours": null,
+                                     "Total_Hours": null,
+                                     "Comments": null
+                                 }
+                             ],
+                             "Monthly_Totals": {
+                                 "Regular_Hours": null,
+                                 "Overtime_Hours": null,
+                                 "Total_Hours": null
+                             },
+                             "reviewedBy": {
+                                 "name": null,
+                                 "position": null,
+                                 "date": null
+                             },
+                             "approvedBy": {
+                                 "name": null,
+                                 "position": null,
+                                 "date": null
+                             }
+                         }
+                     }
+                     ```"""
+            ),
+             "Aramco test ": (
+                """You are a high-performance document parsing assistant, optimized for speed and accuracy. Your primary objective is to extract structured data from the provided document and return it in valid JSON format with minimal processing time.
+
+                    Guidelines for Speed Optimization:
+                    1. Process the document with minimal computation and only extract the required fields.
+                    2. Use null for any fields that are missing or not clearly identifiable.
+                    3. Avoid redundant checks or deep parsing; rely on the most straightforward extraction methods.
+                    4. Always return ONLY valid JSON—no additional text, explanations, or formatting errors.
+                    5. Focus on precision for key-value pairs; skip over ambiguous or irrelevant information.
+                    
+                    Document-Specific JSON Structures:
+                    
+                    1. **Invoice**:
+                       - Extract financial and customer details efficiently.
+                       - JSON format:
+                         ```json
+                         {
+                             "invoice": {
+                                 "date": null,
+                                 "dueDate": null,
+                                 "accountNumber": null,
+                                 "invoiceNumber": null,
+                                 "customerContact": null,
+                                 "kintecContact": null,
+                                 "accountsContact": null,
+                                 "periodEnd": null,
+                                 "contractNo": null,
+                                 "specialistsName": null,
+                                 "rpoNumber": null,
+                                 "assignmentProject": null,
+                                 "workLocation": null,
+                                 "expenses": null,
+                                 "regularHours": null,
+                                 "overtime": null,
+                                 "mobilisationAllowance": null,
+                                 "dailyHousing": null,
+                                 "opPipTechnical": null,
+                                 "code": null,
+                                 "vatBasis": null,
+                                 "vatRate": null,
+                                 "vatAmount": null,
+                                 "totalExclVat": null,
+                                 "totalInclVat": null
+                             }
+                         }
+                         ```
+                    
+                    2. **Timesheet**:
+                       - Extract time tracking and approval data swiftly.
+                       - JSON format:
+                         ```json
+                         {
+                             "timesheet": {
+                                 "Year": null,
+                                 "RPO_Number": null,
+                                 "PMC_Name": null,
+                                 "Project_Location": null,
+                                 "Project_and_Package": null,
+                                 "Month": null,
+                                 "Timesheet_Details": [
+                                     {
+                                         "Week": null,
+                                         "Regular_Hours": null,
+                                         "Overtime_Hours": null,
+                                         "Total_Hours": null,
+                                         "Comments": null
+                                     },
+                                     {
+                                         "Week": null,
+                                         "Regular_Hours": null,
+                                         "Overtime_Hours": null,
+                                         "Total_Hours": null,
+                                         "Comments": null
+                                     }
+                                 ],
+                                 "Monthly_Totals": {
+                                     "Regular_Hours": null,
+                                     "Overtime_Hours": null,
+                                     "Total_Hours": null
+                                 },
+                                 "reviewedBy": {
+                                     "name": null,
+                                     "position": null,
+                                     "date": null
+                                 },
+                                 "approvedBy": {
+                                     "name": null,
+                                     "position": null,
+                                     "date": null
+                                 }
+                             }
+                         }
+                         ```
+                    
+                    3. **Purchase Order**:
+                       - Extract contract and pricing details with minimal overhead.
+                       - JSON format:
+                         ```json
+                         {
+                             "purchaseOrder": {
+                                 "contractNo": null,
+                                 "relPoNo": null,
+                                 "version": null,
+                                 "title": null,
+                                 "startDate": null,
+                                 "endDate": null,
+                                 "costCenter": null,
+                                 "purchasingGroup": null,
+                                 "contractor": null,
+                                 "location": null,
+                                 "workDescription": null,
+                                 "pricing": {
+                                     "regularRate": null,
+                                     "overtimeRate": null,
+                                     "totalBudget": null
+                                 }
+                             }
+                         }
+                         ```
+                    
+                    4. **Travel Booking**:
+                       - Extract essential travel and employee data efficiently.
+                       - JSON format:
+                         ```json
+                         {
+                             "travelBooking": {
+                                 "requestId": null,
+                                 "approvalStatus": null,
+                                 "employee": {
+                                     "name": null,
+                                     "id": null,
+                                     "email": null,
+                                     "firstName": null,
+                                     "lastName": null,
+                                     "gradeCodeGroup": null
+                                 },
+                                 "defaultManager": {
+                                     "name": null,
+                                     "email": null
+                                 },
+                                 "sender": {
+                                     "name": null,
+                                     "email": null
+                                 },
+                                 "travel": {
+                                     "startDate": null,
+                                     "endDate": null,
+                                     "requestPolicy": null,
+                                     "requestType": null,
+                                     "employeeType": null,
+                                     "travelActivity": null,
+                                     "tripType": null
+                                 },
+                                 "cost": {
+                                     "companyCode": null,
+                                     "costObject": null,
+                                     "costObjectId": null
+                                 },
+                                 "transport": {
+                                     "type": null,
+                                     "comments": null
+                                 },
+                                 "changeRequired": null,
+                                 "comments": null
+                             }
+                         }
+                         ```
+                    
+                    Ensure your parsing method balances accuracy and speed, prioritizing quick turnaround without compromising JSON validity or structural integrity.
+                    """
             )
         }