Update app.py

app.py

@@ -61,19 +61,19 @@ def text_to_speech(text):
     return wav, rate
 
 def process_pdf(uploaded_file, hf_model_name):
-    """
-    Processes the uploaded PDF file to extract, summarize the abstract, and convert it to speech.
-    """
-    abstract = extract_and_clean_abstract(uploaded_file)
-    summary = summarize_text(hf_model_name, abstract)
-    wav, rate = text_to_speech(summary)
-    sf.write('/tmp/speech_output.wav', wav, rate)
-    return '/tmp/speech_output.wav'
+    if uploaded_file.name.lower().endswith('.pdf'):
+        abstract = extract_and_clean_abstract(uploaded_file)
+        summary = summarize_text(hf_model_name, abstract)
+        wav, rate = text_to_speech(summary)
+        sf.write('/tmp/speech_output.wav', wav, rate)
+        return '/tmp/speech_output.wav'
+    else:
+        return "Error: Please upload a PDF file."
 
 iface = gr.Interface(
     fn=process_pdf,
     inputs=[
-        gr.File(label="Upload PDF", type="pdf"),
+        gr.File(label="Upload PDF"),
         gr.Textbox(label="Hugging Face Model Name for Summarization")
     ],
     outputs=gr.Audio(label="Audio Summary"),