Update app.py

app.py

@@ -130,24 +130,66 @@ def bot_streaming(prompt_option, max_new_tokens=8192):
         # Define predetermined prompts
         prompts = {
             "Timesheet Details (Full Extraction)": (
-                "Based on the provided timesheet details, extract the following information:\n"
-                "   - Full name of the person\n"
-                "   - Position title of the person\n"
-                "   - Work location\n"
-                "   - Contractor's name\n"
-                "   - NOC ID\n"
-                "   - Month and year (in MM/YYYY format)\n"
-                "And from the bottom table:\n"
-                "   - Number of service days onshore\n"
-                "   - Number of standby days onshore in Doha\n"
-                "   - Number of service days offshore\n"
-                "   - Number of service days during weekends or public holidays\n"
-                "   - Number of standby and extended hitch days offshore\n"
-                "   - Number of extended hitch days onshore for rotational personnel\n"
-                "   - Number of Per Diem days for onshore/offshore rotational personnel\n"
-                "   - Number of training days\n"
-                "   - Number of travel days\n"
-                "   - Number of offshore days"
+                """Extract structured information from the provided timesheet. The extracted details should include:
+
+1. Personnel Details:
+
+Name
+
+Position Title
+
+Work Location
+
+Contractor Status (Yes/No)
+
+NOC ID
+
+Month and Year
+
+
+
+2. Service and Activity Summary:
+
+Regular Service Days (ONSHORE)
+
+Standby Days (ONSHORE in Doha)
+
+Offshore Days
+
+Standby & Extended Hitch Days (OFFSHORE)
+
+Extended Hitch Days (ONSHORE Rotational)
+
+Service during Weekends & Public Holidays
+
+
+
+3. Overtime and Compensation:
+
+ONSHORE Overtime Hours (Over 8 hours)
+
+OFFSHORE Overtime Hours (Over 12 hours)
+
+Per Diem Days (ONSHORE/OFFSHORE Rotational Personnel)
+
+
+
+4. Training and Travel:
+
+Training Days
+
+Travel Days
+
+
+
+5. Totals:
+
+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)": (
                 "Based on the provided timesheet details, extract the following information:\n"