MDR / app.py
Gaautamm's picture
Update app.py
7e4f24b
raw
history blame
853 Bytes
import openai
import gradio as gr
# Set your OpenAI API key
openai.api_key = "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"
messages = [{"role": "system", "content": "You are a medical device classification expert based on the risk level as per Indian regulation"}]
def CustomChatGPT(user_input):
messages.append({"role": "user", "content": user_input})
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=messages
)
ChatGPT_reply = response["choices"][0]["message"]["content"]
messages.append({"role": "assistant", "content": ChatGPT_reply})
return ChatGPT_reply
# Define your Gradio interface here with custom port range
gr.Interface(fn=CustomChatGPT, inputs="text", outputs="text", title="Medical Device Regulation Chat Bot").launch(share=True, port=8888, websocket_port=8889)