Update app.py

app.py

@@ -276,7 +276,17 @@ Answer in English:
 """.strip()
     return prompt
 
-
+def highlight_sources(retrieved):
+    html = "<h4>📚 Source Passages</h4>"
+    for i, (chunk, score) in enumerate(retrieved):
+        html += f"""
+        <div style='padding:10px; background:#eef6ff; margin:8px 0;'>
+            <b>[{i+1}] Score = {1-score:.3f}</b><br>
+            {chunk[:400]}...
+        </div>
+        """
+    return html
+    
 def answer_question(question: str, language: str):
     if question.strip() == "":
         return "Please enter a question.", ""
@@ -340,93 +350,77 @@ def load_url_ui(url: str, language: str):
 
 def create_app():
     with gr.Blocks() as demo:
-        
+        gr.Markdown("<h1 style='text-align:center;'> Multilingual Chat with PDF / URL</h1>")
         gr.Markdown(
-            """
-            <h1 style='text-align:center;'>Multilingual Chat with PDF / URL</h1>
-            <p style='text-align:center;'>
-                Load a PDF or Article and ask questions. The app retrieves relevant chunks 
-                and answers using a local LLM with multilingual support.
-            </p>
-            """
+            "Upload a PDF or paste a URL (PDF or article). "
+            "The app creates embeddings, retrieves the most relevant chunks, "
+            "and answers using a small local LLM."
         )
 
+        # Language selector
         lang = gr.Dropdown(
             ["auto"] + list(LANG_CODES.keys()),
             value="auto",
             label="Answer Language",
         )
 
-        chat = gr.Chatbot(label="Chat History", height=350)
-        user_question = gr.Textbox(label="Ask a question", placeholder="Type your question here...", lines=2)
-        ask_btn = gr.Button("Ask", variant="primary")
-
-        status = gr.HTML()
-
-        with gr.Group():
-            gr.Markdown("### Load PDF or URL")
-
-            with gr.Row():
-                pdf = gr.File(
-                    label="Upload PDF",
-                    file_types=[".pdf"],
-                    height=70,
-                    interactive=True
-                )
-                url = gr.Textbox(
-                    label="Or paste URL",
-                    placeholder="https://arxiv.org/pdf/...  or research article",
-                )
-
-            load_pdf_btn = gr.Button("Load PDF", variant="secondary")
-            load_url_btn = gr.Button("Load URL", variant="secondary")
+        # ---------------------------
+        #   DOCUMENT LOADING AREA
+        # ---------------------------
+        gr.Markdown("### Load PDF or Article")
 
-        gr.Markdown("### Example Questions")
         with gr.Row():
-            ex1 = gr.Button("Summarize the document")
-            ex2 = gr.Button("What are the key findings?")
-            ex3 = gr.Button("Explain the methodology")
-            ex4 = gr.Button("List the conclusions")
-            ex5 = gr.Button("Explain in simple terms")
-
-        pdf_id_state = gr.State("")
-
-        # --- Reset chat when new PDF is loaded ---
-        def reset_chat_on_new_pdf(message, old_id):
-            if message != old_id:
-                return [], message
-            return gr.skip(), old_id
-
-        load_pdf_btn.click(load_pdf_ui, [pdf, lang], status).then(
-            reset_chat_on_new_pdf, [status, pdf_id_state], [chat, pdf_id_state]
+            pdf = gr.File(
+                label="Upload PDF",
+                file_types=[".pdf"],
+                height=70   # smaller box
+            )
+            pdf_status = gr.HTML()
+
+        # Auto load PDF on upload
+        pdf.upload(load_pdf_ui, [pdf, lang], pdf_status)
+
+        gr.Markdown("---")
+
+        # URL input
+        url = gr.Textbox(
+            label="Enter URL",
+            placeholder="https://example.com/article.pdf",
         )
+        url_status = gr.HTML()
 
-        load_url_btn.click(load_url_ui, [url, lang], status).then(
-            reset_chat_on_new_pdf, [status, pdf_id_state], [chat, pdf_id_state]
-        )
+        # Auto load URL on pressing Enter
+        url.submit(load_url_ui, [url, lang], url_status)
+
+        # ---------------------------
+        #   CHAT AREA
+        # ---------------------------
+        gr.Markdown("### Ask Questions About the Loaded Document")
 
-        # --- ASK ---
-        def chat_wrapper(question, history, lang):
-            if not question.strip():
-                return "", history
-            ans = answer_question(question, lang)
+        q = gr.Textbox(label="Your Question")
+        a = gr.HTML(label="Answer")
+        cits = gr.HTML(label="Source Passages")
 
-            history.append({"role": "user", "content": question})
-            history.append({"role": "assistant", "content": ans})
+        ask_btn = gr.Button("Ask", variant="primary")
+        ask_btn.click(answer_question, [q, lang], [a, cits])
 
-            return "", history
+        # Pre-set example question buttons
+        gr.Markdown("### Example Questions")
+        with gr.Row():
+            b1 = gr.Button("Summarize the document")
+            b2 = gr.Button("What are the key findings?")
+            b3 = gr.Button("Explain the methodology used")
+            b4 = gr.Button("What are the main limitations?")
+            b5 = gr.Button("What is the conclusion of this paper?")
 
-        ask_btn.click(chat_wrapper, [user_question, chat, lang], [user_question, chat])
-        user_question.submit(chat_wrapper, [user_question, chat, lang], [user_question, chat])
+        b1.click(lambda: "Summarize the document.", None, q)
+        b2.click(lambda: "What are the key findings?", None, q)
+        b3.click(lambda: "Explain the methodology used in this study.", None, q)
+        b4.click(lambda: "What are the main limitations of this study?", None, q)
+        b5.click(lambda: "What is the conclusion of this paper?", None, q)
 
-        # Examples
-        ex1.click(lambda: "Summarize the document", None, user_question)
-        ex2.click(lambda: "What are the key findings?", None, user_question)
-        ex3.click(lambda: "Explain the methodology used", None, user_question)
-        ex4.click(lambda: "List the main conclusions", None, user_question)
-        ex5.click(lambda: "Explain in simple terms", None, user_question)
+        return demo
 
-    return demo