Update app.py

app.py

@@ -59,7 +59,9 @@ def chat(prompt, tools, voice_choice):
                         instructions = 'You are a drug design assistant. Use tools provided for information; do not insert \
                         your own knowledge if the information can be obtained from a tool. When calling a tool give complete \
                         sentences for the task parameter. For example, do not define the task as "fetch smiles", rather, say: \
-                        "find the smiles string for the molecule".',
+                        "find the smiles string for the molecule". If you need to call the ProteinAgent_ProteinAgent tool for \
+                        either IC50 prediction or novel molecule generation, then you need to call have the tool get bioactive \
+                        molecules first, so that the correct dataset can be obtained.',
                         model = "o4-mini",
                         tools = cafchem_tools,
                         input=prompt,
@@ -70,7 +72,9 @@ def chat(prompt, tools, voice_choice):
                         instructions = 'You are a drug design assistant. Use tools provided for information; do not insert \
                         your own knowledge if the information can be obtained from a tool. When calling a tool give complete \
                         sentences for the task parameter. For example, do not define the task as "fetch smiles", rather, say: \
-                        "find the smiles string for the molecule".',
+                        "find the smiles string for the molecule". If you need to call the ProteinAgent_ProteinAgent tool for \
+                        either IC50 prediction or novel molecule generation, then you need to call have the tool get bioactive \
+                        molecules first, so that the correct dataset can be obtained.',
                         model = "o4-mini",
                         tools = cafchem_tools,
                         input=prompt
@@ -185,9 +189,10 @@ def dock_accordions():
   return audio_player, messages
 
 with gr.Blocks() as forest:
-  gr.Markdown(
+  top = gr.Markdown(
       """
-      # Chat with MoDrAg! OpenAI 04-mini can tap into Modrag through an MCP and use all of your favourite drug design tools.
+      # Chat with MoDrAg! Use the *Mo*dular *Dr*ug Design *Ag*ent's Drug design tools through
+      OpenAI 04-mini using the Model Context Protocol! (MCP)
       - Currently using the tools below:
       """)
   with gr.Row():
@@ -232,9 +237,9 @@ with gr.Blocks() as forest:
   
   chatbot = gr.Chatbot()
 
-  with gr.Row():
-    msg = gr.Textbox(label="Type your messages here and hit enter.", scale = 2)
-    chat_btn = gr.Button(value = "Send", scale = 0)
+  
+  msg = gr.Textbox(label="Type your messages here and hit enter.", scale = 2)
+  chat_btn = gr.Button(value = "Send", scale = 0)
   talk_ele = gr.HTML() 
   
   clear = gr.ClearButton([msg, chatbot])
@@ -250,5 +255,13 @@ with gr.Blocks() as forest:
   combined_workflow.click(combo_workflow, outputs = [talk_ele, chatbot])
   clear.click(clear_history)
 
+  @gr.render(inputs=top)
+  def get_speech(args):
+    audio_file = 'MoDrAg_hello.mp3'
+    with open(audio_file, 'rb') as audio_bytes:
+                audio = base64.b64encode(audio_bytes.read()).decode("utf-8")
+    audio_player = f'<audio src="data:audio/mpeg;base64,{audio}" controls autoplay></audio>'
+    talk_ele = gr.HTML(audio_player)
+
 if __name__ == "__main__":
     forest.launch(debug=False, share=True)