update

app.py

@@ -13,6 +13,7 @@ import warnings
 import socket
 import tempfile
 import json # New import
+import re
 
 from google.cloud import bigquery
 
@@ -68,6 +69,18 @@ else:
 def create_field_mode_ui():
     """Creates the Gradio UI for the offline Field Mode."""
 
+    def clean_diagnosis_text(diagnosis: str) -> str:
+        """
+        Cleans the diagnosis text by removing special characters and extra whitespace.
+        This function is designed to prepare the text for a BigQuery SEARCH query.
+        """
+        # Remove special characters, keeping only letters, numbers, and basic punctuation.
+        # This will remove characters like *, #, etc.
+        cleaned_text = re.sub(r'[^\w\s.\-,"]', '', diagnosis)
+        # Replace multiple whitespace characters with a single space
+        cleaned_text = re.sub(r'\s+', ' ', cleaned_text).strip()
+        return cleaned_text
+
     def search_bigquery_for_remedy(search_query: str) -> str:
         """
         Searches the BigQuery table for a remedy using the SEARCH function.
@@ -86,20 +99,10 @@ def create_field_mode_ui():
                   SEARCH(remedy_description, @query)
             """
 
-            # Escape special characters for BigQuery SEARCH
-            # The SEARCH function in BigQuery requires special characters like '*' to be escaped.
-            escaped_search_query = search_query
-            reserved_chars = r'\\*?:.()[]{}|&^$#@+-\"\'`'
-            for char in reserved_chars:
-                if char in escaped_search_query:
-                    escaped_search_query = escaped_search_query.replace(char, f'\\\\{char}')
-
-
-
             # Set up the query parameters.
             job_config = bigquery.QueryJobConfig(
                 query_parameters=[
-                    bigquery.ScalarQueryParameter("query", "STRING", escaped_search_query),
+                    bigquery.ScalarQueryParameter("query", "STRING", search_query),
                 ]
             )
 
@@ -141,10 +144,11 @@ def create_field_mode_ui():
             print(f"Diagnosis received: {diagnosis}")
 
             if "Could not parse" in diagnosis:
-                return f"Sorry, I couldn\'t identify the condition from the image. Raw output: {diagnosis}"
+                return f"Sorry, I couldn't identify the condition from the image. Raw output: {diagnosis}"
 
-            search_query = diagnosis
-            report_title = diagnosis
+            # Clean the diagnosis text before using it as a search query
+            search_query = clean_diagnosis_text(diagnosis)
+            report_title = diagnosis # Keep the original diagnosis for the report
 
             results = search_bigquery_for_remedy(search_query)
             
@@ -192,7 +196,7 @@ def create_field_mode_ui():
             if temp_file_path and os.path.exists(temp_file_path):
                 os.remove(temp_file_path)
 
-    # ... (the rest of your Gradio Interface definition remains the same)
+    # ... (the rest of your Gradio Interface definition remains the same) 
     css = """
     footer {visibility: hidden !important;}
     .gradio-container {font-family: 'IBM Plex Sans', sans-serif;}