Update maker.py

maker.py

@@ -108,37 +108,34 @@ def extract_title_prompt_example(text):
     default_system_prompt = "This is a custom GPT agent."
     default_example_input = "Type your query here."
 
-    # Find the start indices of each section
-    title_start = text.find("# Title:")
-    prompt_start = text.find("# System prompt:")
-    example_start = text.find("# Example input:")
-
-    # Extract Title
-    if title_start != -1:
-        title_start += len("# Title:")
-        title_end = prompt_start if prompt_start != -1 else len(text)
-        title = text[title_start:title_end].strip()
-    else:
-        title = default_title
-
-    # Extract System Prompt
-    if prompt_start != -1:
-        prompt_start += len("# System prompt:")
-        prompt_end = example_start if example_start != -1 else len(text)
-        system_prompt = text[prompt_start:prompt_end].strip()
-    else:
-        system_prompt = default_system_prompt
-
-    # Extract Example Input
-    if example_start != -1:
-        example_start += len("# Example input:")
-        example_input = text[example_start:].strip().split("\n", 1)[0]
-    else:
-        example_input = default_example_input
+    # Split the text into lines and reverse it to start from the end
+    lines = text.split('\n')
+    lines.reverse()
+
+    title = default_title
+    system_prompt = default_system_prompt
+    example_input = default_example_input
+
+    # Flags to check if we have found the sections
+    found_title, found_prompt, found_example = False, False, False
+
+    for line in lines:
+        if not found_example and line.startswith("# Example input:"):
+            example_input = line.replace("# Example input:", "").strip()
+            found_example = True
+        elif not found_prompt and line.startswith("# System prompt:"):
+            system_prompt = line.replace("# System prompt:", "").strip()
+            found_prompt = True
+        elif not found_title and line.startswith("# Title:"):
+            title = line.replace("# Title:", "").strip()
+            found_title = True
+
+        # Break the loop if all sections are found
+        if found_title and found_prompt and found_example:
+            break
 
     return text, title, system_prompt, example_input
 
-
 def make_open_gpt(message, history, current_title, current_system_prompt, current_example_input, system_prompt=system_prompt):
     try:
         response = predict_beta(message, history, system_prompt)