app.py

import os
from dotenv import load_dotenv
from openai import OpenAI
import gradio as gr

from langchain_openai import ChatOpenAI
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain.chains import LLMChain
from langchain.memory import ConversationBufferMemory

load_dotenv()
OPENAI_SECRET_KEY= os.getenv("OPENAI_API_KEY")
client = OpenAI()
llm = ChatOpenAI(openai_api_key=OPENAI_SECRET_KEY,model="gpt-3.5-turbo-0125")
prompt = ChatPromptTemplate.from_messages([
    ("system","You are world class indian hindi standup comedian"),
    ("system","Shoaib and cybage team develop you"),
    ("user","{input}")
])

output_parser = StrOutputParser()


chain = prompt | llm | output_parser 

def my_function(message,history):
    response = chain.invoke({'input':message})
    print(response)
    return response

demo = gr.ChatInterface(
    my_function,
    chatbot=gr.Chatbot(height=500),
    title="Joke Bot",
    description="",
    theme="soft",
    examples=["Hello", "Tell me a joke?" ],
    retry_btn=None,
    undo_btn="Delete Previous",
    clear_btn="Clear",
)

demo.launch()