added bot prediction for assessments

test.py

@@ -6,9 +6,7 @@ input_base_path = '/root/ds_erp_ai/data/raw/erp_assessment_prediction'  # The ba
 pipeline = CompanyModelPipeline(company_ids=company_ids, input_base_path=input_base_path)
 pipeline.run_pipeline()'''
 
-from src.pipeline.inference import AssessmentInference
-
-
+'''from src.pipeline.inference import AssessmentInference
 
 inference = AssessmentInference(
     company_id="testid",num_assessments=2
@@ -18,3 +16,32 @@ result = inference.run()
 
 
 print(result)
+'''
+'''
+response2 = bot.predict_next_n_assessment(
+    company_info=company_info,
+    companyid="testid",
+    N=3
+)
+
+print(f"Predictions {response2}")
+'''
+
+from src.services.chatbot import Chatbot
+company_info = {
+    'company_name': "ABC Corp",
+    'company_size': "Medium",  # Can be "Small", "Medium", or "Large"
+    'departments': ["Sales", "Marketing", "IT", "Finance", "HR", "Logistics"]
+}
+bot = Chatbot()
+response = bot.predict_based_on_past_assessment(
+    query="Should i make my next assessment weekly or biweekly to meet up to deadline?",
+    company_info=company_info,
+    companyid="testid"
+)
+
+print(f"Result: {response}")
+
+
+
+