Update app.py

app.py

@@ -1,17 +1,25 @@
 import os
 import pandas as pd
 
-from dotenv import load_dotenv
-
 from flask import request, Flask, jsonify
+from flask_httpauth import HTTPBasicAuth
 
 from langchain.agents.agent_types import AgentType
 from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent
 
 from langchain_openai import ChatOpenAI
 
-load_dotenv()
+auth = HTTPBasicAuth()
+
+users = {
+    "demouser": os.environ.get("PASSWD")
+}
 
+@auth.verify_password
+def verify_password(username, password):
+    if username in users and users[username] == password:
+        return username
+        
 gpt35 = ChatOpenAI(
     model_name="gpt-3.5-turbo",
     api_key=os.environ.get("OPENAI_API_KEY"),
@@ -35,6 +43,7 @@ def home():
     return 'Welcome to the DataFrame Agent'
 
 @dataframe_agent_api.post('/v1/input')
+@auth.login_required
 def predict():
 
    user_input = request.get_data(as_text=True)