Update app.py

app.py

@@ -1,9 +1,6 @@
 import openai
 import gradio as gr
 
-# Set your password
-my_password = "patti-ai-kb"
-
 # Initialize OpenAI API key
 openai.api_key = "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"
 
@@ -22,19 +19,12 @@ def generate_text(prompt):
     return response.choices[0].text.strip()
 
 # Create Gradio interface
-password_text = gr.Textbox(label="Enter password")
-input_text = gr.Textbox(label="Enter prompt")
-output_text = gr.Textbox(label="AI response")
-
-def generate_with_password(prompt, password):
-    if password != "my_password":
-        return "Invalid password. Please try again."
-    response = generate_text(prompt)
-    return response
-
-demo = gr.Interface(
-    fn=generate_with_password,
-    inputs=[input_text, password_text],
+    password = "mypassword"
+    input_text = gr.inputs.Textbox(label="Enter prompt", placeholder="Enter your question here")
+    output_text = gr.outputs.Textbox(label="AI response", placeholder="Waiting for input...")
+    demo = gr.Interface(
+    fn=generate_text,
+    inputs=input_text,
     outputs=output_text,
     title="AI Chatbot for PlanetTogether Knowledge Base",
     description="Ask a question about the PlanetTogether APS:",
@@ -43,5 +33,20 @@ demo = gr.Interface(
     theme="default"
 )
 
+# Add password field to header
+    demo.add_header("Enter Password", type="password")
+
+# Function to check password and show/hide inputs/output
+    def check_password(password_input):
+    if password_input == password:
+        demo.show_input()
+        demo.show_output()
+    else:
+        demo.hide_input()
+        demo.hide_output()
+
+# Set the callback function for password field
+demo.set_config(callback=check_password)
+
 # Launch demo
-demo.launch()
+demo.launch(share=True)