Update app.py

app.py

@@ -13,7 +13,6 @@ coach_name_upper = os.getenv("COACH_NAME_UPPER")
 coach_name_long = os.getenv("COACH_NAME_LONG")
 sys_prompt_new = os.getenv("PROMPT_NEW")
 sys_prompt_hist = os.getenv("PROMPT_HIST")
-pics_prompt = os.getenv("PROMPT_PICS")
 theme=os.getenv("THEME")
 ################# End PERSONA-SPECIFIC VALUES ######################
 
@@ -27,9 +26,12 @@ openai_model = os.getenv("OPENAI_MODEL")
 # define file name prefix
 prefix = os.getenv("PREFIX")  #  "/data/" if in HF or "data/" if local
 tx = os.getenv("TX")
+overwrite = os.getenv("OVERWRITE")
 
 # Assistants API prompts
 bullet_instructions = os.getenv("PROMPT_BULLET")
+##################################################################
+########################### END OF HF SECRETS ####################
 
 # Get dateTime string to build a filename reflecting the UserID + Timestamp
 dt = datetime.now()
@@ -73,9 +75,7 @@ def get_system_prompt(user_id):
 
         # System prompt for returning users includes past summary
         return [
-            {"role": "system", "content": sys_prompt_hist + past_summary +
-                "If the user asks how to share or upload pictures, refer them to the 'DatingFinesse.com' website for instructions. The user may have shared pictures with you. Each picture will be separated by 'IMAGE - Date/Time:'. If the user has shared images, most recently shared will be listed last and they would be summarized here: " +
-                user_pics_content + pics_prompt}
+            {"role": "system", "content": sys_prompt_hist + past_summary + "The user may have shared pictures with you. If so, each picture will have a title and detailed description and would be summarized here: " + user_pics_content}
         ]
     else:
         # System prompt for new users
@@ -97,7 +97,6 @@ def on_timeout(user_id):
     global last_file_mod_time
     current_mod_time = get_user_hist_file_mod_time(user_id)
     if current_mod_time is not None and (last_file_mod_time is None or current_mod_time > last_file_mod_time):
-        #print(f"User with ID {user_id} has been inactive, but the history file was updated. Running timeout script.")
         last_file_mod_time = current_mod_time
         # Construct the filename for the user's bullet file 
         user_summ_file_bullet = f"{prefix}{coach_code}-{user_id.upper().replace(' ', '')}-bullet.txt"
@@ -206,12 +205,32 @@ inactivity_thread = threading.Thread(target=check_for_inactivity)
 inactivity_thread.daemon = True
 inactivity_thread.start()
 
+#### UPDATE FILE
+def update_file(file_path, content):
+    try:
+        # Create a backup of the existing file
+        backup_file_path = file_path + "-OLD"
+        if os.path.exists(file_path):
+            # If the backup file already exists, delete it
+            if os.path.exists(backup_file_path):
+                os.remove(backup_file_path)
+            # Rename the original file to the backup file
+            os.rename(file_path, backup_file_path)
+        
+        # Write the new content to the file
+        with open(file_path, "w", encoding="UTF-8") as file:
+            file.write(content)
+        
+        return "File updated successfully. Backup created as " + backup_file_path
+    except Exception as e:
+        return f"Error updating file: {str(e)}"
+
 ###############  CHAT  ###################
 def predict(user_input, history):
     max_length = 2000
     if len(user_input) > max_length:
         user_input = ""
-
+    
     global last_interaction_time, user_id, last_file_mod_time
     # ... [other global variables that are used]
 
@@ -219,8 +238,81 @@ def predict(user_input, history):
     last_interaction_time = time.time()
 
     user_hist_file = f"{prefix}{coach_code}-{user_id}.txt"
+    user_summ_file_bullet = f"{prefix}{coach_code}-{user_id.upper().replace(' ', '')}-bullet.txt"
     pics_file_path = f"{prefix}{coach_code}-{user_id}-pics.txt"
 
+    ####### READ #######
+    if user_input == tx + coach_code:
+        try:
+            # Prepare the transcript for the Textbox output
+            if os.path.exists(user_summ_file_bullet):
+                with open(user_summ_file_bullet, "r", encoding="UTF-8") as file:
+                    output = file.read()
+                yield output
+                return
+        except FileNotFoundError:
+            yield "File '" + user_summ_file_bullet + "' not found."
+            return
+    
+    if user_input == tx + coach_code + "hist":
+        try:
+            # Prepare the transcript for the Textbox output
+            if os.path.exists(user_hist_file):
+                with open(user_hist_file, "r", encoding="UTF-8") as file:
+                    output = file.read()
+                yield output
+                return
+        except FileNotFoundError:
+            yield "File '" + user_hist_file + "' not found."
+            return
+        
+    if user_input == tx + coach_code + "pics":
+        try:
+            # Prepare the transcript for the Textbox output
+            if os.path.exists(pics_file_path):
+                with open(pics_file_path, "r", encoding="UTF-8") as file:
+                    output = file.read()
+                yield output
+                return
+        except FileNotFoundError:
+            yield "File '" + pics_file_path + "' not found."
+            return  
+        
+    ####### UPDATE #######    
+    if user_input.startswith(overwrite + "bull:"):
+        try:
+            file_path, content = user_input[12:].split("|", 1)
+            file_path = user_summ_file_bullet.strip()
+            content = content.strip()
+            
+            if file_path and content:
+                result = update_file(file_path, content)
+                yield result
+                return
+            else:
+                yield "Invalid format. Please use 'update_file: <file_path> | <content>'."
+                return
+        except Exception as e:
+            yield f"Error updating file: {str(e)}"
+            return
+        
+    if user_input.startswith(overwrite + "pics:"):
+        try:
+            file_path, content = user_input[12:].split("|", 1)
+            file_path = pics_file_path.strip()
+            content = content.strip()
+            
+            if file_path and content:
+                result = update_file(file_path, content)
+                yield result
+                return
+            else:
+                yield "Invalid format. Please use 'update_file: <file_path> | <content>'."
+                return
+        except Exception as e:
+            yield f"Error updating file: {str(e)}"
+            return
+
     if user_id == "NOTLOGGEDIN":
         raise gr.Error(f"You must be LOGGED IN to Chat. Refresh the page and try again.")    
     if user_id == "":
@@ -306,7 +398,7 @@ def get_user_summary(name):
 def admin():
     if user_id == "POIPOIPOI":
         # Set the directory path
-        directory_path = "" # different on huggingface
+        directory_path = "data" # different on huggingface
 
         # Get a list of all files in the directory
         file_list = os.listdir(directory_path)
@@ -330,7 +422,9 @@ with gr.Blocks(theme) as demo:
             
             def submit_name(name, password):
                 name = name + "7777"
+                name = name.upper()
                 pwd = password
+                pwd = pwd.upper()
                 if name != pwd:
                     name = "NOTLOGGEDIN"
                     raise gr.Error(f"You must be LOGGED IN to Chat. Refresh the page, log in, and try again.")