index

app.py

@@ -14,6 +14,8 @@ import socket
 import tempfile
 import json # New import
 
+from google.cloud import bigquery
+
 # Suppress potential warnings for a cleaner console
 warnings.filterwarnings("ignore")
 os.environ["TORCH_COMPILE_DISABLE"] = "1"  # Ensure torch compile is off
@@ -65,6 +67,54 @@ else:
 
 def create_field_mode_ui():
     """Creates the Gradio UI for the offline Field Mode."""
+
+    def search_bigquery_for_remedy(search_query: str) -> str:
+        """
+        Searches the BigQuery table for a remedy using the SEARCH function.
+        """
+        try:
+            # Initialize the BigQuery client. Your project ID is used here.
+            client = bigquery.Client(project="gem-creation")
+
+            # Define the SQL query with the SEARCH function and a parameter for safety.
+            query = """
+                SELECT
+                  remedy_description
+                FROM
+                  `gem-creation.maize_remedies.remedies`
+                WHERE
+                  SEARCH(remedy_description, @query)
+            """
+
+            # Set up the query parameters.
+            job_config = bigquery.QueryJobConfig(
+                query_parameters=[
+                    bigquery.ScalarQueryParameter("query", "STRING", search_query),
+                ]
+            )
+
+            # Execute the query.
+            print(f"Executing BigQuery search for: '{search_query}'")
+            query_job = client.query(query, job_config=job_config)
+
+            # Process the results.
+            results = list(query_job)  # Get all results
+
+            if not results:
+                print("No remedy found in BigQuery.")
+                return "No remedy found in the cloud knowledge base for this condition."
+            else:
+                # The result is a Row object; access the column by its name.
+                remedy = results[0].remedy_description
+                print("Remedy found in BigQuery.")
+                return remedy
+
+        except Exception as e:
+            error_message = f"An error occurred while querying the BigQuery database: {e}"
+            print(f"❌ {error_message}")
+            return error_message
+
+
     def get_diagnosis_and_remedy(uploaded_image: Image.Image) -> str:
         if uploaded_image is None:
             return "Please upload an image of a maize plant first."
@@ -86,12 +136,12 @@ def create_field_mode_ui():
             search_query = diagnosis
             report_title = diagnosis
 
-            results = KB.search(search_query)
+            results = search_bigquery_for_remedy(search_query)
             
             if not results:
                 remedy = "No remedy found in the local knowledge base."
             else:
-                remedy = results[0]['content']
+                remedy = results
 
             final_response = f"""
             ## Diagnosis Report