Update main.py

main.py

@@ -109,6 +109,7 @@ def getPrompt(prompt, modelID, attempts=1):
         try:    
             apiData={"inputs":prompt, "parameters": parameters, "options": options, "timeout": 45}
             response = requests.post(API_URL + modelID, headers=headers, data=json.dumps(apiData))
+            print(response)
             if response.status_code == 200:
                 try:
                     responseData = response.json()
@@ -123,6 +124,7 @@ def getPrompt(prompt, modelID, attempts=1):
             print(f"An error occurred: {e}")
             if attempts < 3:
                 getPrompt(prompt, modelID, attempts + 1)
+    print(str(response))
     return response.json()
 
 @app.post("/inferencePrompt")
@@ -131,8 +133,9 @@ 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[0]["generated_text"])
+        print(plain_response_data)
         returnJson = {"plain": plain_response_data[0]["generated_text"], "magic": item.itemString + magic_response_data[0]["generated_text"]}
+        print(f'Return Json {returnJson}')
         return returnJson
     except Exception as e:
         returnJson = {"plain": f'An Error occured: {e}', "magic": f'An Error occured: {e}'}