updated app.py

app.py

@@ -72,10 +72,15 @@ def test_rag(query):
 
 # OAuth Login Functionality
 def oauth_login():
-    client_id = os.getenv("OAUTH_CLIENT_ID")
-    redirect_uri = f"https://{os.getenv('SPACE_HOST')}/login/callback"
-    state = "random_string"  # Ideally generate a secure random string
-    login_url = f"https://huggingface.co/oauth/authorize?redirect_uri={redirect_uri}&scope=openid%20profile&client_id={client_id}&state={state}"
+    params = {
+        'client_id': CLIENT_ID,
+        'response_type': 'code',
+        'redirect_uri': REDIRECT_URI,
+        'response_mode': 'query',
+        'scope': SCOPE,
+        'state': 'random_state_string'  # Optional: Use for security
+    }
+    login_url = f"{AUTH_URL}?{urlencode(params)}"
     return login_url
 
 # Define the Gradio interface
@@ -95,7 +100,6 @@ def clear_input():
 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.")
@@ -110,7 +114,11 @@ with gr.Blocks() as interface:
     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])
+    def show_components():
+        return gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+
+    # After a successful login, show the input box and buttons
+    submit_btn.click(show_components, outputs=[input_box, submit_btn, chat_history])
     submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
 
 interface.launch()