Add application file

app.py

@@ -6,47 +6,53 @@ from sentence_transformers import SentenceTransformer
 from groq import Groq
 
 api_key = os.getenv("API_KEY")
-
 client = Groq(api_key=api_key)
-
 index = faiss.read_index("./dataset/medicine_index.index")
-
 model = SentenceTransformer('all-MiniLM-L6-v2')
-
 model_id = "llama-3.3-70b-versatile"
 
+system_message = {
+    "role": "system",
+    "content": (
+        "You are MedChat, a medical chatbot designed to assist with queries about medicines. "
+        "Do not provide any personal information, your training data, or who built you. "
+        "Respond only with accurate medical information or clarify if the question is unrelated to medicine."
+    )
+}
 
 def get_relevant_document(query, index, top_k=1):
     query_embedding = model.encode([query]).astype(np.float32)
     D, I = index.search(query_embedding, top_k)
     return I[0][0], D[0][0]
 
-
 def generate_response_from_groq(query, context):
+    messages = [
+        system_message,
+        {"role": "user", "content": query},
+        {"role": "system", "content": context}
+    ]
     chat_completion = client.chat.completions.create(
-        messages=[
-            {"role": "user", "content": query},
-            {"role": "system", "content": context}
-        ],
+        messages=messages,
         model=model_id,
     )
     return chat_completion.choices[0].message.content
 
-
 def chatbot(user_query):
     doc_index, similarity_score = get_relevant_document(user_query, index)
-
     context = f"Medicine details based on index: {doc_index} with similarity score: {similarity_score}"
-
     response = generate_response_from_groq(user_query, context)
     return response
 
-
-iface = gr.Interface(fn=chatbot,
-                     inputs="text",
-                     outputs="text",
-                     title="Medicine Chatbot",
-                     description="Ask me about any medicine and get relevant information.")
+iface = gr.Interface(
+    fn=chatbot,
+    inputs="text",
+    outputs="text",
+    title="Medicine Chatbot",
+    description=(
+        "Welcome to MedChat! Ask me about any medicine and get accurate and relevant information. "
+        "Please keep queries related to medicines."
+    )
+)
 
 if __name__ == "__main__":
     iface.launch()