Update app.py

app.py

@@ -1,6 +1,38 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 from google import genai
+import json
+
+from IPython.display import display, HTML, Markdown
+
+
+def show_json(obj):
+  print(json.dumps(obj.model_dump(exclude_none=True), indent=2))
+  return json.dumps(obj.model_dump(exclude_none=True), indent=2)
+
+def show_parts(r):
+  parts = r.candidates[0].content.parts
+  if parts is None:
+    finish_reason = r.candidates[0].finish_reason
+    print(f'{finish_reason=}')
+    return
+  for part in r.candidates[0].content.parts:
+    if part.text:
+      display(Markdown(part.text))
+      output = part.text
+    elif part.executable_code:
+      display(Markdown(f'```python\n{part.executable_code.code}\n```'))
+      output = part.executable_code
+    else:
+      show_json(part)
+
+  grounding_metadata = r.candidates[0].grounding_metadata
+  if grounding_metadata and grounding_metadata.search_entry_point:
+    display(HTML(grounding_metadata.search_entry_point.rendered_content))
+  return output
+
+
+
 client = genai.Client(api_key="AIzaSyD6voSAiSUim17kB90skpdisMMyFXZPxMo")
 MODEL_ID = "gemini-2.0-flash-exp"
 
@@ -49,7 +81,8 @@ def respond(
         response += token
         yield response
     '''
-    response = soccer_chat.send_message(f'''Analyze the conversation history so far: {messages} and come up with a response that continues the conversation forward.''')
+    r = soccer_chat.send_message(f'''Analyze the conversation history so far: {messages} and come up with a response that continues the conversation forward.''')
+    response = show_parts(r)
     print (response)
     yield response