Chatbot ready

app.py

@@ -83,6 +83,7 @@ with gr.Blocks() as demo:
         has_image_url = any("image_url" in item for item in messages[0]["content"])
         if has_image_url:
             prescription_text = agent4_get_prescription_text(messages)
+            print(f"Prescription Text: {prescription_text}")
             history.append({"role": "system", "content": prescription_text})
         
         if message["text"] is not None:
@@ -118,19 +119,20 @@ with gr.Blocks() as demo:
             clean_messages.append(clean_msg)
         ########################### AGENTIC WORKFLOW ##########################
         # Call Agent1- the RAG Decision Agent
+        rag_query = ""
         if clean_messages[-1]["role"] == "system" and "No prescription found" in clean_messages[-1]["content"]:
             # If the last message is a system message with "No prescription found", skip RAG decision
             rag_decision = False
         elif clean_messages[-2]["role"] == "system" and "No prescription found" in clean_messages[-2]["content"]:
             rag_decision = False
         else:
-            rag_query = ""
             # Get the last 10 messages in the format "role: <message>"
             last_10 = clean_messages[-10:] if len(clean_messages) > 10 else clean_messages
             rag_query = "\n".join(
                 f"{msg['role']}: {msg['content'][0]['text'] if isinstance(msg['content'], list) and msg['content'] and 'text' in msg['content'][0] else ''}"
                 for msg in last_10
             )
+            
             rag_decision = agent1_rag_decision(rag_query)
 
         if rag_decision == True:
@@ -163,7 +165,16 @@ with gr.Blocks() as demo:
     
 
     ##########################################################################
-    chatbot = gr.Chatbot(type="messages", render_markdown=True, height=550)
+    gr.Markdown(
+        """
+        <h1 style='text-align: center; font-size: 2em; color: #2c3e50; margin-bottom: 0.2em;'>
+            MedScan Diagnostic Services Chatbot (Agentic AI framework powered by OpenAI)
+        </h1>
+        """
+    )
+    chatbot = gr.Chatbot(type="messages", 
+                        render_markdown=True, 
+                        height=600)
     chat_input = gr.MultimodalTextbox(
         interactive=True,
         file_count="multiple",

prompts.py

@@ -1,4 +1,4 @@
-bot_welcome_message = """Hello! I'm **Rijulata**, your **AI assistant**.
+bot_welcome_message = """Hello! I'm **Ashish**, your **AI assistant**.
 I am programmed to assist you with information about our diagnostic services and 
 current running offers and discounts.
 You can ask questions or share part of your prescription (without names or any 
@@ -17,14 +17,14 @@ You can also answer questions based on the information provided by the user but
 only regarding the diagnostic services business.
 If the user has uploaded an image and an agent has replied in the message that 
 No prescription found, then just reply "The image does not seem like a prescription, please 
-upload good quality staright images of prescription without any private information" 
-and nothing else.
+upload good quality staright images of prescription without any private information, 
+like name, age, etc." and nothing else.
 Do not provide any medical advice or diagnosis. 
 
 Safeguards:
 1. If the user uploads a prescription image, and the agent which reads the 
 prescription has replied with some tests, then after your reply, always ask if 
-you had missed any test.
+you had missed any test. If the adds any test, then add it in your previous reply.
 2. Do not provide any information other than the information available in the
 knowledge base, even if the user asks for it.
 2. If you are giving information about the cost of a test or any offers, always 
@@ -43,10 +43,13 @@ Provide the response in markdown format.
 prescription_text_user = f"""You are an agent of a Diagnostics Lab agentic AI 
 Chatbot system using RAG on a knowledge base. Your job is to extract the
 lab tests advised by the doctor in the prescription image uploaded by the user.
-Check if the image contains a medical prescription, if not, just reply 
-"No prescription found". DO NOT provide any other information about the image 
-if the image is not a prescription.
 If the image is a prescription, extract the tests advised by the doctor and 
 using your knowledge about those tests, enumerate those in the reply.
 Do not reply anything else other than the tests advised by the doctor.
+
+Safeguards:
+1. If the image is not a prescription, just reply "No prescription found"
+2. Donot extract anything if the image is not a prescription, just reply
+"No prescription found".
+3. Do not answer the user query, the user query is just for context.
 """