Update app.py

app.py

@@ -89,19 +89,30 @@ def chat(message, history):
     return history, ""
 #============starting extract_docx_text
 def respond(message, history):
-    answer = qa_chain.invoke(message)
-    docs = retriever.invoke(message)
-    refs = [f"Page {d.metadata.get('page', 'N/A')}" for d in docs]
-    full_answer = f"{answer}\n\n**References:**\n" + "\n".join(refs)
-    
-    # CRITICAL: Append ONLY pure dicts - no metadata, tuples, or extras
-    new_history = history + [  # Or history.append() then return history
-        {"role": "user", "content": message},
-        {"role": "assistant", "content": full_answer}
-    ]
+    word_count = len(message.strip().split())
     
-    # Clear input
-    return "", new_history  # Return cleared msg, updated history
+    # If less than 3 words, do not call LLM, just ask user to clarify
+    if word_count < 3:
+        correction_msg = "Please **clarify** or expand your question (at least 3 words)."
+        new_history = history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": correction_msg},
+        ]
+        return "", new_history
+    else:
+        answer = qa_chain.invoke(message)
+        docs = retriever.invoke(message)
+        refs = [f"Page {d.metadata.get('page', 'N/A')}" for d in docs]
+        full_answer = f"{answer}\n\n**References:**\n" + "\n".join(refs)
+        
+        # CRITICAL: Append ONLY pure dicts - no metadata, tuples, or extras
+        new_history = history + [  # Or history.append() then return history
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": full_answer}
+        ]
+        
+        # Clear input
+        return "", new_history  # Return cleared msg, updated history
 #====================
 def extract_docx_text(file_path):
     doc = Document(file_path)
@@ -326,14 +337,10 @@ with gr.Blocks(css=css) as demo:
             out = gr.HTML(label="Compliance Result")
             #out = gr.Textbox(lines=15, label="Compliance Result",elem_id="compliance-out")
             run_btn.click(check_compliance, inputs=inp, outputs=out)
-        with gr.TabItem("Compliance Check of user technical doc"):
-            gr.HTML("""
-                        <div style="color: white; background: black; padding: 20px; text-align: center; font-size: 24px;">
-                            🚧 Coming Soon 🚧
-                        </div>"""
-                   )
-        with gr.TabItem("ChatBot-NRL manual-goods"):
-                gr.Markdown("# RAG Chatbot")
+        
+        with gr.TabItem("NRL ChatBot"):
+                gr.Markdown("""# RAG Chatbot - Ask question and answer will be generated by AI from Only MANUAL FOR PROCUREMENT OF
+                            GOODS - NRL revised on 16.03.23""")
                 chatbot = gr.Chatbot(height=500)  # Defaults to messages
                 msg = gr.Textbox(placeholder="Ask a question...", label="Query")
                 submit_btn = gr.Button("Submit")
@@ -341,4 +348,22 @@ with gr.Blocks(css=css) as demo:
                 # Events
                 submit_btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
                 msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
+        with gr.TabItem("Compliance Check of user technical doc"):
+            gr.HTML("""
+                        <div style="color: white; background: black; padding: 20px; text-align: center; font-size: 24px;">
+                            🚧 Coming Soon 🚧
+                        </div>"""
+                   )
+        with gr.TabItem("Compliance Check of ATC"):
+            gr.HTML("""
+                        <div style="color: white; background: black; padding: 20px; text-align: center; font-size: 24px;">
+                            🚧 Coming Soon 🚧
+                        </div>"""
+                   )
+        with gr.TabItem("Compliance Check of Draft GeM Tender"):
+            gr.HTML("""
+                        <div style="color: white; background: black; padding: 20px; text-align: center; font-size: 24px;">
+                            🚧 Coming Soon 🚧
+                        </div>"""
+                   )
 demo.queue().launch()