Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import pipeline
-from pptx import Presentation  # Import the Presentation class
+from pptx import Presentation
 import re
 
 # Create a text classification pipeline
@@ -16,20 +16,28 @@ def extract_text_from_pptx(file_path):
                 text.append(shape.text)
     return "\n".join(text)
 
+def limit_text_length(text, max_length=512):
+    # Truncate or limit the text length
+    return text[:max_length]
+
 def predict_pptx_content(file_path):
     try:
         extracted_text = extract_text_from_pptx(file_path)
         cleaned_text = re.sub(r'\s+', ' ', extracted_text)
+        
+        # Limit text length before classification
+        limited_text = limit_text_length(cleaned_text)
+
         # Perform inference using the pipeline
-        result = classifier(extracted_text)
+        result = classifier(limited_text)
 
         predicted_label = result[0]['label']
         predicted_probability = result[0]['score']
-        summary = summarizer(extracted_text, max_length=222, min_length=30, do_sample=False)[0]['summary_text']
+        summary = summarizer(extracted_text, max_length=80, min_length=30, do_sample=False)[0]['summary_text']
         prediction = {
             "Summary": summary,
             "Evaluation": f"Evaluate the topic according to {predicted_label} is: {predicted_probability}",
-            "Predicted_Label": predicted_label,  # Adjusted key to match Gradio output key
+            "Predicted_Label": predicted_label,
         }
 
         return prediction
@@ -43,8 +51,8 @@ def predict_pptx_content(file_path):
 iface = gr.Interface(
     fn=predict_pptx_content,
     inputs=gr.File(type="filepath", label="Upload PowerPoint (.pptx) file"),
-    outputs=["text"],  # Adjusted output keys
-    live=False,  # Change to True for one-time analysis
+    outputs=[gr.Textbox("Summary"), gr.Textbox("Evaluation"), gr.Textbox("Predicted_Label")],
+    live=False,
     title="<h1 style='color: lightgreen; text-align: center;'>HackTalk Analyzer</h1>",
 )