Created Full Table Path

app.py

@@ -60,12 +60,21 @@ def update_tables(schema_name):
 
 # Get Schema
 def get_table_schema(table):
-    result = conn.sql(f"SELECT sql FROM duckdb_tables() where table_name ='{table}';").df()
+    result = conn.sql(f"SELECT sql, database_name, schema_name FROM duckdb_tables() where table_name ='{table}';").df()
     ddl_create = result.iloc[0,0]
+    parent_database = result.iloc[0,1]
+    schema_name = result.iloc[0,2]
+    full_path = f"{parent_database}.{schema_name}.{table}"
+    if schema_name != "main":
+        old_path = f"{schema_name}.{table}"
+    else:
+        old_path = table
+    ddl_create = ddl_create.replace(old_path, full_path)
     return ddl_create
 
 # Get Prompt
-def get_prompt(schema, query_input):
+def get_prompt(schema, query_input, parent_database):
+    schema = schema.replace()
     text = f"""
     ### Instruction:
     Your task is to generate valid duckdb SQL query to answer the following question.
@@ -98,9 +107,9 @@ def text2sql(table, query_input):
             result_output:pd.DataFrame([{"error": f"❌ Unable to get the SQL query based on the text. {e}"}])
         }
 
-    schema = get_table_schema(table)
+    schema, parent_database = get_table_schema(table)
     print(f'Schema Generated...')
-    prompt = get_prompt(schema, query_input)
+    prompt = get_prompt(schema, query_input, parent_database)
     print(f'Prompt Generated...')
     try:
         print(f'Generating SQL... {model.device}')
@@ -131,9 +140,6 @@ def text2sql(table, query_input):
         result_output:query_result
     }
 
-# Load Databases Names
-
-
 # Custom CSS styling
 custom_css = """
 .gradio-container {