Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

@@ -83,11 +83,10 @@ def text2ppt(token_key, input_prompt, input_theme):
     md_text = reply[4:] if reply[:3] == "---" else reply
     md_text_list = md_text.split('\n')
     print(md_text_list)
-    base_line = md_text_list[1][:7]
 
     f = open("text2ppt_input.md", 'w')
     for i in range(0, len(md_text_list)):
-        if md_text_list[i][:7] == base_line:
+        if md_text_list[i][0] == "<":
             data = "---" + "\n"
         data = md_text_list[i] + "\n"
         print(data)