Update app.py

app.py

@@ -199,7 +199,15 @@ prompt_templates = {
         Provide a comprehensive guide on securing wireless networks and IoT devices, including the use of encryption, network segmentation, and regular vulnerability assessments.
         """
     ),
-    # Add all other prompts here...
+    # General Prompt
+    "general": PromptTemplate(
+        input_variables=["query", "context"],
+        template="""
+        Context: You are a cybersecurity expert with extensive experience in all sub-streams of the industry, including but not limited to network security, application security, cloud security, threat intelligence, penetration testing, and incident response. {context}
+        Query: {query}
+        Please provide a detailed and professional response to the query based on your expertise in cybersecurity and the provided context.
+        """
+    ),
 }
 
 # Initialize chains for each prompt
@@ -252,7 +260,7 @@ def classify_query(query: str) -> str:
         return "patch_management"
     elif "wireless and IoT security" in query.lower():
         return "wireless_iot_security"
-    # Add more conditions for other prompts...
+    # Default to the general prompt
     else:
         return "general"
 
@@ -269,7 +277,7 @@ async def process_search(search_query: SearchQuery):
         if query_type in chains:
             response = chains[query_type].run(query=search_query.query, context=context)
         else:
-            response = "I am not sure how to help with that. Please provide more details."
+            response = chains["general"].run(query=search_query.query, context=context)
 
         return {
             "status": "success",