Update gradio_nlp_group_project.py

gradio_nlp_group_project.py

@@ -53,7 +53,14 @@ class TextProcessor:
 
 def get_transcript(video_id):
 
-  transcripts = YouTubeTranscriptApi.list_transcripts(video_id)
+    try:
+        transcripts = YouTubeTranscriptApi.list_transcripts(video_id)
+    except NoTranscriptFound:
+        return "No transcript found for this video.", 'en'
+    except TranscriptsDisabled:
+        return "Transcripts are disabled for this video.", 'en'
+    except Exception as e:
+        return f"An error occurred: {str(e)}", 'en'
 
   available_languages = []
 
@@ -79,38 +86,36 @@ def get_transcript(video_id):
 
 def process_text(video_id):
     transcript, language = get_transcript(video_id)
-    if "The following languages were not available" in transcript:
+    if transcript.startswith("An error occurred:") or "No transcript" in transcript:
         return {"Error": transcript, "Language Detected": "None"}
-    processor = TextProcessor(transcript)
-
-    results = {"Language Detected": language}  # Include language in the output for debugging
 
-    if language == 'en':
+    processor = TextProcessor(transcript)
+    try:
         summarized_text = processor.summarize_text(transcript)
         translated_text = processor.translate_text(summarized_text)
         classification_result = processor.classify_text(summarized_text)
-        results.update({
+        results = {
+            'Language Detected': language,
             'Summarized Text': summarized_text,
             'Translated Text': translated_text,
             'Classification Result': classification_result
-        })
-    else:
-        results.update({'Error': 'Unsupported language'})
+        }
+    except Exception as e:
+        results = {'Error': f"An error occurred during processing: {str(e)}"}
 
     return results
-
+    
 iface = gr.Interface(
     fn=process_text,
     inputs=[gr.Textbox(label="YouTube Video ID")],
     outputs=[gr.JSON(label="Results")],
     title="Text Processing App with YouTube Transcript",
-    description="This app allows you to fetch, summarize, translate, and classify YouTube video transcripts based on their language. Currently, English to Spanish Translation is currently supported."
+    description="This app allows you to fetch, summarize, translate, and classify YouTube video transcripts. Errors are handled and displayed."
 
 )
 
 def main():
-
-  iface.launch()
+    iface.launch()
 
 if __name__ == '__main__':
-  main()
+    main()