test tool integration

app.py

@@ -22,14 +22,12 @@ ui = gr.Interface(
             info="Provide the context/documents to use for answering. The API expects a list of dictionaries, but the UI should except anything"
         ),
     ],
-    outputs=gr.JSON(
-        label="Generated Answer"
-        #lines=6,
-        #show_copy_button=True
-    ),
-    title="ChatFed Generation Module",
-    description="Ask questions based on provided context. Intended for use in RAG pipelines as an MCP server with other ChatFed modules (i.e. context supplied by semantic retriever service).",
-    api_name="generate"
+    outputs=[
+        gr.Text(label="Generated Answer", lines=6, show_copy_button=True),
+        gr.JSON(label="Raw JSON Output")],
+        title="ChatFed Generation Module",
+        description="Ask questions based on provided context. Intended for use in RAG pipelines as an MCP server with other ChatFed modules (i.e. context supplied by semantic retriever service).",
+        api_name="generate"
 )
 
 # Launch with MCP server enabled

utils/generator.py

@@ -222,7 +222,10 @@ async def generate(query: str, context: Union[str, List[Dict[str, Any]]]) -> str
     try:
         messages = build_messages(query, formatted_context)
         answer = await _call_llm(messages)
-        return answer
+
+        # Return string to present in app and JSON for the HuggingChat tool
+        return answer, {"answer": answer}
+        
     except Exception as e:
         logging.exception("Generation failed")
         return f"Error: {str(e)}"