llm setting

AI_generator.py

@@ -80,7 +80,7 @@ class SqlToPlotlyStreamlit:
         return [first_prompt, second_prompt, third_prompt, fourth_prompt]
 
     @staticmethod
-    def generate_anwers() -> list:
+    def generate_anwers(llm) -> list:
         first_prompt, second_prompt, third_prompt, fourth_prompt = (
             SqlToPlotlyStreamlit.chat_prompt_template()
         )
@@ -165,7 +165,7 @@ class SqlToPlotlyStreamlit:
 
         try:
             overall_chain = SequentialChain(
-                chains=list(SqlToPlotlyStreamlit.generate_anwers()),
+                chains=list(SqlToPlotlyStreamlit.generate_anwers(self.llm)),
                 input_variables=["schema", "question"],
                 output_variables=["query", "python_graph", "json"],
                 verbose=True,
@@ -182,8 +182,6 @@ class SqlToPlotlyStreamlit:
             error_occurred = True
 
         if not error_occurred:
-            # st.write("## Generated Query")
-            # st.code(result["json"].split("```sql")[-1].split("```")[0].replace("```", ""))
             st.write("## Generated Python")
             st.code(self.generated_python_code)
 

app.py

@@ -4,7 +4,7 @@ from AI_generator import SCHEMA_STR, SqlToPlotlyStreamlit
 from dashboard import dashboard
 from dotenv import load_dotenv
 import os
-import hashlib
+
 
 load_dotenv()
 
@@ -13,7 +13,8 @@ password = os.getenv("PASSWORD")
 
 
 st.sidebar.markdown("""
-<footer style="position: fixed; bottom: 0; text-align: center; background-color: #002b36; border-top: 1px solid #ddd;">
+<footer style="position: fixed; bottom: 0; text-align: center;
+background-color: #002b36; border-top: 1px solid #ddd;">
   &copy; 2023 LPO. All rights reserved.
 </footer>
 """, unsafe_allow_html=True)