Update main.py

main.py

@@ -117,12 +117,7 @@ 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)
-        print(plain_response_data['choices'])
-        print(plain_response_data['choices'][0])
-        print(plain_response_data['choices'][0]['message'])
-        plain_response = plain_response_data['choices'][0]['message']['content']
-        returnJson = {"plain": plain_response, "magic": item.itemString + magic_response_data[0]["generated_text"]}
+        returnJson = {"plain": plain_response_data.choices[0].message.content, "magic": item.itemString + magic_response_data[0]["generated_text"]}
         print(f'Return Json {returnJson}')
         return returnJson
     except Exception as e: