Create app.py

app.py

@@ -0,0 +1,80 @@
+from openai import OpenAI
+from dotenv import load_dotenv
+import gradio as gr
+import MAIAI
+
+# Load environment variables
+load_dotenv()
+
+def chat(text, history, native_language, language, persona, tone = "Casual", model = "gpt-4o-mini"):
+    
+    print(tone, native_language, language, persona)
+    
+    casual = "This is in a casual, internet texting context, use of local slangs is encouraged." if tone == "Casual" else ""
+    
+    teacher = MAIAI.Agent(model=model, temperature=0.5, role=f"You are a {language} teacher teaching {native_language} speaking student.")
+    responder = MAIAI.Agent(model=model, temperature=0.5, role=f"""You are a {language} {persona}. Respond to the user's text in {language} and give {native_language} translation. Refer to Chat History for context. Prompt the user to keep the conversation going. {casual}""")
+    
+    feedback_task = MAIAI.Task(
+        agent=teacher,
+        goal=f"""Text: {text}
+
+Correct any linguistic error in the text and give example driven feedback on how to improve the text in {language}.
+Translate any non-{language} from the text into {language}.
+You MUST give your feedback in {native_language}.
+{casual}
+"""
+    )
+    
+    respond_task = MAIAI.Task(
+        agent=responder,
+        goal=f"""{text}
+        
+        Respond to the text above in {language} and give {native_language} translation.
+        Refer to Chat History for context.
+        Chat History: {history}"""
+    )
+    
+    feedback = feedback_task.execute()
+    
+    yield f"""Feedback:
+{feedback}"""
+
+    response = respond_task.execute()
+    
+    output = f"""
+***Feedback:***
+
+{feedback}
+
+-----------
+
+***{persona}:***
+
+{response}
+    """
+
+    yield output
+
+# Sample Function Call ------------------------------------
+    
+# feedback,response = chat("Soy jugando Demonslayer! Y tu?", "English","Spanish","friendly lady",,casual_tone="Casual")
+
+# print(f"""
+# Feedback: {feedback}
+
+# Reply: {response}
+#       """)
+
+# Gradio ----------------------------------------------------
+chat_interface = gr.ChatInterface(
+    fn=chat, 
+    additional_inputs=[
+        gr.components.Dropdown(choices=["English","中文","Spanish"], value="English", label="I speak"),
+        gr.components.Dropdown(choices=["English","中文","Spanish"], value="English", label="I want to learn"), 
+        gr.components.Textbox(value = "LinguAI Chatbot", label="I want to talk to"),
+        gr.components.Dropdown(choices=["Casual","Formal"], value="Casual", label="Tone")],
+    title="LinguAI"
+)
+
+chat_interface.launch()