deadshot2003 commited on
Commit
daf8fe4
·
verified ·
1 Parent(s): 338b259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -13,8 +13,8 @@ import openai
13
  # Load environment variables
14
  load_dotenv()
15
 
16
- # Initialize the OpenAI client
17
- client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
18
 
19
  # Set up logging
20
  logging.basicConfig(level=logging.INFO)
@@ -30,18 +30,17 @@ def transcribe_audio(file):
30
  logging.info("Transcribing audio file")
31
  file = convert_to_supported_format(file)
32
  logging.info("Converted file to WAV format")
33
-
34
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_file:
35
  temp_file.write(file.getvalue())
36
  temp_file_path = temp_file.name
37
 
38
  try:
39
- with open(temp_file_path, "rb") as audio_file:
40
- transcript = client.audio.transcriptions.create(
41
- model="whisper-1",
42
- file=audio_file
43
- )
44
- return transcript.text
45
  except Exception as e:
46
  logging.error(f"Error in transcription: {str(e)}")
47
  return f"Error in transcription: {str(e)}"
@@ -63,7 +62,7 @@ def get_transcript(url):
63
 
64
  def summarize_text(text):
65
  try:
66
- response = client.chat.completions.create(
67
  model="gpt-3.5-turbo",
68
  messages=[
69
  {"role": "system", "content": "You are a helpful assistant."},
@@ -71,7 +70,7 @@ def summarize_text(text):
71
  ],
72
  max_tokens=150
73
  )
74
- return response.choices[0].message.content.strip()
75
  except Exception as e:
76
  return f"Error in summarizing text: {str(e)}"
77
 
@@ -100,3 +99,5 @@ elif option == "Upload audio/video file":
100
  transcript_text = transcribe_audio(uploaded_file)
101
  summary = handle_summary(transcript_text)
102
  st.text_area("Summary", summary, height=200)
 
 
 
13
  # Load environment variables
14
  load_dotenv()
15
 
16
+ # Set up OpenAI API
17
+ openai.api_key = os.getenv("OPENAI_API_KEY")
18
 
19
  # Set up logging
20
  logging.basicConfig(level=logging.INFO)
 
30
  logging.info("Transcribing audio file")
31
  file = convert_to_supported_format(file)
32
  logging.info("Converted file to WAV format")
33
+
34
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_file:
35
  temp_file.write(file.getvalue())
36
  temp_file_path = temp_file.name
37
 
38
  try:
39
+ with warnings.catch_warnings():
40
+ warnings.simplefilter("ignore")
41
+ with open(temp_file_path, "rb") as audio_file:
42
+ transcript = openai.Audio.transcribe("whisper-1", audio_file)
43
+ return transcript["text"]
 
44
  except Exception as e:
45
  logging.error(f"Error in transcription: {str(e)}")
46
  return f"Error in transcription: {str(e)}"
 
62
 
63
  def summarize_text(text):
64
  try:
65
+ response = openai.ChatCompletion.create(
66
  model="gpt-3.5-turbo",
67
  messages=[
68
  {"role": "system", "content": "You are a helpful assistant."},
 
70
  ],
71
  max_tokens=150
72
  )
73
+ return response['choices'][0]['message']['content'].strip()
74
  except Exception as e:
75
  return f"Error in summarizing text: {str(e)}"
76
 
 
99
  transcript_text = transcribe_audio(uploaded_file)
100
  summary = handle_summary(transcript_text)
101
  st.text_area("Summary", summary, height=200)
102
+ if __name__ == "__main__":
103
+ main()