test logs output

app.py

@@ -1,3 +1,4 @@
+import io
 import json
 import os
 import re
@@ -574,7 +575,19 @@ def format_result_string(result):
     {result['GDC-QAG results']['Question']}
     ```
     ```
-    Query augmented generation final response:  
+    QAG gene entities:  
+    {result['GDC-QAG results']['Gene entities']}
+    ```
+    ```
+    QAG mutation entities:  
+    {result['GDC-QAG results']['Mutation entities']}
+    ```
+    ```
+    QAG cancer entities:  
+    {result['GDC-QAG results']['Cancer entities']}
+    ```
+    ```
+    QAG final response:  
     {result['GDC-QAG results']['Query augmented generation']}
     ```
     
@@ -592,7 +605,19 @@ def format_result_string_multi(result):
     {result['GDC-QAG results']['Question']}
     ```
     ```
-    Query augmented generation final response:  
+    QAG gene entities:  
+    {result['GDC-QAG results']['Gene entities']}
+    ```
+    ```
+    QAG mutation entities:  
+    {result['GDC-QAG results']['Mutation entities']}
+    ```
+    ```
+    QAG cancer entities:  
+    {result['GDC-QAG results']['Cancer entities']}
+    ```
+    ```
+    QAG final response:  
     {multi_result}
     ```
     
@@ -603,6 +628,11 @@ def format_result_string_multi(result):
 
 @utilities.timeit
 def execute_pipeline(question: str):
+    # redirect stdout to logs
+    buffer = io.StringIO()
+    sys_stdout = sys.stdout
+    sys.stdout = buffer
+    
     df = pd.DataFrame({"questions": [question]})
     print(f"\n\nQuestion received: {question}\n")
 
@@ -672,7 +702,15 @@ def execute_pipeline(question: str):
     except Exception as e:
         result_string = format_error_string()
 
-    return result_string
+    
+    # Restore stdout
+    sys.stdout = sys_stdout
+    logs = buffer.getvalue()
+
+    # Format logs for Markdown
+    logs_string = f"```bash\n{logs}\n```"
+    
+    return result_string, logs_string
 
 
 def visible_component(input_text):
@@ -721,17 +759,13 @@ with gr.Blocks(title="GDC QAG MCP server", css="""
         """
     )
     
-    #output = gr.Markdown(
-    #    label="Query Result",
-    #    lines=10,
-    #    max_lines=25,
-    #    info="The Result of the Query will appear here",
-    #)
+    logs = gr.Markdown()
 
+    
     execute_button.click(
         fn=execute_pipeline,
         inputs=[query_input],
-        outputs=output,
+        outputs=[output, logs],
     )