update app.py

app.py

@@ -10,7 +10,6 @@ import torch
 import re
 import transformers
 import spaces
-import requests
 
 # Initialize embeddings and ChromaDB
 model_name = "sentence-transformers/all-mpnet-base-v2"
@@ -92,19 +91,26 @@ def chat(query, history=None):
 def clear_input():
     return "",  # Return empty string to clear input field
 
+# Function to show/hide components based on login status
+def on_login(success):
+    return gr.update(visible=success), gr.update(visible=success)
+
 # Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("## RAG Chatbot")
     gr.Markdown("Ask a question and get answers based on retrieved documents.")
     
-    input_box = gr.Textbox(label="Enter your question", placeholder="Type your question here...")
-    submit_btn = gr.Button("Submit")
-    chat_history = gr.Chatbot(label="Chat History")
+    # Hidden components initially
+    input_box = gr.Textbox(label="Enter your question", placeholder="Type your question here...", visible=False)
+    submit_btn = gr.Button("Submit", visible=False)
+    chat_history = gr.Chatbot(label="Chat History", visible=False)
 
     # Sign-In Button
     login_btn = gr.Button("Sign In with HF")
     login_btn.click(lambda: oauth_login(), outputs=None)  # Redirect user for OAuth login
-
+    
+    # Show components after login
+    submit_btn.click(on_login, inputs=[True], outputs=[input_box, submit_btn, chat_history])
     submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
 
 interface.launch()