Pushing code to HuggingFace space

app.py

@@ -0,0 +1,82 @@
+import gradio as gr
+import os
+import subprocess
+import logging
+import asyncio 
+from src.agents.nl_sql_agent import NLSQLAgent
+
+logging.basicConfig(level=logging.INFO)
+logging.getLogger("httpx").setLevel(logging.WARNING)
+logging.getLogger("chromadb").setLevel(logging.WARNING)
+
+print("--- Starting Hugging Face Space setup ---")
+
+SETUP_DATABASE_SCRIPT = os.path.join('src', 'setup_database.py')
+RAG_INDEX_SCRIPT = os.path.join('src', 'rag_index.py')
+
+if not os.environ.get("NEBIUS_API_KEY"):
+    print("FATAL ERROR: NEBIUS_API_KEY environment variable not set. This is required for Nebius LLMs and Embeddings.")
+
+try:
+    print(f"Running database setup script: {SETUP_DATABASE_SCRIPT}...")
+    subprocess.run(["python", SETUP_DATABASE_SCRIPT], check=True, capture_output=True)
+    print("Database setup complete.")
+
+    print(f"Running RAG indexing script: {RAG_INDEX_SCRIPT}...")
+    subprocess.run(["python", RAG_INDEX_SCRIPT], check=True, capture_output=True)
+    print("RAG indexing complete.")
+
+except subprocess.CalledProcessError as e:
+    print(f"Error during initial setup. Script failed with exit code {e.returncode}.")
+    print(f"Stdout:\n{e.stdout.decode()}")
+    print(f"Stderr:\n{e.stderr.decode()}")
+    print("Exiting application due to critical setup failure.")
+    exit(1) 
+
+print("--- Hugging Face Space setup complete. Initializing Agent ---")
+
+# --- Initialize the NL-to-SQL Agent ---
+nl_sql_agent_instance = NLSQLAgent() 
+print("NLSQLAgent initialized.")
+
+# --- Define Gradio Interface Functions ---
+async def query_agent_gradio(user_query: str) -> str:
+    if not user_query.strip():
+        return "Please enter a question to get started!"
+
+    try:
+        response = await nl_sql_agent_instance.process_query(user_query)
+        return response
+    except Exception as e:
+        logging.error(f"Error processing query in Gradio app: {e}", exc_info=True) 
+        return f"An internal error occurred: {type(e).__name__}: {str(e)}. Please check the Space logs for more details."
+
+# --- Create Gradio Interface ---
+iface = gr.Interface(
+    fn=query_agent_gradio, 
+    inputs=gr.Textbox(lines=2, placeholder="Ask a question about your sales data (e.g., 'What is the total number of sales?', 'Show me top 5 products by quantity sold')..."),
+    outputs=gr.Markdown(), 
+    title="📊 AI-Powered Data Analyst Assistant (NL-to-SQL Demo)",
+    description=(
+        "Ask natural language questions to get insights from your sales database. "
+        "This demo exclusively uses the NL-to-SQL Agent, leveraging fine-tuned LLMs, RAG for schema context, and SQLite. "
+        "Example questions: 'What are the names of customers in the North region?', 'How much revenue did we generate from Electronics products?', 'Show me the total revenue for each month over the last six months, ordered by month.'"
+    ),
+    examples=[
+        ["What is the total number of sales?"],
+        ["What are the names of customers in the North region?"],
+        ["How much revenue did we generate from Electronics products?"],
+        ["Which customer has the most sales in the past one month?"],
+        ["Show me the total revenue for each month over the last six months, ordered by month."],
+        ["Which are our top 5 products by total quantity sold across all time?"],
+        ["What is the average amount per sale for each product category?"],
+        ["How many unique customers have made a purchase in each region over the last year?"],
+        ["What is the total revenue generated by customers from each region?"]
+    ],
+    allow_flagging="auto", 
+    analytics_enabled=True 
+)
+
+if __name__ == "__main__":
+    print("Launching Gradio app...")
+    iface.launch(share=False)

src/agents/agent_tools/schema_retriever_tool.py

@@ -1,5 +1,3 @@
-# src/agents/tools/schema_retriever_tool.py
-
 import os
 import logging
 import chromadb

src/agents/nl_sql_agent.py

@@ -12,7 +12,7 @@ logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
 logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
 
 class NLSQLAgent:
-    def __init__(self, model_name: str, api_key: str, base_url: str):
+    def __init__(self):
         """
         Initializes the NL-to-SQL Agent, which translates natural language to SQL, executes it, and provides answers.
 
@@ -98,11 +98,7 @@ if __name__ == "__main__":
     elif YOUR_FINETUNED_MODEL_ID == "your-finetuned-model-id":
         print("Error: Please replace 'your-finetuned-model-id' with your actual model ID.")
     else:
-        nl_sql_agent = NLSQLAgent(
-            model_name=YOUR_FINETUNED_MODEL_ID,
-            api_key=NEBIUS_API_KEY,
-            base_url=NEBIUS_BASE_URL
-        )
+        nl_sql_agent = NLSQLAgent()
 
         print("\nNL-to-SQL Agent initialized. Ask a question about your sales database.")