Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

@@ -73,8 +73,8 @@ def generate_text2ppt_input_prompt(input_type, input_value, input_pages):
 
 # Function to execute text2ppt
 def text2ppt(token_key, input_prompt, input_theme):
-    reply = query({"inputs": "You are a kind helpful PPT designer. "+input_prompt})
-
+    output = query({"inputs": "You are a kind helpful PPT designer. "+input_prompt})
+    reply = reply[0]['generated_text'][len("You are a kind helpful PPT designer. "+input_prompt):]
     print(reply)
     md_text = reply[4:] if reply[:3] == "---" else reply
     md_text_list = md_text.split('\n')
@@ -133,6 +133,7 @@ def ppt2script(token_key, input_file, input_type):
 
     input_prompt = header + text
 
-    reply = query({"inputs": "You are a kind helpful PPT Assistant."+input_prompt})
+    output = query({"inputs": "You are a kind helpful PPT Assistant."+input_prompt})
+    reply = reply[0]['generated_text'][len("You are a kind helpful PPT Assistant."+input_prompt:]
 
     return reply