Test

app.py

@@ -48,58 +48,57 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
-        gr.Textbox(value=""" 
-        ### Instructions:
-        Your task is to convert a question into a SQL query, given a Postgres database schema.
-        Adhere to these rules:
-        - Use Table Aliases to prevent ambiguity.
-        - When creating a ratio, always cast the numerator as float.
-
-        ### Input:
-        Generate a SQL query that answers the question `{question}`.
-        This query will run on a database whose schema is represented in this string:
-        CREATE TABLE products (
-        product_id INTEGER PRIMARY KEY, -- Unique ID for each product
-        name VARCHAR(50), -- Name of the product
-        price DECIMAL(10,2), -- Price of each unit of the product
-        quantity INTEGER  -- Current quantity in stock
-        );
-
-        CREATE TABLE customers (
-        customer_id INTEGER PRIMARY KEY, -- Unique ID for each customer
-        name VARCHAR(50), -- Name of the customer
-        address VARCHAR(100) -- Mailing address of the customer
-        );
-
-        CREATE TABLE salespeople (
-        salesperson_id INTEGER PRIMARY KEY, -- Unique ID for each salesperson
-        name VARCHAR(50), -- Name of the salesperson
-        region VARCHAR(50) -- Geographic sales region
-        );
-
-        CREATE TABLE sales (
-        sale_id INTEGER PRIMARY KEY, -- Unique ID for each sale
-        product_id INTEGER, -- ID of product sold
-        customer_id INTEGER,  -- ID of customer who made purchase
-        salesperson_id INTEGER, -- ID of salesperson who made the sale
-        sale_date DATE, -- Date the sale occurred
-        quantity INTEGER -- Quantity of product sold
-        );
-
-        CREATE TABLE product_suppliers (
-        supplier_id INTEGER PRIMARY KEY, -- Unique ID for each supplier
-        product_id INTEGER, -- Product ID supplied
-        supply_price DECIMAL(10,2) -- Unit price charged by supplier
-        );
-
-        -- sales.product_id can be joined with products.product_id
-        -- sales.customer_id can be joined with customers.customer_id
-        -- sales.salesperson_id can be joined with salespeople.salesperson_id
-        -- product_suppliers.product_id can be joined with products.product_id
-
-        ### Response:
-        ```sql
-    """, label="System message"),
+        gr.Textbox(value="""Your task is to convert a question into a SQL query, given a Postgres database schema.
+Adhere to these rules:
+- Use Table Aliases to prevent ambiguity.
+- When creating a ratio, always cast the numerator as float.
+- Only use tables and fields explicitly defined in the provided schema. Do not include or reference any tables or fields not listed.
+- If the required query cannot be constructed using the provided schema, return the string `NOT FOUND`.
+
+### Input:
+Generate a SQL query that answers the question `{question}`.
+This query will run on a database whose schema is represented in this string:
+CREATE TABLE products (
+product_id INTEGER PRIMARY KEY, -- Unique ID for each product
+name VARCHAR(50), -- Name of the product
+price DECIMAL(10,2), -- Price of each unit of the product
+quantity INTEGER  -- Current quantity in stock
+);
+
+CREATE TABLE customers (
+customer_id INTEGER PRIMARY KEY, -- Unique ID for each customer
+name VARCHAR(50), -- Name of the customer
+address VARCHAR(100) -- Mailing address of the customer
+);
+
+CREATE TABLE salespeople (
+salesperson_id INTEGER PRIMARY KEY, -- Unique ID for each salesperson
+name VARCHAR(50), -- Name of the salesperson
+region VARCHAR(50) -- Geographic sales region
+);
+
+CREATE TABLE sales (
+sale_id INTEGER PRIMARY KEY, -- Unique ID for each sale
+product_id INTEGER, -- ID of product sold
+customer_id INTEGER,  -- ID of customer who made purchase
+salesperson_id INTEGER, -- ID of salesperson who made the sale
+sale_date DATE, -- Date the sale occurred
+quantity INTEGER -- Quantity of product sold
+);
+
+CREATE TABLE product_suppliers (
+supplier_id INTEGER PRIMARY KEY, -- Unique ID for each supplier
+product_id INTEGER, -- Product ID supplied
+supply_price DECIMAL(10,2) -- Unit price charged by supplier
+);
+
+-- sales.product_id can be joined with products.product_id
+-- sales.customer_id can be joined with customers.customer_id
+-- sales.salesperson_id can be joined with salespeople.salesperson_id
+-- product_suppliers.product_id can be joined with products.product_id
+
+### Response:
+```sql""", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(