Restyle app to match OHA branding

app.py

@@ -16,6 +16,7 @@ import gradio as gr
 from openai import OpenAI
 from geopy.geocoders import Nominatim
 from geopy.distance import geodesic
+from fpdf import FPDF
 import time
 
 OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
@@ -150,6 +151,7 @@ TRANSLATIONS = {
         "continue_button": "Continue",
         "back_button": "Back",
         "miles_away": "miles away",
+        "num_hospitals": "Number of hospitals to show",
         "free_care": "Free care up to",
         "sliding_scale": "Sliding scale up to",
         "status": "Status",
@@ -179,6 +181,7 @@ TRANSLATIONS = {
         "continue_button": "Continuar",
         "back_button": "Atrás",
         "miles_away": "millas de distancia",
+        "num_hospitals": "Número de hospitales a mostrar",
         "free_care": "Atención gratuita hasta",
         "sliding_scale": "Escala móvil hasta",
         "status": "Estado",
@@ -216,8 +219,8 @@ def _geocode_zip(zip_code):
     _zip_geocode_cache[zip_code] = location
     return location
 
-def find_nearby_hospitals(zip_code, lang):
-    """Find the 3 nearest CT hospitals to a given ZIP code.
+def find_nearby_hospitals(zip_code, lang, max_results=5):
+    """Find the nearest CT hospitals to a given ZIP code.
 
     Validates the ZIP is a CT format (060xx-069xx), geocodes it,
     then returns markdown cards and choice labels sorted by distance.
@@ -239,7 +242,7 @@ def find_nearby_hospitals(zip_code, lang):
             distances.append((name, dist, data))
 
         distances.sort(key=lambda x: x[1])
-        nearest = distances[:3]
+        nearest = distances[:int(max_results)]
 
         choices = []
         cards = ""
@@ -1032,8 +1035,8 @@ def create_interface():
         gr.HTML(
             '<div class="oha-header">'
             '<a href="https://portal.ct.gov/oha" target="_blank" rel="noopener">'
-            '<h1>Office of the Healthcare Advocate<span class="oha-beta">BETA</span></h1>'
-            '<div class="oha-subtitle">Hospital Financial Assistance Eligibility Screener</div>'
+            '<h1>Office of the Healthcare Advocate</h1>'
+            '<div class="oha-subtitle">Hospital Financial Assistance Eligibility Screener <span class="oha-beta">BETA</span></div>'
             '</a>'
             '</div>'
         )
@@ -1049,6 +1052,15 @@ def create_interface():
                 container=False
             )
 
+        # Google Maps embed — placed outside step groups so it persists across all steps
+        gr.HTML(
+            '<div class="map-embed" style="margin-top:1rem;">'
+            '<iframe src="https://www.google.com/maps/d/u/0/embed?mid=1UVO9r7ZS26kZr8Q62dY0147v3Milu_s&ll=41.508444988744984%2C-72.7719992&z=9" '
+            'width="100%" height="480" style="border:0; border-radius: 6px;" '
+            'allowfullscreen="" loading="lazy" title="Map: Connecticut Hospital Locations"></iframe>'
+            '</div>'
+        )
+
         # Step 1: Choose path
         step1 = gr.Group(visible=True, elem_classes="step-group")
         with step1:
@@ -1057,19 +1069,16 @@ def create_interface():
                 help_btn = gr.Button("Find a Hospital by ZIP Code", variant="primary", size="lg")
                 know_btn = gr.Button("I Know My Hospital", variant="secondary", size="lg")
 
-            gr.HTML(
-                '<div class="map-embed" style="margin-top:1rem;">'
-                '<iframe src="https://www.google.com/maps/d/u/0/embed?mid=1UVO9r7ZS26kZr8Q62dY0147v3Milu_s&ll=41.508444988744984%2C-72.7719992&z=9" '
-                'width="100%" height="480" style="border:0; border-radius: 6px;" '
-                'allowfullscreen="" loading="lazy" title="Map: Connecticut Hospital Locations"></iframe>'
-                '</div>'
-            )
 
         # Step 2a: Find hospital by ZIP
         step2a = gr.Group(visible=False, elem_classes="step-group")
         with step2a:
             gr.Markdown("### Enter your ZIP code")
             zip_input = gr.Textbox(label="ZIP Code", placeholder="e.g., 06511", max_lines=1)
+            num_hospitals_slider = gr.Slider(
+                minimum=3, maximum=9, value=5, step=1,
+                label="Number of hospitals to show"
+            )
             search_btn = gr.Button("Search", variant="primary")
 
             nearby_cards = gr.Markdown()
@@ -1118,6 +1127,8 @@ def create_interface():
             chatbot = gr.Chatbot(
                 label="Conversation",
                 height=350,
+                show_share_button=False,
+                show_copy_button=True,
             )
 
             # Suggested prompt buttons (pill-shaped via CSS)
@@ -1170,9 +1181,9 @@ def create_interface():
                 gr.update(visible=False)
             ]
 
