Update app.py

app.py

@@ -33,7 +33,7 @@ class QADataGenerator:
         self._setup_providers()
         self._setup_input_handlers()
         self._initialize_session_state()
-        # Prompt template with a dynamic {num_examples} parameter and escaped curly braces
+        # Updated prompt template with dynamic {num_examples} parameter and escaped curly braces
         self.custom_prompt_template: str = (
             "You are an expert in extracting question and answer pairs from documents. "
             "Generate {num_examples} Q&A pairs from the following data, formatted as a JSON list of dictionaries. "
@@ -90,28 +90,11 @@ class QADataGenerator:
             "api_key": "",
             "inputs": [],       # List to store input sources
             "qa_pairs": None,   # Generated Q&A pairs output
-            "error_logs": [],   # To store error messages
+            "error_logs": [],   # To store any error messages
         }
         for key, value in defaults.items():
             if key not in st.session_state:
                 st.session_state[key] = value
-
-        # Pre-populate configuration from URL query parameters (if provided)
-        params = st.query_params
-        if "provider" in params:
-            st.session_state.config["provider"] = params["provider"][0]
-        if "model" in params:
-            st.session_state.config["model"] = params["model"][0]
-        if "temperature" in params:
-            try:
-                st.session_state.config["temperature"] = float(params["temperature"][0])
-            except ValueError:
-                pass
-        if "num_examples" in params:
-            try:
-                st.session_state.config["num_examples"] = int(params["num_examples"][0])
-            except ValueError:
-                pass
     
     def log_error(self, message: str) -> None:
         """Log an error message to session state and display it."""
@@ -302,44 +285,24 @@ class QADataGenerator:
 # ============ UI Components ============
 
 def config_ui(generator: QADataGenerator) -> None:
-    """Display configuration options in the sidebar and update URL query parameters."""
+    """Display configuration options in the sidebar."""
     with st.sidebar:
         st.header("Configuration")
-        # Retrieve any query parameters from the URL
-        params = st.query_params
-        default_provider = params.get("provider", ["OpenAI"])[0]
-        default_model = params.get("model", ["gpt-4-turbo"])[0]
-        default_temperature = float(params.get("temperature", [DEFAULT_TEMPERATURE])[0])
-        default_num_examples = int(params.get("num_examples", [3])[0])
-        
-        provider_options = list(generator.providers.keys())
-        provider = st.selectbox("Select Provider", provider_options, 
-                                index=provider_options.index(default_provider) if default_provider in provider_options else 0)
+        provider = st.selectbox("Select Provider", list(generator.providers.keys()))
         st.session_state.config["provider"] = provider
         provider_cfg = generator.providers[provider]
         
-        model_options = provider_cfg["models"]
-        model = st.selectbox("Select Model", model_options,
-                             index=model_options.index(default_model) if default_model in model_options else 0)
+        model = st.selectbox("Select Model", provider_cfg["models"])
         st.session_state.config["model"] = model
         
-        temperature = st.slider("Temperature", 0.0, 1.0, default_temperature)
+        temperature = st.slider("Temperature", 0.0, 1.0, DEFAULT_TEMPERATURE)
         st.session_state.config["temperature"] = temperature
         
-        num_examples = st.number_input("Number of Q&A Pairs", min_value=1, max_value=10, 
-                                       value=default_num_examples, step=1)
+        num_examples = st.number_input("Number of Q&A Pairs", min_value=1, max_value=10, value=3, step=1)
         st.session_state.config["num_examples"] = num_examples
         
         api_key = st.text_input(f"{provider} API Key", type="password")
         st.session_state.api_key = api_key
-        
-        # Update the URL query parameters for sharing/pre-populating configuration
-        st.experimental_set_query_params(
-            provider=st.session_state.config["provider"],
-            model=st.session_state.config["model"],
-            temperature=st.session_state.config["temperature"],
-            num_examples=st.session_state.config["num_examples"],
-        )
 
 def input_ui(generator: QADataGenerator) -> None:
     """Display input data source options using tabs."""
@@ -388,7 +351,7 @@ def input_ui(generator: QADataGenerator) -> None:
             st.success("Database input added!")
 
 def output_ui(generator: QADataGenerator) -> None:
-    """Display the generated Q&A pairs and provide download options (JSON and CSV)."""
+    """Display the generated Q&A pairs and provide download options."""
     st.subheader("Q&A Pairs Output")
     if st.session_state.qa_pairs:
         st.write("### Generated Q&A Pairs")