refactor: improve footnote formatting

main.py

@@ -9,13 +9,11 @@ from google.genai import types
 from tools import dailymed, semantic_scholar
 
 INSTRUCTION = """\
-You are a medical research expert.
+You are a medical research expert. Provide a concise answer to the query below, using no more than 250 words.
 
-Please answer the user's query clearly and concisely, using no more than 250 words.
+Base every claim or statement strictly on the sources returned from the tool calls. For each claim, include a citation referencing the source's ID (do not include the citation in the `text` field). A claim may be supported by one or multiple sources, but only cite sources that directly support the claim. Do not add unnecessary citations.
 
-Base every claim or statement strictly on the provided sources. For each claim, include a citation referencing the source's ID (do not include the citation in the `text` field). A claim may be supported by one or multiple sources, but only cite sources that directly support the claim. Do not add unnecessary citations.
-
-You can use markdown format, such as **, to highlight the key part of the answer. But do not return the response in a markdown code block.
+You may use markdown formatting, such as **, to highlight key parts of the text. Do not return the response in a markdown code block.
 
 If none of the sources contain relevant information to answer the query, politely inform the user that an answer cannot be provided, and do not use any citations.
 
@@ -68,14 +66,21 @@ def format_output(response: str) -> tuple[str, str]:
         answer = ""
         citations = {}
         for statement in statements:
-            answer += statement["text"]
+            text = statement["text"].strip()
+            answer = (
+                answer + f"\n{text}"
+                if text.startswith("*") or text.startswith("-")
+                else answer + f" {text}"
+            )
+            citation_ids = []
             for source in statement.get("sources", []):
                 source_str = f"[{source['title']}]({source['url']})"
                 if not (citation_id := citations.get(source_str)):
                     citation_id = len(citations) + 1
                     citations[source_str] = citation_id
-                answer += f"[^{citation_id}]"
-            answer += " "
+                citation_ids.append(citation_id)
+            if citation_ids:
+                answer += " ".join(f"[^{i}]" for i in sorted(citation_ids))
     except KeyError as err:
         print(err)
         return response, ""
@@ -87,7 +92,7 @@ def format_output(response: str) -> tuple[str, str]:
 def main():
     gemini_client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
-    st.title("Ask ~~Jeeves~~ Elna")
+    st.title("Elna")
     with st.form("search", border=False):
         query = st.text_input("Your medical question")
         submit = st.form_submit_button("Ask")