Update app.py

first to try adding password

app.py

@@ -167,6 +167,12 @@ import shutil
 # Initialize the Groq client
 client = Groq(api_key=os.environ["keko"])
 
+def check_password(password):
+    correct_password = "zoo"  # Set your desired password here
+    return password == correct_password
+
+
+
 def save_audio(audio_file):
     if audio_file is None:
         return None, "No audio file provided."
@@ -219,50 +225,79 @@ def generate_text(text, prompt_type):
     
     return response.choices[0].message.content
 
+
+
+
+
 # Create the Gradio interface using Blocks
 with gr.Blocks(title="Audio Recorder, Transcriber, and Text Generator") as demo:
     gr.Markdown("# Audio Recorder, Transcriber, and Text Generator")
-    gr.Markdown("Record audio, transcribe it, and optionally generate text based on the transcription.")
-    
-    with gr.Row():
-        audio_input = gr.Audio(sources=["microphone", "upload"], type="filepath", label="Record or Upload Audio")
-    
-    with gr.Row():
-        save_btn = gr.Button("Save Audio")
-        transcribe_btn = gr.Button("Transcribe Audio")
-    
-    with gr.Row():
-        audio_output = gr.Audio(label="Saved Audio (MP3)", format="mp3")
-        save_msg = gr.Textbox(label="Save Status")
-    
-    transcription_output = gr.Textbox(label="Transcription", show_copy_button=True)
-    
-    with gr.Row():
-        prompt_type = gr.Dropdown(
-            choices=["dic_clean","summarize", "create_qa", "analyze"],
-            label="Select Text Generation Type",
-            value="dic_clean"
+    gr.Markdown("Enter the correct password to access the application.")
+    
+    password_input = gr.Textbox(type="password", label="Enter Password")
+    login_button = gr.Button("Login")
+    login_message = gr.Markdown()
+
+    with gr.Column(visible=False) as main_interface:
+        gr.Markdown("Record audio, transcribe it, and optionally generate text based on the transcription.")
+        
+        with gr.Row():
+            audio_input = gr.Audio(sources=["microphone", "upload"], type="filepath", label="Record or Upload Audio")
+        
+        with gr.Row():
+            save_btn = gr.Button("Save Audio")
+            transcribe_btn = gr.Button("Transcribe Audio")
+        
+        with gr.Row():
+            audio_output = gr.Audio(label="Saved Audio (MP3)", format="mp3")
+            save_msg = gr.Textbox(label="Save Status")
+        
+        transcription_output = gr.Textbox(label="Transcription", show_copy_button=True)
+        
+        with gr.Row():
+            prompt_type = gr.Dropdown(
+                choices=["dic_clean","summarize", "create_qa", "analyze"],
+                label="Select Text Generation Type",
+                value="dic_clean"
+            )
+            generate_btn = gr.Button("Generate Text")
+        
+        generated_text_output = gr.Textbox(label="Generated Text", show_copy_button=True)
+        
+        save_btn.click(
+            save_audio,
+            inputs=[audio_input],
+            outputs=[audio_output, save_msg]
         )
-        generate_btn = gr.Button("Generate Text")
-    
-    generated_text_output = gr.Textbox(label="Generated Text", show_copy_button=True)
-    
-    save_btn.click(
-        save_audio,
-        inputs=[audio_input],
-        outputs=[audio_output, save_msg]
-    )
-    
-    transcribe_btn.click(
-        transcribe_audio,
-        inputs=[audio_input],
-        outputs=[transcription_output]
-    )
-    
-    generate_btn.click(
-        generate_text,
-        inputs=[transcription_output, prompt_type],
-        outputs=[generated_text_output]
+        
+        transcribe_btn.click(
+            transcribe_audio,
+            inputs=[audio_input],
+            outputs=[transcription_output]
+        )
+        
+        generate_btn.click(
+            generate_text,
+            inputs=[transcription_output, prompt_type],
+            outputs=[generated_text_output]
+        )
+
+    def login(password):
+        if check_password(password):
+            return {
+                main_interface: gr.Column(visible=True),
+                login_message: gr.Markdown("Login successful. You can now use the application.")
+            }
+        else:
+            return {
+                main_interface: gr.Column(visible=False),
+                login_message: gr.Markdown("Incorrect password. Please try again.")
+            }
+
+    login_button.click(
+        login,
+        inputs=[password_input],
+        outputs=[main_interface, login_message]
     )
 
 # Launch the interface