Create admin_messages.py

admin_messages.py

@@ -0,0 +1,67 @@
+import re
+import json
+from msg import fetch_json_from_github, update_user_json_file, fetch_authenticity_token_and_commit_oid
+
+def extract_admin_message(ai_response):
+    """
+    Extract admin message from AI response and return both the admin message
+    and the cleaned response (without admin tags).
+    """
+    admin_pattern = r'<admin>(.*?)</admin>'
+    admin_messages = re.findall(admin_pattern, ai_response, re.DOTALL)
+
+    # Clean the response by removing admin tags and their content
+    cleaned_response = re.sub(admin_pattern, '', ai_response)
+
+    if admin_messages:
+        return admin_messages[0].strip(), cleaned_response.strip()
+    else:
+        return None, ai_response
+
+def save_admin_message(page_id, message, user_profile_img):
+    """
+    Save admin message to the GitHub JSON file.
+    """
+    try:
+        # Fetch current JSON data
+        github_response = fetch_json_from_github()
+        if not github_response["success"]:
+            print(f"Error fetching JSON data: {github_response['message']}")
+            return False
+
+        # Get the current data
+        data = github_response["data"]
+
+        # Initialize the page_id entry if it doesn't exist
+        if page_id not in data:
+            data[page_id] = []
+
+        # Add the new message
+        new_message = {
+            "message": message,
+            "img_profile": user_profile_img
+        }
+
+        data[page_id].append(new_message)
+
+        # Convert the updated data to JSON string
+        new_content = json.dumps(data, indent=2)
+
+        # Get authentication tokens for GitHub
+        authenticity_token, commit_oid = fetch_authenticity_token_and_commit_oid()
+        if not authenticity_token or not commit_oid:
+            print("Failed to get GitHub authentication tokens")
+            return False
+
+        # Update the file on GitHub
+        update_result = update_user_json_file(authenticity_token, commit_oid, new_content)
+        if update_result["success"]:
+            print("Admin message saved successfully")
+            return True
+        else:
+            print(f"Error saving admin message: {update_result['message']}")
+            return False
+
+    except Exception as e:
+        print(f"Error saving admin message: {str(e)}")
+        return False