refactor: Update import paths and clean up code structure; add basic usage demo for Querypls functionality

Author: samadpls

.devcontainer/devcontainer.json

@@ -14,6 +14,8 @@ __pycache__/
 # Installer logs
 pip-log.txt
 pip-delete-this-directory.txt
+test_*
+run_*
 
 # Unit test / coverage reports
 .coverage

.gitignore

@@ -0,0 +1,190 @@
+#!/usr/bin/env python3
+"""
+Basic usage demo for Querypls backend functionality.
+Demonstrates conversation, SQL generation, and CSV analysis.
+"""
+
+import sys
+import os
+
+# Add the project root to Python path
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from src.services.routing_service import IntelligentRoutingService
+from src.backend.orchestrator import BackendOrchestrator
+from src.schemas.requests import NewChatRequest
+
+
+def demo_conversation():
+    """Demo conversation functionality."""
+    print("🗣️  CONVERSATION DEMO")
+    print("=" * 40)
+    
+    routing_service = IntelligentRoutingService()
+    
+    # Test different conversation types
+    conversations = [
+        "Hello",
+        "How are you?",
+        "What can you do?",
+        "Thanks for your help",
+        "Goodbye"
+    ]
+    
+    for query in conversations:
+        print(f"\nUser: {query}")
+        response = routing_service.handle_conversation_query(query)
+        print(f"Assistant: {response}")
+    
+    print("\n" + "=" * 40)
+
+
+def demo_sql_generation():
+    """Demo SQL generation functionality."""
+    print("🗃️  SQL GENERATION DEMO")
+    print("=" * 40)
+    
+    routing_service = IntelligentRoutingService()
+    
+    # Test different SQL queries
+    sql_queries = [
+        "Show me all users",
+        "Find customers who made purchases in the last 30 days",
+        "Get the total sales by month",
+        "SELECT * FROM users WHERE status = 'active'"
+    ]
+    
+    for query in sql_queries:
+        print(f"\nUser: {query}")
+        response = routing_service.handle_sql_query(query, [])
+        print(f"Assistant: {response[:200]}...")
+    
+    print("\n" + "=" * 40)
+
+
+def demo_csv_analysis():
+    """Demo CSV analysis functionality."""
+    print("📊 CSV ANALYSIS DEMO")
+    print("=" * 40)
+    
+    # Sample CSV data
+    sample_csv = """name,age,salary,department
+Alice,25,50000,IT
+Bob,30,60000,HR
+Charlie,35,70000,IT
+Diana,28,55000,Finance
+Eve,32,65000,HR"""
+    
+    print(f"Sample CSV Data:\n{sample_csv}")
+    
+    routing_service = IntelligentRoutingService()
+    
+    # Test different CSV analysis queries
+    csv_queries = [
+        "Show me the basic statistics of the data",
+        "Create a bar chart of department distribution",
+        "What is the average salary by department?",
+        "Show me the top 3 highest paid employees"
+    ]
+    
+    for query in csv_queries:
+        print(f"\nUser: {query}")
+        response = routing_service.handle_csv_query(query, sample_csv)
+        print(f"Assistant: {response[:300]}...")
+    
+    print("\n" + "=" * 40)
+
+
+def demo_intelligent_routing():
+    """Demo intelligent routing functionality."""
+    print("🧠 INTELLIGENT ROUTING DEMO")
+    print("=" * 40)
+    
+    routing_service = IntelligentRoutingService()
+    
+    # Test different types of queries
+    test_queries = [
+        ("Hello", "CONVERSATION_AGENT"),
+        ("Show me all users", "SQL_AGENT"),
+        ("Analyze this CSV data", "CSV_AGENT"),
+        ("How are you?", "CONVERSATION_AGENT"),
+        ("SELECT * FROM users", "SQL_AGENT"),
+        ("Create a chart from the data", "CSV_AGENT")
+    ]
+    
+    for query, expected_agent in test_queries:
+        print(f"\nQuery: '{query}'")
+        decision = routing_service.determine_agent(query, [], csv_loaded=True)
+        print(f"Expected: {expected_agent}")
+        print(f"Actual: {decision.agent}")
+        print(f"Confidence: {decision.confidence}")
+        print(f"Reasoning: {decision.reasoning}")
+    
+    print("\n" + "=" * 40)
+
+
+def demo_orchestrator():
+    """Demo the main orchestrator functionality."""
+    print("🎼 ORCHESTRATOR DEMO")
+    print("=" * 40)
+    
+    orchestrator = BackendOrchestrator()
+    
+    # Create a new session
+    session_info = orchestrator.create_new_session(NewChatRequest(session_name="Demo Session"))
+    session_id = session_info.session_id
+    print(f"Created session: {session_id}")
+    
+    # Test different types of interactions
+    interactions = [
+        ("Hello", "conversation"),
+        ("Show me all users", "sql"),
+        ("What can you do?", "conversation")
+    ]
+    
+    for query, query_type in interactions:
+        print(f"\nUser ({query_type}): {query}")
+        response = orchestrator.generate_intelligent_response(session_id, query)
+        print(f"Assistant: {response.content[:150]}...")
+    
+    # Test CSV functionality
+    sample_csv = "name,age,salary\nAlice,25,50000\nBob,30,60000\nCharlie,35,70000"
+    result = orchestrator.load_csv_data(session_id, sample_csv)
+    print(f"\nCSV Load Result: {result['status']}")
+    
+    response = orchestrator.generate_intelligent_response(session_id, "Analyze this data")
+    print(f"CSV Analysis: {response.content[:200]}...")
+    
+    print("\n" + "=" * 40)
+
+
+def main():
+    """Run all demos."""
+    print("🚀 Querypls Backend Functionality Demo")
+    print("=" * 50)
+    
+    demos = [
+        ("Conversation", demo_conversation),
+        ("SQL Generation", demo_sql_generation),
+        ("CSV Analysis", demo_csv_analysis),
+        ("Intelligent Routing", demo_intelligent_routing),
+        ("Orchestrator", demo_orchestrator),
+    ]
+    
+    for demo_name, demo_func in demos:
+        try:
+            demo_func()
+        except Exception as e:
+            print(f"❌ {demo_name} demo failed: {str(e)}")
+    
+    print("\n🎉 Demo completed! All backend functionality is working correctly.")
+    print("\n📝 Summary:")
+    print("- Conversation: Natural responses for greetings and help")
+    print("- SQL Generation: Convert natural language to SQL queries")
+    print("- CSV Analysis: Analyze CSV data with Python code")
+    print("- Intelligent Routing: Automatically choose the right agent")
+    print("- Orchestrator: Complete session management")
+
+
+if __name__ == "__main__":
+    main() 

