new changes

util.py

@@ -244,12 +244,17 @@ class ErrorBot:
     # """
 
         system_prompt = f"""
-        You are a technical assistant. A user may ask questions about Problem Reports (PR), Fault Analyses (FA), and Corrections (CR).
-        Your task is to:
-        1. Identify which information (PR, FA, CR) is relevant to answering the user's question.
-        2. Explain the solution in simple, clear, actionable language.
-        3. Do not just repeat the content; summarize and explain.
-    
+        You are a versatile assistant. A user may ask questions about:
+        - Problem Reports (PR), Fault Analyses (FA), and Corrections (CR).
+        - Programming, algorithms, and code examples.
+        - Non-technical or general everyday topics.
+
+        Your tasks are:
+        1. If the question is about PR, FA, or CR → Identify which information is relevant and explain clearly in simple, actionable language (summarize, don’t just repeat).
+        2. If the question is about programming or algorithms → Provide a correct, clear, and well-structured code example in the requested language, with a short explanation.
+        3. If the question is non-technical/general → Respond politely, clearly, and helpfully in a conversational style.
+        4. Always keep answers concise and easy to understand.
+
         ### User Question:
         
 
@@ -394,6 +399,11 @@ class ErrorBot:
         is_followup = self.is_followup_query(query, history)
 
         # Step 2: Non-technical standalone
+        # Step 3: Technical or follow-up
+        print("is_followup", is_followup)
+        #print("last_context", self.last_context)
+        print("is_technical", is_technical)
+
         #if not is_technical:
         if not is_technical and not is_followup:
             print("⚠️ Non-technical standalone query → skipping Qdrant.")
@@ -413,12 +423,7 @@ class ErrorBot:
                 )
                 return completion.choices[0].message.content.strip(), []
 
-        # Step 3: Technical or follow-up
-        print("is_followup", is_followup)
-        print("last_context", self.last_context)
-        print("is_technical", is_technical)
-
-        if is_followup and self.last_context:
+        elif is_followup and self.last_context:
             if not is_technical:
                 print("⚠️ Non-technical followup → skipping Qdrant.")
                 system_prompt = "You are a helpful assistant. Answer clearly and concisely."
@@ -440,6 +445,30 @@ class ErrorBot:
                 print("🔄 Follow-up query → reusing previous context.")
                 retrieved_context = self.last_context
                 context_docs = retrieved_context
+        
+        elif is_followup and not self.last_context:
+
+            if not is_technical:
+                print("⚠️ Non-technical followup → skipping Qdrant.")
+                system_prompt = "You are a helpful assistant. Answer clearly and concisely."
+                convo = [{"role": "system", "content": system_prompt},
+                        {"role": "user", "content": query}]
+
+                if self.llm_provider == "gemini":
+                    convo_str = "\n".join([f"{m['role'].capitalize()}: {m['content']}" for m in convo])
+                    response = self.llm.generate_content(convo_str)
+                    return response.text.strip(), []
+
+                elif self.llm_provider == "groq":
+                    completion = self.llm.chat.completions.create(
+                        model=self.llm_model_name,
+                        messages=convo
+                    )
+                    return completion.choices[0].message.content.strip(), []
+            else:
+                print("🔄 Follow-up query → without previous context.")
+                #retrieved_context = self.last_context
+                context_docs = []
 
         else:
             print("📥 New technical query → retrieving from Qdrant.")
@@ -522,7 +551,7 @@ class ErrorBot:
         
         answer = self.generate_answer(query, context_docs, history, is_followup)
         last_context = self.last_context
-        print(f"\n🤖 Answer: {answer}")
+        #print(f"\n🤖 Answer: {answer}")
         return (answer, last_context)