Update app.py

app.py

@@ -80,11 +80,12 @@ def load_model():
 
 model, tokenizer = load_model()
 
-def analyze_dna(sequence, password):
-
-    pw_token = os.getenv('PASSWORD')
-    if password != pw_token:
-        return {"error": "Invalid password"}, ""
+def analyze_dna(username, password, sequence):
+    env_username = os.getenv('USERNAME')
+    env_password = os.getenv('PASSWORD')
+    
+    if username != env_username or password != env_password:
+        return {"error": "Invalid username or password"}, ""
 
     try:
         if not all(nucleotide in 'ACTGN' for nucleotide in sequence):
@@ -105,7 +106,7 @@ def analyze_dna(sequence, password):
         fig, ax = plt.subplots(figsize=(10, 6))
         ax.barh(top_5_labels, top_5_probs, color='skyblue')
         ax.set_xlabel('Probability')
-        ax.set_title('Top 5 Most Likely Nations of Origin')
+        ax.set_title('Top 5 Most Likely Labels')
         plt.gca().invert_yaxis()
 
         buf = io.BytesIO()
@@ -122,7 +123,11 @@ def analyze_dna(sequence, password):
 # Create a Gradio interface
 demo = gr.Interface(
     fn=analyze_dna,
-    inputs=[gr.Textbox(label="DNA Sequence"), gr.Textbox(label="Password", type="password")],
+    inputs=[
+        gr.Textbox(label="Username"),
+        gr.Textbox(label="Password", type="password"),
+        gr.Textbox(label="DNA Sequence")
+    ],
     outputs=["json", "html"]
 )