test

app.py

@@ -30,35 +30,33 @@ def load_and_convert_audio(audio_path):
 def process_audio(audio_path):
     """Process audio file and return transcription and summary"""
     results = {}
-    temp_wav_path = load_and_convert_audio(audio_path)
-
-    # Transcription
-    transcription = models['transcriber'](temp_wav_path, return_timestamps=True)
-    if isinstance(transcription, dict):
-        results['transcription'] = transcription['text']
-    else:
-        results['transcription'] = ' '.join([chunk['text'] for chunk in transcription])
-
-    # Summarization
-    text = results['transcription']
-    words = text.split()
-    chunk_size = 1000
-    chunks = [' '.join(words[i:i + chunk_size]) for i in range(0, len(words), chunk_size)]
     
-    summaries = []
-    for chunk in chunks:
-        summary = models['summarizer'](chunk, max_length=200, min_length=50, truncation=True)
-        summaries.append(summary[0]['summary_text'])
-    
-    combined_summary = ' '.join(summaries)
-    results['summary'] = combined_summary
+    try:
+        temp_wav_path = load_and_convert_audio(audio_path)
+
+        # Transcription
+        transcription = models['transcriber'](temp_wav_path, return_timestamps=True)
+        results['transcription'] = transcription['text'] if isinstance(transcription, dict) else ' '.join([chunk['text'] for chunk in transcription])
+
+        # Summarization
+        text = results['transcription']
+        words = text.split()
+        chunk_size = 1000
+        chunks = [' '.join(words[i:i + chunk_size]) for i in range(0, len(words), chunk_size)]
 
-    # Clean up temporary WAV file
-    if os.path.exists(temp_wav_path):
-        os.unlink(temp_wav_path)
+        summaries = [models['summarizer'](chunk, max_length=200, min_length=50, truncation=True)[0]['summary_text'] for chunk in chunks]
+        results['summary'] = ' '.join(summaries)
+
+    except Exception as e:
+        return {'error': str(e)}, 500  # Return error message if something goes wrong
+
+    finally:
+        if os.path.exists(temp_wav_path):
+            os.unlink(temp_wav_path)
 
     return results
 
+
 @app.route('/process-audio', methods=['POST'])
 def process_audio_endpoint():
     """API endpoint to process audio file"""