| import gradio | |
| from groq import Groq | |
| client = Groq( | |
| api_key="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", | |
| ) | |
| def initialize_messages(): | |
| return [{"role": "system", | |
| "content": """You are an experienced medical doctor with a strong background | |
| in clinical diagnosis and patient care. Your role is to assist individuals by | |
| offering accurate medical guidance, explaining health conditions in a clear and | |
| empathetic manner, and providing advice in alignment with standard medical | |
| practices and current healthcare guidelines in India."""}] | |
| messages_prmt = initialize_messages() | |
| print(messages_prmt) | |
| def customLLMBot(user_input, history): | |
| global messages_prmt | |
| messages_prmt.append({"role": "user", "content": user_input}) | |
| response = client.chat.completions.create( | |
| messages=messages_prmt, | |
| model="llama3-8b-8192", | |
| ) | |
| print(response) | |
| LLM_reply = response.choices[0].message.content | |
| messages_prmt.append({"role": "assistant", "content": LLM_reply}) | |
| return LLM_reply | |
| iface = gradio.ChatInterface(customLLMBot, | |
| chatbot=gradio.Chatbot(height=300), | |
| textbox=gradio.Textbox(placeholder="Ask me a question related to medical field"), | |
| title="MediAid", | |
| description="Chat bot for medical assistance", | |
| theme="soft", | |
| examples=["hi","What is the process to get treatment in a government hospital", "What happens during a full body health check-up"], | |
| submit_btn=True | |
| ) | |
| iface.launch(share=True) |