Update main.py

main.py

@@ -133,8 +133,8 @@ def inferencePrompt(item: Core):
     try:
         plain_response_data = getPrompt(item.itemString, prompt_model)
         magic_response_data = getPrompt(item.itemString, magic_prompt_model)
-        print(plain_response_data)
-        returnJson = {"plain": plain_response_data[0]["generated_text"], "magic": item.itemString + magic_response_data[0]["generated_text"]}
+        plain_response = plain_response_data['choices'][0]['message']['content'])
+        returnJson = {"plain": plain_response, "magic": item.itemString + magic_response_data[0]["generated_text"]}
         print(f'Return Json {returnJson}')
         return returnJson
     except Exception as e: