Spaces:
Sleeping
Sleeping
File size: 2,573 Bytes
e266142 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
import logging
import gradio as gr
from sql_generator import SQLGenerator
from intent_classifier import IntentClassifier
from rag_system import RAGSystem
# Initialize logging
logging.basicConfig(
level=logging.INFO, # Set the logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
format='%(asctime)s - %(levelname)s - %(message)s' # Format for the log messages
)
class UnifiedSystem:
def __init__(self):
self.sql_generator = SQLGenerator()
self.intent_classifier = IntentClassifier()
self.rag_system = RAGSystem()
def process_query(self, query):
logging.info(f"Processing query: {query}") # Log the incoming query
intent, confidence = self.intent_classifier.classify(query)
logging.info(f"Classified intent: {intent} with confidence: {confidence:.2f}")
if intent == "database_query":
sql_query = self.sql_generator.generate_query(query) # Assuming this method is correct
products = self.sql_generator.fetch_shopify_data("products")
if products and 'products' in products:
results = "\n".join([
f"Title: {p['title']}, Vendor: {p['vendor']}"
for p in products['products']
])
return f"Intent: Database Query (Confidence: {confidence:.2f})\n\n" \
f"SQL Query: {sql_query}\n\nResults:\n{results}"
else:
logging.warning("No results found or error fetching data from Shopify.")
return "No results found or error fetching data from Shopify."
elif intent == "product_description":
rag_response = self.rag_system.process_query(query)
return f"Intent: Product Description (Confidence: {confidence:.2f})\n\n" \
f"Response: {rag_response}"
logging.error("Intent not recognized.")
return "Intent not recognized."
def create_interface():
system = UnifiedSystem()
iface = gr.Interface(
fn=system.process_query,
inputs=gr.Textbox(
label="Enter your query",
placeholder="e.g., 'Show me all T-shirts' or 'Describe the product features'"
),
outputs=gr.Textbox(label="Response"),
title="Unified Query Processing System",
description="Enter a natural language query to search products or get descriptions."
)
return iface
if __name__ == "__main__":
iface = create_interface()
iface.launch()
|