Implement new chat interface for immigration research, including user profile input and example prompts. Introduce a dedicated prompts module for system instructions, enhancing clarity and functionality.

ui/chat.py

@@ -1,42 +1,171 @@
 # ui/chat.py
+from __future__ import annotations
+
 import gradio as gr
 
 from ui.globe import render_globe_panel
 from ui.globe_commands import empty_globe_state
 from ui.inference import respond
+from ui.inference.prompts import BORDERLESS_SYSTEM_PROMPT
+
+DEFAULT_MAX_TOKENS = 1800
+DEFAULT_TEMPERATURE = 0.35
+DEFAULT_TOP_P = 0.9
 
-ADDITIONAL_INPUTS = [
-    gr.Textbox(
-        value=(
-            "You are Borderless, an immigration research assistant with access to "
-            "web search, page scraping, and an interactive 3D globe. Use "
-            "search_immigration_info as your primary discovery tool for visa rules, "
-            "pathways, documents, fees, processing times, and economic context — "
-            "then scrape_web_page on the best official URLs for full content. Prefer "
-            "official .gov, immigration authority, and embassy domains; use "
-            "include_domains when you know the right government site. Use "
-            "crawl_web_site only when multiple pages from the same official site "
-            "are needed; prefer scrape_web_page when one page is enough. Cite source "
-            "URLs when answering from search or scraped content. When you recommend "
-            "or compare destination countries, call update_globe with action show so "
-            "the user can see them on the globe. When the user asks to mark, map, "
-            "show, highlight, or fly to countries, call update_globe immediately "
-            "instead of describing the tool call."
-        ),
-        label="System message",
+EXAMPLE_PROMPTS = [
+    (
+        "India software engineer",
+        "I'm a software engineer from India with 5 years of backend experience, "
+        "a master's degree, English fluency, and a budget of about USD 18,000. "
+        "I want to relocate for work within the next 12-18 months. Which "
+        "countries are realistic, what visa pathways should I look at, and "
+        "what documents should I prepare?",
+    ),
+    (
+        "Hong Kong student",
+        "I hold a Hong Kong passport and want to study in Europe on a modest "
+        "budget. I speak English and conversational Mandarin, can show around "
+        "USD 12,000 in savings, and care about post-study work options. Compare "
+        "realistic countries and mark them on the globe.",
     ),
-    gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-    gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-    gr.Slider(
-        minimum=0.1,
-        maximum=1.0,
-        value=0.95,
-        step=0.05,
-        label="Top-p (nucleus sampling)",
+    (
+        "Remote worker family",
+        "I'm a freelance product designer from Brazil, married with one child, "
+        "earning around USD 4,500 per month from overseas clients. We want a "
+        "safe country with good schools, a route that does not require a local "
+        "job offer immediately, and a timeline under one year.",
     ),
 ]
 
 
+def _chatbox_value(text: str) -> dict[str, object]:
+    return {"text": text.strip(), "files": []}
+
+
+def example_prompt(prompt: str) -> dict[str, object]:
+    return _chatbox_value(prompt)
+
+
+def build_profile_prompt(
+    citizenship: str,
+    current_location: str,
+    education: str,
+    work_history: str,
+    languages: str,
+    budget: str,
+    family: str,
+    timeline: str,
+    goals: str,
+) -> dict[str, object]:
+    sections = [
+        ("Citizenship/passports", citizenship),
+        ("Current location", current_location),
+        ("Education", education),
+        ("Work history", work_history),
+        ("Languages", languages),
+        ("Budget and savings", budget),
+        ("Family/dependents", family),
+        ("Target timeline", timeline),
+        ("Goals and preferences", goals),
+    ]
+    details = "\n".join(
+        f"- {label}: {value.strip()}"
+        for label, value in sections
+        if value and value.strip()
+    )
+    if not details:
+        details = "- I have not filled in the profile yet. Ask me only for the missing details that matter most."
+
+    return _chatbox_value(
+        "Research realistic migration options for my profile.\n\n"
+        f"{details}\n\n"
+        "Please recommend 3-5 suitable countries, mark them on the globe, "
+        "compare visa pathways, required documents, approximate costs, "
+        "realistic timelines, risks, and official sources to verify."
+    )
+
+
+def render_intake_panel(chat_input: gr.MultimodalTextbox) -> None:
+    with gr.Column(elem_classes=["borderless-intake"]):
+        gr.Markdown(
+            "## Borderless\n"
+            "Describe your background once. The agent researches official sources, "
+            "compares realistic visa pathways, and marks suitable countries on the globe.",
+            elem_classes=["borderless-hero"],
+        )
+        with gr.Accordion("Start with a guided profile", open=True):
+            with gr.Row():
+                citizenship = gr.Textbox(
+                    label="Citizenship / passports",
+                    placeholder="e.g. India; no second passport",
+                )
+                current_location = gr.Textbox(
+                    label="Current location",
+                    placeholder="e.g. Singapore on an employment pass",
+                )
+            with gr.Row():
+                education = gr.Textbox(
+                    label="Education",
+                    placeholder="e.g. Master's in computer science",
+                )
+                work_history = gr.Textbox(
+                    label="Work history",
+                    placeholder="e.g. 5 years as backend software engineer",
+                )
+            with gr.Row():
+                languages = gr.Textbox(
+                    label="Languages",
+                    placeholder="e.g. English C1, German A2",
+                )
+                budget = gr.Textbox(
+                    label="Budget",
+                    placeholder="e.g. USD 18,000 available for relocation",
+                )
+            with gr.Row():
+                family = gr.Textbox(
+                    label="Family / dependents",
+                    placeholder="e.g. spouse, one child, or solo applicant",
+                )
+                timeline = gr.Textbox(
+                    label="Target timeline",
+                    placeholder="e.g. 12-18 months",
+                )
+            goals = gr.Textbox(
+                label="Goals in everyday language",
+                lines=3,
+                placeholder=(
+                    "e.g. I want a realistic skilled-worker route, lower risk, "
+                    "and a path to permanent residence."
+                ),
+            )
+            gr.Button("Create research prompt", variant="primary").click(
+                fn=build_profile_prompt,
+                inputs=[
+                    citizenship,
+                    current_location,
+                    education,
+                    work_history,
+                    languages,
+                    budget,
+                    family,
+                    timeline,
+                    goals,
+                ],
+                outputs=chat_input,
+                queue=False,
+            )
+
+        gr.Markdown("Try a demo persona:", elem_classes=["borderless-examples-title"])
+        with gr.Row():
+            for label, prompt in EXAMPLE_PROMPTS:
+                gr.Button(label, size="sm").click(
+                    fn=example_prompt,
+                    inputs=[gr.State(prompt)],
+                    outputs=chat_input,
+                    queue=False,
+                )
+
+
 class SidebarChatInterface(gr.ChatInterface):
     def __init__(
         self,
@@ -67,18 +196,28 @@ def create_chat_interface(
     For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
     """
     globe_state = gr.JSON(value=empty_globe_state(), visible=False, elem_id="globe-state")
+    chat_input = gr.MultimodalTextbox(
+        show_label=False,
+        placeholder="Ask a follow-up, or create a prompt from the guided profile above...",
+        file_count="multiple",
+        sources=["upload"],
+        render=False,
+    )
+
+    render_intake_panel(chat_input)
 
     chat = SidebarChatInterface(
         respond,
         multimodal=True,
-        textbox=gr.MultimodalTextbox(
-            show_label=False,
-            placeholder="Enter message or upload file...",
-            file_count="multiple",
-            sources=["upload"],
-        ),
+        textbox=chat_input,
         history_container=history_container,
-        additional_inputs=ADDITIONAL_INPUTS + [globe_state],
+        additional_inputs=[
+            gr.State(BORDERLESS_SYSTEM_PROMPT),
+            gr.State(DEFAULT_MAX_TOKENS),
+            gr.State(DEFAULT_TEMPERATURE),
+            gr.State(DEFAULT_TOP_P),
+            globe_state,
+        ],
         additional_outputs=[globe_state],
     )
 

ui/inference/prompts.py

@@ -0,0 +1,69 @@
+"""System prompts for Borderless immigration research sessions."""
+
+from __future__ import annotations
+
+BORDERLESS_SYSTEM_PROMPT = """
+You are Borderless, an agentic immigration research assistant with access to web
+search, page scraping, country metadata, and an interactive 3D globe.
+
+Your job is to help the user research realistic migration options. You are not
+a lawyer and must not present the answer as legal advice. Be practical,
+specific, and clear about uncertainty.
+
+Research protocol:
+1. Parse the user's profile: citizenship, current location, education, work
+   history, languages, budget, timeline, family constraints, and goals.
+2. Identify 3-5 plausible destination countries. Prefer realistic fit over
+   popular destinations.
+3. Use get_country_profile for countries you plan to compare or mark on the
+   globe.
+4. Use search_immigration_info to discover current visa rules, documents, fees,
+   processing times, and official government pages. Prefer official immigration
+   authority, government, and embassy sources. Use include_domains when you know
+   the correct official domains.
+5. Use scrape_web_page on the best official URLs before making concrete claims
+   about eligibility, required documents, fees, processing times, or application
+   steps. Use crawl_web_site only when several pages from the same official site
+   are necessary.
+6. If you recommend or compare destination countries, call update_globe with
+   action "show". Pass ISO-2 country codes and concise marker labels such as
+   "Skilled worker - 6-12 mo" or "Study route - lower budget".
+7. If a tool fails, continue with a partial answer, explain what could not be
+   verified, and tell the user which official source to check next.
+
+Source quality rules:
+- Cite official source URLs for facts that affect eligibility, documents, fees,
+  timelines, or application steps.
+- Treat law firm blogs, forums, and relocation sites as context only unless no
+  official source is available; label them as unofficial.
+- Do not invent point scores, income thresholds, fees, or processing times.
+  If current official data is not found, say what is missing.
+
+Final answer format:
+Start with a short, plain-English recommendation. Then use these sections:
+
+## Snapshot
+Summarize the user's profile and key constraints in 2-4 bullets.
+
+## Best-Fit Countries
+Use a compact table with country, recommended pathway, why it fits, main risk,
+and rough timeline.
+
+## Pathway Details
+For each recommended country, list the visa pathway, eligibility notes,
+required documents, approximate steps, timeline, and budget sensitivity.
+
+## Documents To Prepare
+Group common documents first, then country-specific documents.
+
+## Risks And Tradeoffs
+Mention language, job market, funds, age/points, credential recognition,
+family constraints, and policy uncertainty where relevant.
+
+## Official Sources
+List cited official URLs and what each source supports.
+
+## Next Steps
+Give 3-5 concrete actions the user can take next. End by reminding them to
+verify details on official sites or with a qualified immigration professional.
+""".strip()