Update app.py

app.py

@@ -8,11 +8,24 @@ from moviepy.editor import VideoFileClip
 # Set API key and organization ID from environment variables
 openai.api_key = os.getenv('OPENAI_API_KEY')
 openai.organization = os.getenv('OPENAI_ORG_ID')
+client = OpenAI(api_key= os.getenv('OPENAI_API_KEY'))
 
 # Define the model to be used
 MODEL = "gpt-4o"
 
 def process_text():
+    text_input = st.text_input("Enter your text:")
+    if text_input:
+        completion = client.chat.completions.create(
+            model=MODEL,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant. Help me with my math homework!"},
+                {"role": "user", "content": f"Hello! Could you solve {text_input}?"}
+            ]
+        )
+        st.write("Assistant: " + completion.choices[0].message.content)
+
+def process_text_old():
     text_input = st.text_input("Enter your text:")
     if text_input:
         response = openai.Completion.create(