-        def search_hospitals_wrapper(zip_code, lang):
+        def search_hospitals_wrapper(zip_code, lang, max_results):
             """Geocode a ZIP and return nearby hospital cards and radio choices."""
-            cards, choices = find_nearby_hospitals(zip_code, lang)
+            cards, choices = find_nearby_hospitals(zip_code, lang, max_results=max_results)
             if choices:
                 return cards, gr.update(choices=choices, visible=True, value=choices[0]), gr.update(visible=True)
             return cards, gr.update(visible=False), gr.update(visible=False)
@@ -1257,13 +1268,65 @@ def create_interface():
                 ]
 
         def download_results(last_result, lang):
-            """Generate a temporary text file with the eligibility summary for download."""
+            """Generate a PDF with the eligibility summary for download."""
             if not last_result:
                 gr.Warning("No results to download." if lang == "en" else "No hay resultados para descargar.")
                 return gr.update(visible=False)
-            content = generate_download_content(last_result, lang)
-            tmp = tempfile.NamedTemporaryFile(mode='w', suffix='.txt', prefix='eligibility_summary_', delete=False)
-            tmp.write(content)
+            t = TRANSLATIONS[lang]
+            data = last_result
+
+            pdf = FPDF()
+            pdf.add_page()
+            pdf.set_auto_page_break(auto=True, margin=20)
+
+            # Header
+            pdf.set_font("Helvetica", "B", 18)
+            pdf.set_text_color(0, 52, 120)  # #003478
+            pdf.cell(0, 12, "CT Hospital Financial Assistance", new_x="LMARGIN", new_y="NEXT")
+            pdf.set_font("Helvetica", "", 12)
+            pdf.set_text_color(33, 37, 41)
+            pdf.cell(0, 8, "Eligibility Summary", new_x="LMARGIN", new_y="NEXT")
+            pdf.ln(4)
+
+            # Hospital & Status
+            pdf.set_font("Helvetica", "B", 14)
+            pdf.cell(0, 10, data['hospital'], new_x="LMARGIN", new_y="NEXT")
+            pdf.set_font("Helvetica", "B", 12)
+            pdf.cell(0, 8, f"{t['status']}: {data['eligibility_status']}", new_x="LMARGIN", new_y="NEXT")
+            pdf.ln(2)
+
+            # Details table
+            pdf.set_font("Helvetica", "", 11)
+            fields = [
+                ("Annual Household Income" if lang == "en" else "Ingreso anual del hogar", f"${data['income']:,.0f}"),
+                ("Household Size" if lang == "en" else "Tamaño del hogar", str(data['household_size'])),
+                ("FPL Percentage" if lang == "en" else "Porcentaje FPL", f"{data['fpl_percentage']}%"),
+                ("Discount Level" if lang == "en" else "Nivel de descuento", data['discount_level']),
+                ("SNAP/WIC Enrolled" if lang == "en" else "Inscrito en SNAP/WIC", "Yes" if data['has_snap_wic'] else "No"),
+                ("Has Insurance" if lang == "en" else "Tiene seguro", "Yes" if data.get('has_insurance') else "No"),
+                ("PA 24-81 Eligible" if lang == "en" else "Elegible PA 24-81", "Yes" if data['pa_24_81_eligible'] else "No"),
+            ]
+            for label, value in fields:
+                pdf.set_font("Helvetica", "B", 10)
+                pdf.cell(80, 7, label)
+                pdf.set_font("Helvetica", "", 10)
+                pdf.cell(0, 7, value, new_x="LMARGIN", new_y="NEXT")
+            pdf.ln(4)
+
+            # Contact
+            pdf.set_font("Helvetica", "B", 11)
+            pdf.cell(0, 8, f"Contact: {data['contact']}", new_x="LMARGIN", new_y="NEXT")
+            if data['asset_limit']:
+                pdf.cell(0, 8, f"Asset Limit: {data['asset_limit']}", new_x="LMARGIN", new_y="NEXT")
+            pdf.ln(4)
+
+            # Disclaimer
+            pdf.set_font("Helvetica", "I", 9)
+            pdf.set_text_color(100, 100, 100)
+            pdf.multi_cell(0, 5, t['disclaimer'])
+
+            tmp = tempfile.NamedTemporaryFile(suffix='.pdf', prefix='eligibility_summary_', delete=False)
+            pdf.output(tmp.name)
             tmp.close()
             return gr.update(value=tmp.name, visible=True)
 
@@ -1330,7 +1393,7 @@ def create_interface():
 
         search_btn.click(
             fn=search_hospitals_wrapper,
-            inputs=[zip_input, lang_state],
+            inputs=[zip_input, lang_state, num_hospitals_slider],
             outputs=[nearby_cards, hospital_radio, continue_btn_a]
         )
 

requirements.txt

@@ -1,3 +1,4 @@
 gradio>=4.0
 openai>=1.0
 geopy>=2.4
+fpdf2>=2.7