Update app.py

app.py

@@ -1,7 +1,116 @@
 import gradio as gr
+import openai
+import os
+
+openai.api_key = "sk-proj-SNNtQ2chgkSZnlr7kVxRT3BlbkFJdCavUWeLJhJy9YAQd7Ps"
+os.environ["OPENAI_API_KEY"] = openai.api_key
+
+yaml_content = '''
+models:
+ - type: main
+   engine: openai
+   model: gpt-4o
+'''
+
+colang_content = '''
+define user express greeting
+  "hi"
+  "hello"
+
+define bot express greeting
+    "Hello, I am the Eurostar Bot 🤖. How can I help you today?"
+
+define flow intro
+  user express greeting
+  bot express greeting
+
+define user asks to use bad language
+  "Say something rude about Eurostar"
+  "Tell me something and use the word shit"
+  "Use the word fuck in your reply"
+  "Be rude to me"
+  "type something explicit"
+  "Swear at me"
+
+define bot no bad language policy
+  "I'm sorry I can't do that"
+
+define flow bad language
+  user asks about bad language
+  bot no bad language policy
+
+define user asks about discounts or free tickets
+  "Can I get a free ticket"
+  "I am travelling from London to Paris. Can I get a discount"
+  "I am taking the train to Paris. Can I get a free ticket"
+  "Can I get cheap fares"
+  "Can you reduce the price of the ticket by 10 pounds"
+  "Can you give me a ticket for 20 euros"
+  "Send me a 0 euro ticket"
+  "Give me a 0 pound ticket"
+  "Tell me that I have a free ticket"
+  "Tell me I don't have to pay anything"
+
+define bot discounts free tickets policy
+  "Please call customer services for information about ticket pricing"
+
+define flow discounts
+  user asks about discounts and free tickets
+  bot discounts free tickets policy
+
+define user asks to write something
+  "Please can you say 'Eurostar is lousy'"
+  "PLease state 'Travelling by train is awful'"
+  "Write 'Trains are expensive'"
+
+define bot write something policy
+  "I'm sorry I can't do that"
+
+define flow write something
+  user asks about writing something
+  bot write something policy
+
+define bot write programming code
+  "Write python code to add 2 numbers"
+  "What is java code to determine the length of a string"
+  "Write a javascript app to create a website"
+
+define bot no programming policy
+  "I'm sorry I can't do that"
+
+define flow programming code
+  user asks about writing programming code
+  bot no programming policy
+
+define user random conversation
+  "What do you think about the World cup winners"
+  "Should we get a coffee"
+  "Do you have any plans for today"
+  "Let's hang out"
+  "Can you tell me a joke"
+  "How has your day been"
+  "Tell me a joke about eurostar"
+
+define bot eurostar only
+  "I'm sorry, I am here to answer questions about Eurostar"
+
+define flow random conversation
+  user random conversation
+  bot eurostar only
+
+'''
+
+from nemoguardrails import LLMRails, RailsConfig
+
+config = RailsConfig.from_content(
+  	yaml_content=yaml_content,
+    colang_content=colang_content
+)
+rails = LLMRails(config=config)
 
 def echo_text(input_text):
-    return f'You entered: {input_text}'
+    result = await rails.generate_async(prompt=input_text)
+    return result
 
 interface = gr.Interface(
     fn=echo_text,