app.py Beta 2 (Push 3)

app.py

@@ -7,8 +7,8 @@ import spacy
 import os
 import logging
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
+# Set up logging with detailed output
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
 app = FastAPI()
@@ -20,8 +20,8 @@ db = client["test"]
 products_collection = db["products"]
 
 # BlenderBot Setup
-model_repo = "SyedHutter/blenderbot_model"  # Repo ID
-model_subfolder = "blenderbot_model"        # Subdirectory within repo
+model_repo = "SyedHutter/blenderbot_model"
+model_subfolder = "blenderbot_model"
 model_dir = "/home/user/app/blenderbot_model"
 
 if not os.path.exists(model_dir):
@@ -99,21 +99,27 @@ async def root():
 @app.post("/process/", response_model=CombinedResponse)
 async def process_prompt(request: PromptRequest):
     try:
+        logger.info(f"Processing request: {request.input_text}")
         input_text = request.input_text
         history = request.conversation_history[-3:] if request.conversation_history else []
 
         intent = detect_intent(input_text)
         keywords = extract_keywords(input_text)
         ner_response = {"extracted_keywords": keywords}
+        logger.info(f"Intent: {intent}, Keywords: {keywords}")
 
         products = search_products_by_keywords(keywords)
         product_context = get_product_context(products)
+        logger.info(f"Products matched: {len(products)}")
 
         history_str = " || ".join(history)
         full_input = f"{history_str} || {product_context} {context_msg} || {input_text}" if history else f"{product_context} {context_msg} || {input_text}"
+        logger.info(f"Full input to model: {full_input}")
+
         inputs = tokenizer(full_input, return_tensors="pt", truncation=True, max_length=512)
         outputs = model.generate(**inputs, max_length=150, num_beams=5, no_repeat_ngram_size=2)
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        logger.info(f"Model response: {response}")
 
         enhanced_response = format_response(response, products, intent)
         qa_response = {
@@ -128,6 +134,7 @@ async def process_prompt(request: PromptRequest):
             "products_matched": products
         }
     except Exception as e:
+        logger.error(f"Error processing request: {str(e)}", exc_info=True)
         raise HTTPException(status_code=500, detail=f"Oops, something went wrong: {str(e)}. Try again!")
 
 @app.on_event("startup")