feature-modify-system-instructions (#2)

- changed system prompt to check for items to be replaced in Guest Rooms (d25ebd824f059236ee9c03250b7f804544f3a40d)


Co-authored-by: Arjun Shankar <sarjunshankar@users.noreply.huggingface.co>

app.py

@@ -86,75 +86,45 @@ terminators = [
     tokenizer.convert_tokens_to_ids("<|eot_id|>")
 ]
 
-# SYS_PROMPT = """
-# Extract all relevant keywords and add quantity from the following text and format the result in nested JSON, ignoring personal details and focusing only on the scope of work as shown in the example:
-# Good JSON example: {'lobby': {'frcm': {'replace': {'carpet': 1, 'carpet_pad': 1, 'base': 1, 'window_treatments': 1, 'artwork_and_decorative_accessories': 1, 'portable_lighting': 1, 'upholstered_furniture_and_decorative_pillows': 1, 'millwork': 1} } } }
-# Bad JSON example: {'lobby': { 'frcm': { 'replace': [ 'carpet', 'carpet_pad', 'base', 'window_treatments', 'artwork_and_decorative_accessories', 'portable_lighting', 'upholstered_furniture_and_decorative_pillows', 'millwork'] } } }
-# Make sure to fetch details from the provided text and ignore unnecessary information. The response should be in JSON format only, without any additional comments.
-# """
-
 SYS_PROMPT = """
-Extract all relevant keywords and add quantities from the following text and format the result in nested JSON, ignoring personal details and focusing only on the area and furniture items as shown in the example. Each item should have a count, which will be set to 1 for simplicity. The response should be in JSON format only, without any additional comments.
-
-Good JSON example:{
-  "Lobby Area/Entrance": {
-    "Vinyl wall covering": 1,
-    "Decorative hardwired lighting": 1
-  },
-  "Lobby": {
-    "Carpet, carpet pad, and base": 1,
-    "Window treatments": 1,
-    "Artwork and decorative accessories": 1,
-    "Portable lighting": 1,
-    "Upholstered furniture and decorative pillows": 1,
-    "Millwork": 1
-  }
-}
-Make sure to fetch details from the provided text and ignore unnecessary information. The response should be in JSON format only, without any additional comments.
-
-Task:
-Convert the provided extracted text into the JSON format described above.
-
-Provided Text:
-
-PROPERTY IMPROVEMENT PLAN
-PREPARED FOR:
-Springfield, IL
-To be relicensed as Hilton Garden Inn
-...
-Patios/The Terrace - Install patio decorative lighting. Install patio furniture. (lounge chairs, chaise, dining tables/chairs)
+Given the text of a hotel property improvement plan, extract the items to be replaced for only the Guest Rooms/ Suites, Guest Bathrooms/Suite Bathrooms.
+First, find the section of the pdf which describes improvements to be done on the Guest Rooms and Guest Bathrooms, then find the items to be replaced.
+Ignore items from other sections of the hotel.
+Items to be replaced are usually preceded by the words replace, install, or provide. 
+Return the results as a JSON with "Guest Room" and "Guest Bathroom" as keys and each value the list of unique items to be replaced.
+Return only the JSON with no extra text.
+
+Example Text:
+"
+Site & Building Exterior
+Replace all exterior decorative lighting
 ...
-Lobby Area - Replace carpet, carpet pad, and base. Replace window treatments. Replace artwork and decorative accessories. Replace portable lighting. (floor lamps, table lamps) Replace upholstered furniture and decorative pillows. Replace millwork. Replace the television(s).
+Guestrooms
+Replace [ORG] C-Table.
+Provide full length mirror.
+Replace cabinets - Kitchen.
+at doors where brass hardware finishes exist – replace with stainless
 ...
-Registration Area - Replace vinyl wall covering. Replace hard surface floor covering. Replace artwork. Install new signature graphics on the back wall.
+Guest Bathrooms - (FRCM) Replace mirrors. Install a vanity mirror that has integrated lighting
+Guest Bathrooms - (FRCM) Replace artwork and decorative accessories.
 ...
-
-Expected Output (JSON format):
+Suites - Replace microwave, refrigerator, and associated casegood cabinet.
+"
+Example Response:
 {
-  "Patios/The Terrace": {
-    "Patio decorative lighting": 1,
-    "Lounge chairs": 1,
-    "Chaise": 1,
-    "Dining tables": 1,
-    "Dining chairs": 1,
-    "Patio furniture": 1
-  },
-  "Lobby Area": {
-    "Carpet, carpet pad, and base": 1,
-    "Window treatments": 1,
-    "Artwork and decorative accessories": 1,
-    "Portable lighting (floor lamps, table lamps)": 1,
-    "Upholstered furniture and decorative pillows": 1,
-    "Millwork": 1,
-    "Television(s)": 1
-  },
-  "Registration Area": {
-    "Vinyl wall covering": 1,
-    "Hard surface floor covering": 1,
-    "Artwork (new signature graphics on the back wall)": 1
-  }
+"Guest Room": [
+  "C-Table",
+  "full length mirror",
+  "kitchen cabinets",
+  "stainless steel door hardware",
+  "microwave",
+  "refrigerator",
+  "casegood cabinet",],
+"Guest Bathroom": [
+  "vanity mirror with integrated lighting",
+  "artwork",
+  "decorative accessories",],
 }
-
 """
 def chunk_text(text, chunk_size=5000):
     """