Update maker.py

maker.py

@@ -74,32 +74,30 @@ def predict_beta(message, chatbot=[], system_prompt=system_prompt, max_new_token
         error_msg = f"An error occurred: {str(e)}"
         raise gr.Error(error_msg)
 
-
 def extract_title_prompt_example(text, title, system_prompt, example_input):
     try:
-        # Finding the indices of the key terms
-        text_start = text.rfind("<|assistant|>", ) + len("<|assistant|>")
-        text = text[text_start:]
-    except ValueError:
-        pass
-    try:
-        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
+        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 ""
+        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 ""
+        else:
+            system_prompt = ""
+
+        if example_start >= 0:
+            example_input = text[example_start + len("example input:"):].strip().split("\n")[0]
+        else:
+            example_input = ""
+    except Exception as e:
+        print(f"Error in extract_title_prompt_example: {e}")
+        # Return the original values if there's an error
+        return text, title, system_prompt, example_input
+
     return text, title, system_prompt, example_input
 
 def make_open_gpt(message, history, current_title, system_prompt, current_example_input):