Update app.py

app.py

@@ -1,10 +1,9 @@
-from mistralai import Mistral
+import cohere
 import gradio as gr
 import os
 from datetime import datetime
-api_key = os.environ["MISTRAL_API_KEY"]
-model = "mistral-large-latest"
-client = Mistral(api_key=api_key)
+MODEL = "command-a-03-2025"
+co = cohere.ClientV2(api_key=os.environ["COHERE_API_KEY"])
 def interaction_check(prompt):
     """
     Check for drug interactions and provide pharmacist guidance
@@ -38,8 +37,8 @@ Important guidelines:
 
 Format your response with clear sections and use emojis for visual clarity where appropriate."""
 
-        chat_response = client.chat.complete(
-            model=model,
+        chat_response = co.chat(
+            model=MODEL,
             messages=[
                 {
                     "role": "system",
@@ -54,7 +53,7 @@ Format your response with clear sections and use emojis for visual clarity where
             max_tokens=1000   # Ensure comprehensive responses
         )
 
-        response = chat_response.choices[0].message.content
+        response = chat_response.message.content[0].text
 
         # Add disclaimer footer
         disclaimer = "\n\n---\n⚠️ **IMPORTANT DISCLAIMER**: This information is for educational purposes only and should not replace professional medical advice. Always consult with a licensed pharmacist or healthcare provider before making any changes to your medications."
@@ -120,7 +119,7 @@ with gr.Blocks(css=custom_css, title="💊 PharmaSafe - Drug Interaction Checker
         # Header
         gr.Markdown(
             """
-            # 💊 💊 PharmaSafe - Drug Interaction Checker
+            # 💊 💊 PharmaSafety - Drug Interaction Checker
 
             Get preliminary guidance on potential drug interactions and medication safety concerns.
             This tool helps identify possible interactions between medications.