Update app.py

app.py

@@ -96,9 +96,10 @@ def generate_prompt_response(persona_desc: str, msg_prompt: str) -> dict:
             top_k=40,
             repetition_penalty=1.1
         )
+        prompt_template = (persona_desc + user_prompt)
         
-        generated_text = pipe(persona_desc + user_prompt)[0]['generated_text']
-        assistant_response = generated_text.replace("ASSISTANT:", "").strip()
+        generated_response = pipe(prompt_template)[0]['generated_text']
+        assistant_response = generated_response.split('\n\n')[1]
     
         return {"user": msg_prompt, "assistant": assistant_response}
 
@@ -115,8 +116,8 @@ async def api_home():
     """
     return {'detail': 'Welcome to Articko Bot!'}
 
-@app.post('/start_conversation/')
-async def start_conversation(request: Request):
+@app.post('/chat')
+async def chat(request: Request):
     """
     Starts a new conversation thread with a provided prompt.
     
@@ -143,8 +144,8 @@ async def start_conversation(request: Request):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.post('/custom_prompted_chat/')
-async def start_chat(request: Request):
+@app.post('/prompted_chat')
+async def prompted_chat(request: Request):
     """
     Starts a new chat thread with a provided user message prompt and persona description of the ai assistant .