Spaces:
Running
Running
# Path: chatbot_model_based_autocorrect.py | |
import os, getpass | |
from langchain_openai import ChatOpenAI | |
from langchain_core.messages import HumanMessage | |
import gradio as gr | |
openai.api_key = os.getenv("OPENAI_API_KEY") | |
# Define the language model | |
model = ChatOpenAI(model="gpt-4o-mini") | |
# Function to generate a conversational response with model-based autocorrection | |
def chatbot_autocorrect_response(input_text: str): | |
# Step 1: Define the prompt asking the model to correct the sentence | |
prompt = ( | |
f"The user said: '{input_text}'. Please correct this sentence if necessary, and make it sounds friendly, casual tone, acknowledging the correction and make it sounds like american native conversation. If appropriate, make it sounds like an IELTS 9.0 level response. Please only respond with the corrected sentence. If nothing needs to be changed, repeat the sentence." | |
) | |
# Step 2: Send the prompt to the model | |
human_message = HumanMessage(content=prompt) | |
response = model.invoke([human_message]) | |
# Step 3: Return the response from the model | |
return response.content | |
# Example usage | |
def gradio_chatbot(input_text): | |
# Pass the user's input to the chatbot function and get the response | |
return chatbot_autocorrect_response(input_text) | |
# Launch Gradio interface | |
interface = gr.Interface(fn=gradio_chatbot, inputs="text", outputs="text", title="Chatbot with Auto-Correction") | |
interface.launch() |