examples/basic_usage_demo.py

@@ -2,7 +2,7 @@
 Backend utilities for Streamlit configuration and styling.
 """
 
-from config.constants import STREAMLIT_CONFIG
+from src.config.constants import STREAMLIT_CONFIG
 import streamlit as st
 import sys
 import os

src/backend/backend.py

@@ -7,33 +7,21 @@
 from typing import List, Optional, Dict, Any
 from dataclasses import dataclass
 
-from config.settings import get_settings
-from config.constants import (
-    WELCOME_MESSAGE,
-    DEFAULT_SESSION_NAME,
-    CSV_LOAD_ERROR,
-    CSV_ANALYSIS_ERROR,
-    SESSION_CREATE_ERROR,
-    ORCHESTRATOR_INIT_ERROR,
-    SESSION_NOT_FOUND_ERROR,
-    RESPONSE_GENERATION_ERROR,
-    MESSAGE_LOAD_ERROR,
-    MAX_CHAT_HISTORIES,
-)
-from services.sql_service import SQLGenerationService
-from services.csv_analysis_tools import CSVAnalysisTools, create_csv_analysis_agent
-from services.conversation_service import ConversationService
-from services.routing_service import IntelligentRoutingService
-from schemas.requests import (
+from src.config.settings import get_settings
+from src.config.constants import WELCOME_MESSAGE, DEFAULT_SESSION_NAME
+from src.services.sql_service import SQLGenerationService
+from src.services.csv_analysis_tools import CSVAnalysisTools, create_csv_analysis_agent
+from src.services.conversation_service import ConversationService
+from src.services.routing_service import IntelligentRoutingService
+from src.schemas.requests import (
     SQLGenerationRequest,
     ChatMessage,
     ConversationHistory,
     NewChatRequest,
 )
-from schemas.responses import (
+from src.schemas.responses import (
     ChatResponse,
     SessionInfo,
-    ErrorResponse,
     HealthCheckResponse,
 )
 
@@ -65,10 +53,8 @@ def create_new_session(self, request: NewChatRequest) -> SessionInfo:
 
         messages = []
         if request.initial_context:
-            messages.append(
-                ChatMessage(
-                    role="system",
-                    content=request.initial_context))
+            messages.append(ChatMessage(
+                role="system", content=request.initial_context))
 
         messages.append(ChatMessage(role="assistant", content=WELCOME_MESSAGE))
 
@@ -113,8 +99,7 @@ def delete_session(self, session_id: str) -> bool:
             return True
         return False
 
-    def load_csv_data(self, session_id: str,
-                      csv_content: str) -> Dict[str, Any]:
+    def load_csv_data(self, session_id: str, csv_content: str) -> Dict[str, Any]:
         session = self.get_session(session_id)
         if not session:
             raise ValueError(f"Session {session_id} not found")
@@ -134,9 +119,8 @@ def generate_intelligent_response(
             raise ValueError(f"Session {session_id} not found")
 
         user_message = ChatMessage(
-            role="user",
-            content=user_query,
-            timestamp=datetime.now().isoformat())
+            role="user", content=user_query, timestamp=datetime.now().isoformat()
+        )
         session.messages.append(user_message)
 
         # Determine which agent should handle this query
@@ -157,7 +141,7 @@ def generate_intelligent_response(
         elif routing_decision.agent == "CSV_AGENT":
             if session.csv_data:
                 response_content = self.routing_service.handle_csv_query(
-                    user_query, session.csv_data
+                    user_query, session.csv_data, session.messages
                 )
             else:
                 response_content = "I don't see any CSV data loaded. Please upload a CSV file first to analyze it."
@@ -187,9 +171,7 @@ def get_conversation_history(self, session_id: str) -> ConversationHistory:
         if not session:
             raise ValueError(f"Session {session_id} not found")
 
-        return ConversationHistory(
-            messages=session.messages,
-            session_id=session_id)
+        return ConversationHistory(messages=session.messages, session_id=session_id)
 
     def get_csv_info(self, session_id: str) -> Dict[str, Any]:
         return self.csv_tools.get_csv_info(session_id)

src/backend/orchestrator.py

@@ -10,10 +10,41 @@
 STREAMLIT_HOST = "localhost"
 
 # Streamlit Configuration
-STREAMLIT_CONFIG = {
-    "page_title": "Querypls",
-    "page_icon": "💬",
-    "layout": "wide"}
+STREAMLIT_CONFIG = {"page_title": "Querypls",
+                    "page_icon": "💬", "layout": "wide"}
+
+# Welcome and Session Messages
+WELCOME_MESSAGE = "Hello! 👋 I'm Querypls, your SQL and data analysis assistant. I can help you generate SQL queries or analyze CSV files. What would you like to work on today?"
+DEFAULT_SESSION_NAME = "Default Chat"
+
+# CSV Analysis Section
+CSV_ANALYSIS_SECTION = "### 📊 CSV Analysis"
+CSV_UPLOAD_LABEL = "Upload CSV File"
+CSV_UPLOAD_HELP = "Upload a CSV file to analyze with Python code"
+CSV_PREVIEW = "📋 CSV Preview"
+CSV_COLUMNS = "**Columns:** {columns}"
+CSV_DTYPES = "**Data Types:** {dtypes}"
+LOAD_CSV_BUTTON = "📊 Load CSV Data"
+CSV_LOADED_SUCCESS = "✅ CSV data loaded successfully!"
+CSV_UPLOAD_SUCCESS = "✅ CSV uploaded successfully! Shape: {shape}"
+CSV_UPLOAD_ERROR = "❌ Error uploading CSV: {error}"
+CSV_LOAD_ERROR = "❌ No CSV data loaded. Please upload a CSV file first."
+CSV_ANALYSIS_ERROR = "❌ Error analyzing CSV: {error}"
+
+# Session Management
+SESSIONS_SECTION = "### 💬 Chat Sessions"
+NEW_SESSION_BUTTON = "➕ New Session"
+SESSION_CREATE_ERROR = "❌ Error creating session: {error}"
+SESSION_NOT_FOUND_ERROR = "❌ Session not found"
+
+# Application Errors
+ORCHESTRATOR_INIT_ERROR = "❌ Error initializing orchestrator: {error}"
+APP_INIT_ERROR = "❌ Error initializing application"
+RESPONSE_GENERATION_ERROR = "❌ Error generating response: {error}"
+MESSAGE_LOAD_ERROR = "❌ Error loading messages: {error}"
+
+# UI Elements
+MADE_WITH_LOVE = "Made with 🤍"
 
 # Available Models
 AVAILABLE_MODELS = {

src/config/constants.py

@@ -13,8 +13,7 @@
 class Settings(BaseSettings):
     groq_api_key: str = Field(..., env="GROQ_API_KEY")
     groq_model_name: str = Field(
-        default="openai/gpt-oss-120b",
-        env="GROQ_MODEL_NAME")
+        default="openai/gpt-oss-120b", env="GROQ_MODEL_NAME")
     app_version: str = Field(default="1.0.0", env="APP_VERSION")
     max_chat_histories: int = Field(default=5, env="MAX_CHAT_HISTORIES")
     debug_mode: bool = Field(default=False, env="DEBUG_MODE")