Spaces:
Sleeping
Sleeping
File size: 2,797 Bytes
940d88e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import gradio as gr
import openai
import os
from functools import partial
import chat_agent
from langchain.schema import (
HumanMessage
)
def set_api_key(api_key):
openai.api_key = api_key
os.environ["OPENAI_API_KEY"] = api_key
return "API Key set successfully."
def get_response(chatbot, api_key, selected_model, user_input, conversation_history=""):
set_api_key(api_key)
# Preserve the memory of the current chatbot
preserved_memory = chatbot.memory
# Create a new chat agent based on the selected model and seed the memory
chatbot = chat_agent.create_chatbot(model_name=selected_model, seed_memory=preserved_memory)
# Get raw chat response
response = chatbot.agent.run(user_input).strip()
# Iterate through messages in ChatMessageHistory and format the output
updated_conversation = '<div style="background-color: hsl(30, 100%, 30%); color: white; padding: 5px; margin-bottom: 10px; text-align: center; font-size: 1.5em;">Chat History</div>'
for i, message in enumerate(chatbot.memory.chat_memory.messages):
if isinstance(message, HumanMessage):
prefix = "User: "
background_color = "hsl(0, 0%, 40%)" # Dark grey background
text_color = "hsl(0, 0%, 100%)" # White text
else:
prefix = "Chatbot: "
background_color = "hsl(0, 0%, 95%)" # White background
text_color = "hsl(0, 0%, 0%)" # Black text
updated_conversation += f'<div style="color: {text_color}; background-color: {background_color}; margin: 5px; padding: 5px;">{prefix}{message.content}</div>'
return updated_conversation
def main():
api_key = os.environ.get("OPENAI_API_KEY") # "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"
api_key_input = gr.components.Textbox(
lines=1,
label="Enter OpenAI API Key",
value=api_key,
type="password",
)
model_selection = gr.components.Dropdown(
choices=["gpt-3.5-turbo"],
label="Select a GPT Model",
value="gpt-3.5-turbo",
)
user_input = gr.components.Textbox(
lines=3,
label="Enter your message",
)
output_history = gr.outputs.HTML(
label="Updated Conversation",
)
chatbot = chat_agent.create_chatbot(model_name=model_selection.value) # model_selection.value
inputs = [
api_key_input,
model_selection,
user_input,
]
iface = gr.Interface(
fn=partial(get_response, chatbot),
inputs=inputs,
outputs=[output_history],
title="Everything About Martin SeligMan & Positive Psychology",
description=" OTHER PEOPLE MATTER ",
allow_flagging="never",
)
iface.launch()
if __name__ == "__main__":
main()
|