Update app.py

app.py

@@ -328,20 +328,20 @@ with gr.Blocks(theme="gradio/soft") as demo:
 
     with gr.Row():
         submit_btn = gr.Button("🚀 Submit", variant="primary")
-        #if not PUBLIC_MODE:
-        rebuild_btn = gr.Button("🔁 Rebuild Index")
-        rebuild_glossary_btn = gr.Button("📘 Rebuild Glossary")
-        reset_btn = gr.Button("🧹 Reset FAISS Cache (Full Rebuild)")
-        clear_btn = gr.Button("🗑️ Clear Index Only")
+        if not PUBLIC_MODE:
+            rebuild_btn = gr.Button("🔁 Rebuild Index")
+            rebuild_glossary_btn = gr.Button("📘 Rebuild Glossary")
+            reset_btn = gr.Button("🧹 Reset FAISS Cache (Full Rebuild)")
+            clear_btn = gr.Button("🗑️ Clear Index Only")
 
     submit_btn.click(fn=chat_answer, inputs=[query_box], outputs=output_box)
     query_box.submit(fn=chat_answer, inputs=[query_box], outputs=output_box)
 
-    #if not PUBLIC_MODE:
-    rebuild_btn.click(fn=rebuild_index, outputs=output_box)
-    rebuild_glossary_btn.click(fn=rebuild_glossary, outputs=output_box)
-    reset_btn.click(fn=reset_faiss_cache, outputs=output_box)
-    clear_btn.click(fn=clear_index, outputs=output_box)
+    if not PUBLIC_MODE:
+        rebuild_btn.click(fn=rebuild_index, outputs=output_box)
+        rebuild_glossary_btn.click(fn=rebuild_glossary, outputs=output_box)
+        reset_btn.click(fn=reset_faiss_cache, outputs=output_box)
+        clear_btn.click(fn=clear_index, outputs=output_box)
 
 
 # ----------------------------------------------------------