Update app.py

app.py

@@ -387,7 +387,7 @@ def style_metric_container(label, value):
     """, unsafe_allow_html=True)
 
 # --- OpenAI GPT-3 Assistant ---
-def generate_gpt_response(prompt):
+def generate_gpt_response(prompt, dataset):
     """
     First look up the dataset for relevant information. If no matches are found,
     generate a GPT response.
@@ -444,7 +444,7 @@ def generate_gpt_response(prompt):
 
     return assistant_reply
 
-def create_assistant_section():
+def create_assistant_section(dataset):
     st.markdown("""
         <div style='background-color: #f8f9fa; padding: 1.5rem; border-radius: 10px; margin-bottom: 1rem;'>
             <h2 style='color: #1E1E1E; margin-top: 0;'>🤖 Car Shopping Assistant</h2>
@@ -461,7 +461,7 @@ def create_assistant_section():
     if prompt:
         try:
             # Use OpenAI API to generate response
-            response = generate_gpt_response(prompt)
+            response = generate_gpt_response(prompt, dataset)
             st.session_state.assistant_responses.append(response)
         except Exception as e:
             response = f"Sorry, I encountered an error: {str(e)}"
@@ -632,7 +632,7 @@ def predict_with_ranges(inputs, model, label_encoders):
         'max_price': max_price
     }
 # --- Main Application ---
-def main(model, label_encoders):
+def main(model, label_encoders, dataset):
     col1, col2 = st.columns([2, 1])
 
     with col1:
@@ -684,7 +684,7 @@ def main(model, label_encoders):
             st.warning("No graph generated. Please check your data or selection.")
 
     with col2:
-        create_assistant_section()
+        create_assistant_section(dataset)
 
 if __name__ == "__main__":
     try:
@@ -696,7 +696,7 @@ if __name__ == "__main__":
         inspect_model_features(model)
         
         # Call the main function
-        main(model, label_encoders)
+        main(model, label_encoders, original_data)
     except Exception as e:
         st.error(f"Error loading data or models: {str(e)}")
         st.stop()