Update app.py

app.py

@@ -351,24 +351,19 @@ def create_app():
             """
         )
 
-        # Selected answer language
         lang = gr.Dropdown(
             ["auto"] + list(LANG_CODES.keys()),
             value="auto",
             label="Answer Language",
-            scale=1,
         )
 
-        # Chat history (messages format)
         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 box
         status = gr.HTML()
 
-        # ---- LOADER BOX ----
-        with gr.Box():
+        with gr.Group():
             gr.Markdown("### Load PDF or URL")
 
             with gr.Row():
@@ -386,7 +381,6 @@ def create_app():
             load_pdf_btn = gr.Button("Load PDF", variant="secondary")
             load_url_btn = gr.Button("Load URL", variant="secondary")
 
-        # Example Questions
         gr.Markdown("### Example Questions")
         with gr.Row():
             ex1 = gr.Button("Summarize the document")
@@ -395,14 +389,12 @@ def create_app():
             ex4 = gr.Button("List the conclusions")
             ex5 = gr.Button("Explain in simple terms")
 
-        # State for tracking which PDF is loaded
         pdf_id_state = gr.State("")
 
-        # ---- Reset chat when new PDF is loaded ----
+        # --- Reset chat when new PDF is loaded ---
         def reset_chat_on_new_pdf(message, old_id):
-            new_id = message  
-            if new_id != old_id:
-                return [], new_id   
+            if message != old_id:
+                return [], message
             return gr.skip(), old_id
 
         load_pdf_btn.click(load_pdf_ui, [pdf, lang], status).then(
@@ -413,32 +405,21 @@ def create_app():
             reset_chat_on_new_pdf, [status, pdf_id_state], [chat, pdf_id_state]
         )
 
-        # ---- ASK QUESTION ----
+        # --- ASK ---
         def chat_wrapper(question, history, lang):
-            if question.strip() == "":
+            if not question.strip():
                 return "", history
-
             ans = answer_question(question, lang)
 
-            # Correct Chatbot message format
             history.append({"role": "user", "content": question})
             history.append({"role": "assistant", "content": ans})
 
             return "", history
 
-        ask_btn.click(
-            chat_wrapper,
-            [user_question, chat, lang],
-            [user_question, chat]
-        )
+        ask_btn.click(chat_wrapper, [user_question, chat, lang], [user_question, chat])
+        user_question.submit(chat_wrapper, [user_question, chat, lang], [user_question, chat])
 
-        user_question.submit(
-            chat_wrapper,
-            [user_question, chat, lang],
-            [user_question, chat]
-        )
-
-        # ---- EXAMPLES ----
+        # 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)
@@ -449,6 +430,7 @@ def create_app():
 
 
 
+
 demo = create_app()
 
 if __name__ == "__main__":