Update app.py

Changed the /api/custom-qa endpoint as well as the custom_qa_automation function to try and return only string output instead of JSON.

app.py

@@ -221,7 +221,8 @@ def custom_qa_automation(text: str, custom_question: str):
     # Format the answer with the "Comments:" tag in bold and on a new line
     formatted_answer = f"<p><b>Response:</b> {main_answer}</p><p><b>Comment:</b> {details}</p>" #f"**{custom_question}** {main_answer}\n\n**Comments:** {details}"
 
-    return GenerateResponse(text=formatted_answer)
+    #return GenerateResponse(text=formatted_answer)
+    return formatted_answer # Returning string directly instead of being placed within the GenerateResponse class
 
 # Transcription function
 def transcription(file):
@@ -286,7 +287,11 @@ def inference(input_prompt: Prompt):
 def inference(input_prompt: Prompt):
     return check_agent_steps(text=input_prompt.text)
 
-@app.post("/api/custom-qa", summary="Generate text from prompt", tags=["Generate"], response_model=GenerateResponse)
+# @app.post("/api/custom-qa", summary="Generate text from prompt", tags=["Generate"], response_model=GenerateResponse)
+# def inference(input_prompt: Prompt, question: Prompt):
+#     return custom_qa_automation(text=input_prompt.text, custom_question=question.text)
+
+@app.post("/api/custom-qa", summary="Generate text from prompt", response_class=PlainTextResponse) # Attempt to only return HTML string instead of json output
 def inference(input_prompt: Prompt, question: Prompt):
     return custom_qa_automation(text=input_prompt.text, custom_question=question.text)