feat: add headings

app.py

@@ -64,7 +64,7 @@ def build_markdown_links(file_input):
         url = f"{github_base_url}/{item['directory']}/{item['source']}"
         line = f"**Source {idx}:** [{item['source']}]({url}) on page {item['page']}"
         lines.append(line)
-    return "\n# Links to Documents: \n\n".join(lines)
+    return "\n\n".join(lines)
 
 
 def build_markdown_chunks(docs):
@@ -74,7 +74,7 @@ def build_markdown_chunks(docs):
         page = d['title']['page']
         text = d['pages']
         lines.append(f"**Chunk {idx}:** {title} on page {page}\n\n{text}")
-    return "\n# Extracted Source Information: \n\n".join(lines)
+    return "\n\n".join(lines)
 
 @spaces.GPU()
 def qa_fn(question, top_k, temperature, max_tokens):
@@ -99,13 +99,13 @@ def qa_fn(question, top_k, temperature, max_tokens):
         output += tok
     if "Output:" in output:
         output = output.split("Output:", 1)[1].strip()
-    return output, file_links, markdown_chunks
+    return "\n# Generated Answer\n", output,"\n# Used Documents\n",  file_links, "\n# Used Context\n", markdown_chunks
 
-heading_answer  = gr.Markdown("# Generated Answer")
+heading_answer  = gr.Markdown(label="Answer Heading")
 outputs_answer = gr.Textbox(label="Answer")
 outputs_link = gr.Markdown(label="Source Link")
-heading_links  = gr.Markdown("# Links to Documents")
-heading_chunks = gr.Markdown("# Extracted Source Information")
+heading_links  = gr.Markdown(label="Links Heading")
+heading_chunks = gr.Markdown(label="Chunks Heading")
 outputs_chunks = gr.Markdown(label="Used Chunks")
 
 demo = gr.Interface(
@@ -120,8 +120,11 @@ demo = gr.Interface(
     ],
     additional_inputs_accordion="Advanced Options",
     outputs=[
+        heading_answer,
         outputs_answer,
+        heading_links,
         outputs_link,
+        heading_chunks,
         outputs_chunks
     ],
     title="GDPR Legal Assistant",