dindizz commited on
Commit
f63c480
·
verified ·
1 Parent(s): b10c901

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -1,22 +1,25 @@
1
  import gradio as gr
2
- import openai
3
  from deep_translator import GoogleTranslator
4
  import os
5
 
6
- # Initialize OpenAI API
7
- openai.api_key = os.environ.get("OPENAI_API_KEY")
8
 
9
  def translate_text(text, target_lang):
10
  translator = GoogleTranslator(source='auto', target=target_lang)
11
  return translator.translate(text)
12
 
13
  def get_llm_response(prompt):
14
- response = openai.Completion.create(
15
- engine="text-davinci-002",
16
- prompt=prompt,
 
 
 
17
  max_tokens=150
18
  )
19
- return response.choices[0].text.strip()
20
 
21
  def process_input(input_text, output_lang):
22
  # Translate input to English
 
1
  import gradio as gr
2
+ from openai import OpenAI
3
  from deep_translator import GoogleTranslator
4
  import os
5
 
6
+ # Initialize OpenAI client
7
+ client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
8
 
9
  def translate_text(text, target_lang):
10
  translator = GoogleTranslator(source='auto', target=target_lang)
11
  return translator.translate(text)
12
 
13
  def get_llm_response(prompt):
14
+ response = client.chat.completions.create(
15
+ model="gpt-4-0125-preview",
16
+ messages=[
17
+ {"role": "system", "content": "You are a helpful assistant that provides informative and concise responses."},
18
+ {"role": "user", "content": prompt}
19
+ ],
20
  max_tokens=150
21
  )
22
+ return response.choices[0].message['content'].strip()
23
 
24
  def process_input(input_text, output_lang):
25
  # Translate input to English