Update main.py

main.py

@@ -66,9 +66,21 @@ async def generate_response(request: QuestionRequest):
         raise HTTPException(status_code=500, detail="Chatbot not initialized. Please try again later.")
     
     try:
-        # Generate response (non-streaming for simplicity in API context)
-        response = chatbot.chat(request.question, stream=False)
-        return {"response": response}
+        # Generate response (non-streaming for simplicity)
+        response_data = chatbot.chat(request.question, stream=False)
+        
+        # Extract the actual response text
+        # The response may be a dictionary; check for 'gen' or other keys
+        if isinstance(response_data, dict):
+            # Assuming 'gen' contains the response text (list of strings)
+            response_text = "".join(response_data.get("gen", [])) if response_data.get("gen") else "Here's what we can do: Let's discuss your vision!"
+            if not response_text:
+                # Fallback to a default premium response if 'gen' is empty
+                response_text = f"Welcome, valued client! How can Abdullah Ali and our premium team bring your vision to life with a custom website or AI chatbot?"
+        else:
+            response_text = response_data  # Direct string response (if hugchat returns string)
+        
+        return {"response": response_text}
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to generate response: {str(e)}")