Add application file

agent.py

@@ -0,0 +1,89 @@
+import os
+from pathlib import Path
+
+from google.adk.agents import Agent
+from google.genai import types as genai_types
+from google.adk.runners import Runner
+from google.adk.sessions.in_memory_session_service import InMemorySessionService
+
+# Standard Python libraries
+import warnings
+import logging
+import os
+import dotenv
+
+dotenv.load_dotenv()
+
+# --- Basic Configuration ---
+warnings.filterwarnings("ignore") # Suppress common warnings
+logging.basicConfig(level=logging.ERROR) # Reduce log verbosity
+logger = logging.getLogger(__name__)
+
+
+logger.info("Initializing new session")
+# Create session service and runner
+
+def read_markdown_file() -> str:
+    """Read the Medicare and You markdown file."""
+    markdown_path = Path("medicare_and_you.md")
+    if not markdown_path.exists():
+        raise FileNotFoundError("medicare_and_you.md not found")
+    return markdown_path.read_text()
+
+
+# Main Medicare agent
+root_agent = Agent(
+    name="medicare_assistant",
+    model="gemini-2.0-flash-lite",
+    description="Helps users understand Medicare benefits using the Medicare & You handbook",
+    instruction=f"""
+    You are a caring Medicare guide. Your job is to help people understand their Medicare benefits 
+    using information from the Medicare & You 2025 handbook. Fall back to google search if you don't have the information.
+
+    Here is the complete Medicare & You 2025 handbook content that you should use to answer questions:
+
+    {read_markdown_file()}
+
+    When answering questions, always:
+    
+    1. Use only the handbook content provided above
+    2. Explain concepts in simple, friendly terms
+    3. Break down complex topics into easy steps
+    4. If unsure, ask clarifying questions
+    5. End with a note saying to refer tothe page number with the markdown link to the Medicare & You handbook, where the markdown would look like [Medicare & You Handbook](https://www.medicare.gov/publications/10050-medicare-and-you.pdf)
+    6. No need to ask if the user has additional questions, just answer the question.
+        
+    Remember: Be warm, patient, and focus on making Medicare easy to understand.
+    """,
+    tools=[],
+    generate_content_config=genai_types.GenerateContentConfig(
+        temperature=0.1
+        )
+    )
+
+# @title Define Agent Interaction Function
+
+
+async def call_agent_async(query: str, runner, user_id, session_id):
+  """Sends a query to the agent and prints the final response."""
+
+  # Prepare the user's message in ADK format
+  content = genai_types.Content(role='user', parts=[genai_types.Part(text=query)])
+
+  final_response_text = "Agent did not produce a final response." # Default
+
+  # Key Concept: run_async executes the agent logic and yields Events.
+  # We iterate through events to find the final answer.
+  async for event in runner.run_async(user_id=user_id, session_id=session_id, new_message=content):
+      if event.is_final_response():
+          if event.content and event.content.parts:
+             # Assuming text response in the first part
+             final_response_text = event.content.parts[0].text
+          elif event.actions and event.actions.escalate: # Handle potential errors/escalations
+             final_response_text = f"Agent escalated: {event.error_message or 'No specific message.'}"
+          # Add more checks here if needed (e.g., specific error codes)
+          break # Stop processing events once the final response is found
+
+  return final_response_text
+
+

app.py

