Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

@@ -78,8 +78,9 @@ def text2ppt(token_key, input_prompt, input_theme):
                     "return_full_text": False,
                     "max_new_tokens": 200}})
     # reply = output[0]['generated_text'][len("You are a kind helpful PPT designer. "+input_prompt):]
-    reply = output[0]['generated_text'].strip()
+    reply = output[0]['generated_text']
     print(reply)
+    
     md_text = reply[4:] if reply[:3] == "---" else reply
     md_text_list = md_text.split('\n')
     print(md_text_list)
@@ -89,7 +90,7 @@ def text2ppt(token_key, input_prompt, input_theme):
         # data = ""
         # if md_text_list[i] and "<!--" in md_text_list[i]:
         #     data = "---" + "\n"
-        data = md_text_list[i] + "\n"
+        data = md_text_list[i].strip() + "\n"
         # print(data)
         f.write(data)
     f.close()
@@ -147,6 +148,6 @@ def ppt2script(token_key, input_file, input_type):
                     "return_full_text": False,
                     "max_new_tokens": 200}})
     # reply = output[0]['generated_text'][len("You are a kind helpful PPT Assistant."+input_prompt):]
-    reply = output[0]['generated_text'].strip()
+    reply = output[0]['generated_text']
 
     return reply