UI update

app.py

@@ -249,13 +249,13 @@ Answer:"""
             # Basic client-side validation for immediate feedback (redundant but good UX)
             if not api_key or not api_key.strip() or not api_key.startswith("sk-"):
                 return "<div class='error-message'><span class='error-icon'></span>Please provide a valid OpenAI API key (starting with 'sk-'). <a href='https://platform.openai.com/api-keys' target='_blank'>OpenAI</a>.</div>"
-            if not state or state is None or state == "Error: Critical failure loading states": # Added check for error state
+            if not state or state is None or state == "Error: Critical failure loading states" or state == "Error: States unavailable": # Added check for error state
                 return "<div class='error-message'><span class='error-icon'></span>Please select a valid state from the list.</div>"
             if not query or not query.strip():
                 return "<div class='error-message'><span class='error-icon'></span>Please enter your question in the text box.</div>"
 
-            # Call the core processing logic
-            result = self.assistant.process_query(query=query, state=state, openai_api_key=api_key)
+            # Call the core processing logic (using self.process_query)
+            result = self.process_query(query=query, state=state, openai_api_key=api_key)
             answer = result.get("answer", "<div class='error-message'><span class='error-icon'>⚠️</span>An unexpected error occurred.</div>")
 
             # Check if the answer already contains an error message
@@ -266,16 +266,16 @@ Answer:"""
                 return f"<div class='animated-output-content'>{formatted_response_content}</div>"
 
         try:
-            available_states_list = self.assistant.get_states()
+            available_states_list = self.get_states() # Call self.get_states() directly
             print(f"DEBUG: States loaded for selection: {available_states_list}")
             # Ensure radio_choices is not empty if there's an error and handling the error message
-            self.state_choices = available_states_list if available_states_list and "Error" not in available_states_list[0] else ["Error: States unavailable"]
+            radio_choices = available_states_list if available_states_list and "Error" not in available_states_list[0] else ["Error: States unavailable"]
             # Set initial value only if there are valid choices
-            self.initial_state_value = self.state_choices[0] if self.state_choices and "Error" not in self.state_choices[0] else None
+            initial_value_radio = radio_choices[0] if radio_choices and "Error" not in radio_choices[0] else None
         except Exception as e:
             print(f"DEBUG: Error loading states for selection: {e}")
-            self.state_choices = ["Error: Critical failure loading states"]
-            self.initial_state_value = None
+            radio_choices = ["Error: Critical failure loading states"]
+            initial_value_radio = None
 
         example_queries_base = [
             ["What are the rules for security deposit returns?", "California"],
@@ -285,11 +285,11 @@ Answer:"""
             ["What is an implied warranty of habitability?", "Illinois"]
         ]
         example_queries = []
-        if self.state_choices and "Error" not in self.state_choices[0] and len(self.state_choices) > 0:
-            loaded_states_set = set(self.state_choices)
+        if radio_choices and "Error" not in radio_choices[0] and len(radio_choices) > 0:
+            loaded_states_set = set(radio_choices)
             example_queries = [ex for ex in example_queries_base if ex[1] in loaded_states_set]
             if not example_queries: # Fallback if no matching examples found
-                example_queries.append(["What basic rights do tenants have?", self.state_choices[0]])
+                example_queries.append(["What basic rights do tenants have?", radio_choices[0]])
         else: # If states failed to load, provide a generic example
             example_queries.append(["What basic rights do tenants have?", "California"])
 
@@ -856,7 +856,8 @@ Answer:"""
         }}
         """
 
-        # Ensure theme_mode="light" is explicitly set
+        # MODIFIED LINE: Added theme_mode="light"
+        # This will work after upgrading Gradio
         with gr.Blocks(css=custom_css, title="Landlord-Tenant Rights Assistant", theme_mode="light") as demo:
             with gr.Group(elem_classes="app-header-wrapper"):
                 gr.Markdown(
@@ -911,8 +912,8 @@ Answer:"""
                             with gr.Column(elem_classes="input-field", scale=1):
                                 state_input = gr.Radio(
                                     label="Select State",
-                                    choices=self.state_choices, # Use self.state_choices
-                                    value=self.initial_state_value, # Use self.initial_state_value
+                                    choices=radio_choices,
+                                    value=initial_value_radio,
                                     elem_classes=["input-field-group", "gradio-radio-custom"],
                                     interactive=True
                                 )
@@ -968,7 +969,7 @@ Answer:"""
             
             clear_button.click(
                 fn=lambda: (
-                    "", "", self.initial_state_value, "<div class='placeholder'>Inputs cleared. Ready for your next question.</div>"
+                    "", "", initial_value_radio, "<div class='placeholder'>Inputs cleared. Ready for your next question.</div>"
                 ),
                 inputs=[],
                 outputs=[api_key_input, query_input, state_input, output]
@@ -977,6 +978,7 @@ Answer:"""
             return demo
 
 
+
 # --- Main Execution Block (UNCHANGED from original logic) ---
 if __name__ == "__main__":
     logging.info("Starting Landlord-Tenant Rights Bot application...")