Update maker.py

maker.py

@@ -97,31 +97,23 @@ def extract_title_prompt_example(text, title, system_prompt, example_input):
     prompt_start = -1
     example_start = -1
     try:
-        title_start = text.lower().find("# Title:") + len("# Title:")
-        if title_start >= len("# Title:"):
-            prompt_start = text.lower().find("# System prompt:", title_start)
-            title = text[title_start:prompt_start].strip() if prompt_start != -1 else ""
-            print("Title:", title)
-
-        else:
-            title = ""
-
-        if prompt_start >= 0:
-            example_start = text.lower().find("# Example input:", prompt_start)
-            system_prompt = text[prompt_start + len("# System prompt:"):example_start].strip() if example_start != -1 else ""
-            print("System Prompt:", system_prompt)
-        else:
-            system_prompt = ""
-
-        if example_start >= 0:
-            example_input = text[example_start + len("Example input:"):].strip().split("\n")[0]
-            print("Example Input:", example_input)
-
-        else:
-            example_input = ""
-    except Exception as e:
-        print(f"Error in extract_title_prompt_example: {e}")
-        return text, title, system_prompt, example_input
+        title_start = text.lower().rfind("title:") + len("title:")    
+        prompt_start = text.lower().rfind("system prompt:")
+        title = text[title_start:prompt_start].strip()
+    except ValueError:
+        pass
+    try:
+        prompt_start = text.lower().rfind("system prompt:") + len("system prompt:")
+        example_start = text.lower().rfind("example input:")
+        system_prompt = text[prompt_start:example_start].strip()
+    except ValueError:
+        pass
+    try:
+        example_start = text.lower().rfind("example input:") + len("example input:")
+        example_input = text[example_start:].strip()
+        example_input = example_input[:example_input.index("\n")]
+    except ValueError:
+        pass
     return text, title, system_prompt, example_input
 
 def make_open_gpt(message, history, current_title, system_prompt, current_example_input):
@@ -137,8 +129,6 @@ textbox_preview = gr.Textbox(scale=7, container=False)
 
 def test_preview_chatbot(message, history, system_prompt):
     response = predict_beta(message, history, system_prompt)
-    text_start = response.rfind("<|assistant|>", ) + len("<|assistant|>")
-    response = response[text_start:]
     return response