Update app.py

app.py

@@ -672,60 +672,65 @@ class ChatBot:
             return f"Error occurred: {str(e)}"
 
     async def get_response(self,
-                          query: str,
-                          history: List[List[str]],
-                          num_results: int,
-                          max_chars: int,
-                          score_threshold: float,
-                          temperature: float,
-                          scoring_method: str,
-                          selected_engines: List[str],
-                          safe_search: str,
-                          language: str,
-                          force_web_search: bool = False) -> str:
-        """Determine query type and route to appropriate handler with context."""
-        logger.info(f'Processing query: {query}')
-        try:
-            # Update conversation history
-            formatted_history = self.format_chat_history(history)
-            logger.info(f'Current conversation context:\n{formatted_history}')
-
-            # If force_web_search is True, skip query type determination
-            if force_web_search:
-                logger.info('Force web search mode enabled - bypassing query type determination')
-                query_type = QueryType.WEB_SEARCH
-            else:
-                # Determine query type with context
-                query_type = await determine_query_type(query, history, temperature)
-            
-            if query_type == QueryType.KNOWLEDGE_BASE and not force_web_search:
-                logger.info('Using knowledge base to answer query')
-                response = await process_knowledge_base_query(
-                    query=query,
-                    chat_history=history,
-                    temperature=temperature
-                )
-            else:
-                logger.info('Using web search to answer query')
-                response = await self.get_search_results(
-                    query=query,
-                    history=history,
-                    num_results=num_results,
-                    max_chars=max_chars,
-                    score_threshold=score_threshold,
-                    temperature=temperature,
-                    scoring_method=scoring_method,
-                    selected_engines=selected_engines,
-                    safe_search=safe_search,
-                    language=language
-                )
-
-            logger.info(f'Generated response type: {query_type}')
-            return response
+                      query: str,
+                      history: List[List[str]],
+                      num_results: int,
+                      max_chars: int,
+                      score_threshold: float,
+                      temperature: float,
+                      scoring_method: str,
+                      selected_engines: List[str],
+                      safe_search: str,
+                      language: str,
+                      force_web_search: bool = False) -> str:
+    """Determine query type and route to appropriate handler with context."""
+    logger.info(f'Processing query: {query}')
+    try:
+        # Update conversation history
+        formatted_history = self.format_chat_history(history)
+        logger.info(f'Current conversation context:\n{formatted_history}')
+
+        # Convert the force_web_search radio button value to boolean
+        force_web_search = force_web_search == "Web Search Only"
+        logger.info(f'Force web search mode: {force_web_search}')
+
+        # If force_web_search is True, skip query type determination
+        if force_web_search:
+            logger.info('Force web search mode enabled - bypassing query type determination')
+            query_type = QueryType.WEB_SEARCH
+        else:
+            # Determine query type with context
+            query_type = await determine_query_type(query, history, temperature)
+            logger.info(f'Query type determined as: {query_type}')
+        
+        if query_type == QueryType.KNOWLEDGE_BASE and not force_web_search:
+            logger.info('Using knowledge base to answer query')
+            response = await process_knowledge_base_query(
+                query=query,
+                chat_history=history,
+                temperature=temperature
+            )
+        else:
+            logger.info('Using web search to answer query')
+            response = await self.get_search_results(
+                query=query,
+                history=history,
+                num_results=num_results,
+                max_chars=max_chars,
+                score_threshold=score_threshold,
+                temperature=temperature,
+                scoring_method=scoring_method,
+                selected_engines=selected_engines,
+                safe_search=safe_search,
+                language=language
+            )
 
-        except Exception as e:
-            logger.error(f'Error in get_response: {e}')
-            return f"I apologize, but I encountered an error: {str(e)}"
+        logger.info(f'Generated response type: {query_type}')
+        return response
+
+    except Exception as e:
+        logger.error(f'Error in get_response: {e}')
+        return f"I apologize, but I encountered an error: {str(e)}"
 
     def chat(self, 
              message: str,