Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

@@ -74,7 +74,7 @@ def generate_text2ppt_input_prompt(input_type, input_value, input_pages):
 # Function to execute text2ppt
 def text2ppt(token_key, input_prompt, input_theme):
     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):]
+    reply = output[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')
@@ -134,6 +134,6 @@ def ppt2script(token_key, input_file, input_type):
     input_prompt = header + text
 
     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):]
+    reply = output[0]['generated_text'][len("You are a kind helpful PPT Assistant."+input_prompt):]
 
     return reply