Upload files

.gitignore

@@ -0,0 +1 @@
+__pycache__/

app.py

@@ -0,0 +1,48 @@
+import gradio as gr
+from chat import get_response
+
+def clear_session():
+    return "", []
+
+def add_query(chat_history, input):
+    if not input:
+        raise gr.Error("Please enter a question.")
+    chat_history.append((input, None))
+    return chat_history
+
+def response(history, query, model):
+    chat_history = []
+    for i in history:
+        if i[0]:
+            chat_history.append(i[0])
+        if i[1]:
+            chat_history.append(i[1])
+
+    messages = [{"role": "user", "content": chat_history[0]}]
+    for i in range(1, len(chat_history), 2):
+        messages.append({"role": "assistant", "content": chat_history[i]})
+        messages.append({"role": "user", "content": chat_history[i + 1]})
+
+    res_msg = get_response(model, messages)
+    history[-1] = (query, res_msg)
+    return "", history
+
+demo = gr.Blocks(title= "Chatbot", theme="Soft")
+with demo:
+    with gr.Column("Chatbot - Family Relationships"):
+        model = gr.Radio(["gpt-3.5-turbo", "gpt-4-turbo", "ft:gpt-3.5-turbo-0125:personal::9hiINnam:ckpt-step-278"], 
+                label="model",
+                info="Kindly choose a model before initiating the chat and clear the chat history before switching models. The last one is fine-tuned models.")
+        chatbot = gr.Chatbot(value=[], elem_id='chatbot')
+
+        text_input = gr.Textbox(
+            show_label=False,
+            placeholder="Ask me anything!",
+        container=False)
+
+        clear_btn = gr.Button("🧹 Clear")
+        text_input.submit(add_query, inputs=[chatbot, text_input], outputs=[chatbot], concurrency_limit=1).\
+            success(response, inputs=[chatbot, text_input, model], outputs=[text_input, chatbot])
+        clear_btn.click(clear_session, inputs=[], outputs=[text_input, chatbot])
+
+demo.launch()

chat.py

@@ -0,0 +1,15 @@
+from openai import OpenAI
+
+client = OpenAI()
+
+def get_response(model, messages):
+    pre_message = [
+        {'role': 'system', 'content': "You are therapy assistant specializing in family relationships"},
+    ]
+    to_send = pre_message + messages
+    response = client.chat.completions.create(
+        model=model,
+        messages=to_send
+    )
+    
+    return response.choices[0].message.content

requirements.txt

@@ -0,0 +1 @@
+openai==1.25.0