import gradio as gr
from sql_generator import SQLGenerator
from intent_classifier import IntentClassifier
from rag_system import RAGSystem

class UnifiedSystem:
    def __init__(self):
        self.sql_generator = SQLGenerator()
        self.intent_classifier = IntentClassifier()
        self.rag_system = RAGSystem()
        
    def process_query(self, query):
        intent, confidence = self.intent_classifier.classify(query)
        
        if intent == "database_query":
            sql_query = self.sql_generator.generate_query(query)
            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:
                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}"
        
        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()