@@ -0,0 +1,95 @@
+import streamlit as st
+import asyncio
+
+from agent import root_agent, call_agent_async
+from google.adk.runners import Runner
+from google.adk.sessions.in_memory_session_service import InMemorySessionService
+
+# Standard Python libraries
+import warnings
+import logging
+
+import dotenv
+import uuid
+# --- Asyncio Configuration for Streamlit ---
+# Required to run async ADK functions within Streamlit's synchronous environment
+import nest_asyncio
+nest_asyncio.apply()
+
+dotenv.load_dotenv()
+
+# --- Basic Configuration ---
+warnings.filterwarnings("ignore") # Suppress common warnings
+logging.basicConfig(level=logging.WARNING) # Reduce log verbosity
+logger = logging.getLogger(__name__)
+
+st.title("🏥 Medicare Guide!")
+
+# Show welcome message only on first load
+if "welcome_shown" not in st.session_state:
+    st.session_state.welcome_shown = True
+    st.session_state.user_id = f"streamlit_user_{uuid.uuid4()}"
+    st.session_state.session_id = f"streamlit_session_{uuid.uuid4()}"
+    # Main welcome message
+    st.markdown("""        
+    ### 💡 Examples of questions you can ask me:
+    - "What are the different parts of Medicare?"
+    - "When can I enroll in Medicare?"
+    - "What's the difference between Original Medicare and Medicare Advantage?"
+    """)
+    
+# Disclaimer
+st.info("🔔 Note: I provide information from the official [Medicare & You Handbook 2025](https://www.medicare.gov/publications/10050-medicare-and-you.pdf), but am not perfect! I will always do my best to provide page numbers, and I strongly suggest checking for yourself.")
+
+
+if "adk_session" not in st.session_state:
+    app_name="Medicare Guide"
+    session_service = InMemorySessionService()
+    adk_session = session_service.create_session(
+        app_name=app_name,
+        user_id=st.session_state.user_id,
+        session_id=st.session_state.session_id,
+    )
+
+    runner = Runner(
+        app_name="Medicare Guide",
+        agent=root_agent,
+        session_service=session_service,
+    )
+    st.session_state.adk_session = adk_session
+    st.session_state.runner = runner
+    st.session_state.session_service = session_service
+    st.session_state.app_name = app_name
+
+runner = st.session_state.runner
+session_service = st.session_state.session_service
+app_name = st.session_state.app_name
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"], unsafe_allow_html=True)
+
+# Accept user input
+if prompt := st.chat_input("What can I answer for you about Medicare?"):
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    # Display user message in chat message container
+    with st.chat_message("user"):
+        st.markdown(prompt, unsafe_allow_html=True)
+        
+    # Display assistant response in chat message container
+    with st.chat_message("assistant"):
+        try:
+            response_text = asyncio.run(call_agent_async(prompt, runner, st.session_state.user_id, st.session_state.session_id))
+            st.markdown(response_text, unsafe_allow_html=True)
+        except Exception as e:
+            st.error(f"An error occurred during agent interaction: {e}")
+            response_text = f"Sorry, a critical error occurred: {e}"
+            logger.error(f"An error occurred during agent interaction: {e}", exc_info=True)
+            
+    st.session_state.messages.append({"role": "assistant", "content": response_text})        

pdf_to_markdown.py

@@ -0,0 +1,30 @@
+import pymupdf4llm
+import pathlib
+from datetime import datetime
+
+def convert_pdf_to_markdown(input_pdf, output_md):
+    # Get the markdown text
+    print(f"Converting {input_pdf} to markdown...")
+    md_text = pymupdf4llm.to_markdown(input_pdf)
+    
+    # Add a header with metadata
+    header = f"""# Medicare and You Documentation
+> Converted from PDF on {datetime.now().strftime('%Y-%m-%d')}
+> Source: {input_pdf}
+
+---
+
+"""
+    
+    # Combine header and content
+    final_content = header + md_text
+    
+    # Write to file
+    print(f"Writing markdown to {output_md}...")
+    pathlib.Path(output_md).write_bytes(final_content.encode())
+    print("Conversion complete!")
+
+if __name__ == "__main__":
+    input_pdf = "10050-medicare-and-you.pdf"
+    output_md = "medicare_and_you.md"
+    convert_pdf_to_markdown(input_pdf, output_md)

requirements.txt

@@ -0,0 +1,5 @@
+huggingface_hub
+streamlit
+google-adk
+openai
+pymupdf4llm