Update maker.py

maker.py

@@ -20,7 +20,6 @@ What would you like to make?
 
 welcome_preview_message = """
 Welcome to **{}**! Say something like: 
-
 "{}"
 """
 
@@ -47,7 +46,7 @@ Sure, I'd be happy to help you build a bot! I'm generating a title, system promp
 # Example input: Risks of setting up a non-profit board
 
 <|user|>
-Make a chatbot that roasts tech ceos"
+Make a chatbot that roasts tech ceos
 
 <|assistant|>
 Sure, I'd be happy to help you build a bot! I'm generating a title, system prompt, and an example input. How do they sound? Feel free to give me feedback!
@@ -57,7 +56,7 @@ Sure, I'd be happy to help you build a bot! I'm generating a title, system promp
 
 """
 
-def predict_beta(message, chatbot=[], system_prompt=system_prompt, max_new_tokens=650, temperature=0.4, top_p=0.9, repetition_penalty=0.9, advanced=True):
+def predict_beta(message, chatbot=[], system_prompt=system_prompt, max_new_tokens=650, temperature=0.4, top_p=0.90, repetition_penalty=0.90, advanced=True):
     client = Client(tulu)
     try:
         result = client.predict(
@@ -82,39 +81,45 @@ 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):
+def extract_title_prompt_example(text):
     default_title = "Custom GPT Agent"
     default_system_prompt = "This is a custom GPT agent."
     default_example_input = "Type your query here."
 
-    # Initialize variables
-    title_start = -1
-    prompt_start = -1
-    example_start = -1
-    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
+    # 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
+
     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):
     response = predict_beta(message, history, system_prompt)
-    response, title, system_prompt, example_input = extract_title_prompt_example(response, current_title, current_system_prompt, current_example_input)
+    _, title, system_prompt, example_input = extract_title_prompt_example(response)
     return "", history + [(message, response)], title, system_prompt, example_input, [(None, welcome_preview_message.format(title, example_input))], example_input, gr.Column(visible=True), gr.Group(visible=True)
 
 def set_title_example(title, example):