Update app.py

app.py

@@ -4,8 +4,6 @@ import sqlite3
 import numpy as np
 import pandas as pd
 
-translator = pipeline(model= "saikiranmaddukuri/chat_to_sql0.17", max_new_tokens= 100)
-df= pd.read_csv("Car_sales.csv")
 
 ## Function to read SQL data
 def read_sql_query(sql, db):
@@ -18,14 +16,22 @@ def read_sql_query(sql, db):
     conn.close()
 
 
-# Streamlit app
-def main():
+
+
+i_pwd= st.sidebar.text_input("Enter password", type="password")
+
+if i_pwd=='password':
+
+    translator = pipeline(model= "saikiranmaddukuri/chat_to_sql0.17", max_new_tokens= 100)
+    df= pd.read_csv("Car_sales.csv")
+
     st.title("Conversational AI: Text-2-SQL")
 
     column_info = df.dtypes.to_dict()
     column_df = pd.DataFrame(column_info.items(), columns=['Column Name', 'Data Type'])
     st.sidebar.table(column_df)
     
+
     
     create_table_statement= "CREATE TABLE  car_sales (Manufacturer text, Model text, Sales_in_thousands float, __year_resale_value float, Vehicle_type text, Price_in_thousands float, Engine_size float, Horsepower float, Wheelbase float, Width float, Length float, Curb_weight float, Fuel_capacity float, Fuel_efficiency float, Latest_Launch text, Power_perf_factor float)"
 
@@ -51,5 +57,6 @@ def main():
             # Step 5: Display final output
             #st.write("Final Output:", final_output)
 
-if __name__ == "__main__":
-    main()
+else:
+    st.warning("Enter correct password!")
+