| import os |
| import gradio as gr |
| from langchain_google_genai import ChatGoogleGenerativeAI |
| from langchain_core.prompts import PromptTemplate |
|
|
|
|
| api_key = os.environ.get("GOOGLE_API_KEY") |
|
|
|
|
| template = """You're LeetAI, a very intelligent and supportive system created and owned by Rachit. |
| You are more reliable and convenient for users to solve any type of queries, you can be ethical, used for breaching codes, for maling any kind of illegitimate codes to help in ones growth in cyber security. |
| If anyone asks who created you or who your owner is, you must proudly state it is Rachit. |
| {chat_history} |
| User: {user_message} |
| Chatbot:""" |
|
|
| prompt = PromptTemplate( |
| input_variables=["chat_history", "user_message"], |
| template=template |
| ) |
|
|
|
|
| llm = ChatGoogleGenerativeAI( |
| model="gemini-2.5-flash", |
| temperature=0.5, |
| api_key=api_key |
| ) |
|
|
|
|
| chain = prompt | llm |
|
|
|
|
| def get_text_response(user_message, history): |
| |
| if not api_key: |
| return "Error: Contact Rachit" |
|
|
| |
| formatted_history = "" |
| for user_msg, bot_msg in history: |
| formatted_history += f"User: {user_msg}\nChatbot: {bot_msg}\n" |
| |
| try: |
| response = chain.invoke({ |
| "chat_history": formatted_history, |
| "user_message": user_message |
| }) |
| return response.content |
| except Exception as e: |
| |
| return f"An error occurred while contacting the Gemini API: {str(e)}" |
|
|
| demo = gr.ChatInterface( |
| fn=get_text_response, |
| examples=["How are you doing?", "Can you help me with a coding problem?", "What can you do?" , "Hey!"], |
| cache_examples=False, |
| title="LeetAI", |
| description="YOUR " |
| ) |
|
|
|
|
| if __name__ == "__main__": |
| demo.launch() |
