feat: rework ui

app.py

@@ -51,6 +51,7 @@ def retrieve(q, k=3):
         file_sources.append(meta)
     return docs, file_sources
 
+
 def make_prompt(q, docs):
     context = "\n\n".join(f"Title: {d['title']}\nPages: {d['pages']}" for d in docs)
     prompt = f"detailed thinking off\n"
@@ -63,7 +64,8 @@ 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\n".join(lines)
+    return "\n# Links to Documents: \n\n".join(lines)
+
 
 def build_markdown_chunks(docs):
     lines = []
@@ -72,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\n".join(lines)
+    return "\n# Extracted Source Information: \n\n".join(lines)
 
 @spaces.GPU()
 def qa_fn(question, top_k, temperature, max_tokens):
@@ -99,25 +101,29 @@ def qa_fn(question, top_k, temperature, max_tokens):
         output = output.split("Output:", 1)[1].strip()
     return output, file_links, markdown_chunks
 
-with gr.Blocks() as demo:
-    gr.Markdown("## GDPR Legal Assistant")
-    question = gr.Textbox(lines=2, label="Your Question")
-    top_k = gr.Slider(1, 7, value=4, step=1, label="Top-K Documents")
-    temperature = gr.Slider(0.1, 1.0, value=0.6, step=0.05, label="Temperature")
-    max_tokens = gr.Slider(64, 1024, value=512, step=64, label="Max Answer Length")
-    submit = gr.Button("Submit")
-    answer_output = gr.Textbox(label="Answer")
-    link_output   = gr.Markdown(label="Source Link")
-    chunks_output = gr.Markdown(label="Used Chunks")
+outputs_answer = gr.Textbox(label="Answer")
+outputs_link = gr.Markdown(label="Source Link")
+outputs_chunks = gr.Markdown(label="Used Chunks")
 
 
-    submit.click(
-      fn=qa_fn,
-      inputs=[question, top_k, temperature, max_tokens],
-      outputs=[answer_output, link_output, chunks_output],
-      show_progress="full",
-      show_progress_on=[answer_output]
-    )
-
+demo = gr.Interface(
+    fn=qa_fn,
+    inputs=[
+        gr.Textbox(lines=2, label="Your Question"),
+    ],
+    additional_inputs_accordion=[
+        gr.Slider(1, 7, value=4, step=1, label="Top-K Documents"),
+        gr.Slider(0.1, 1.0, value=0.6, step=0.05, label="Temperature"),
+        gr.Slider(64, 1024, value=512, step=64, label="Max Answer Length")
+    ],
+    outputs=[outputs_answer, outputs_link, outputs_chunks],
+    title="GDPR Legal Assistant",
+    description="Ask any question about GDPR or EDPB documents.",
+    allow_flagging="never",
+    layout="vertical",
+    fill_width = True,
+    live = True,
+)
 
-demo.queue().launch(share=True)
+if __name__ == "__main__":
+    demo.launch(share=True)