Update app.py

app.py

@@ -295,8 +295,7 @@ async def generate_streaming_response(messages: List[Dict], use_search: bool, te
 
 # --- Streaming Chat Endpoint ---
 @app.post("/chat/stream")
-async def chat_stream_endpoint(request: Request#, _: None = Depends(verify_origin)
-                              ):
+async def chat_stream_endpoint(request: Request, _: None = Depends(verify_origin)):
     if not client:
         raise HTTPException(status_code=500, detail="LLM client not configured")
         
@@ -305,6 +304,7 @@ async def chat_stream_endpoint(request: Request#, _: None = Depends(verify_origi
         user_message = data.get("message", "").strip()
         use_search = data.get("use_search", False)  # Default: False
         temperature = max(0, min(2, data.get("temperature", 0.7)))  # Clamp to valid range
+        system_prompt = data.get("system_prompt")
         conversation_history = data.get("history", [])
         
         if not user_message:
@@ -312,7 +312,8 @@ async def chat_stream_endpoint(request: Request#, _: None = Depends(verify_origi
 
         # Prepare messages
         current_date = datetime.now().strftime("%Y-%m-%d")
-        system_content = (SYSTEM_PROMPT_WITH_SEARCH if use_search else SYSTEM_PROMPT_NO_SEARCH).format(current_date=current_date)
+        system_content = (SYSTEM_PROMPT_WITH_SEARCH if use_search else system_prompt#SYSTEM_PROMPT_NO_SEARCH).format(current_date=current_date
+                         )
         messages = [{"role": "system", "content": system_content}] + conversation_history + [{"role": "user", "content": user_message}]
         
         logger.info(f"Stream request - search: {use_search}, temp: {temperature}")