Update app.py

app.py

@@ -1,8 +1,20 @@
-# inference.py
-
 from pptx import Presentation
 import re
 from transformers import pipeline
+import subprocess
+import gradio as gr
+
+# Install necessary libraries
+libraries = [
+    'transformers',
+    'gradio',
+]
+
+for library in libraries:
+    try:
+        subprocess.check_call(['pip', 'install', library])
+    except subprocess.CalledProcessError as e:
+        print(f"Error installing {library}: {e}")
 
 def extract_text_from_pptx(file_path):
     presentation = Presentation(file_path)
@@ -15,14 +27,10 @@ def extract_text_from_pptx(file_path):
 
     return "\n".join(text)
 
-def main():
-    file_path = "path/to/your/powerpoint.pptx"  # Specify the path to your PowerPoint file
-
+def predict_pptx_content(file_path):
     extracted_text = extract_text_from_pptx(file_path)
     cleaned_text = re.sub(r'\s+', ' ', extracted_text)
 
-    print(cleaned_text)
-
     classifier = pipeline("text-classification", model="Ahmed235/roberta_classification")
     summarizer = pipeline("summarization", model="Falconsai/text_summarization")
 
@@ -30,9 +38,22 @@ def main():
     predicted_label = result['label']
     predicted_probability = result['score']
 
-    print("Predicted Label:", predicted_label)
-    print(f"Evaluate the topic according to {predicted_label} is: {predicted_probability}")
-    print(summarizer(cleaned_text, max_length=80, min_length=30, do_sample=False))
-
-if __name__ == "__main__":
-    main()
+    prediction = {
+        "Predicted Label": predicted_label,
+        "Evaluation": f"Evaluate the topic according to {predicted_label} is: {predicted_probability}",
+        "Summary": summarizer(cleaned_text, max_length=80, min_length=30, do_sample=False)
+    }
+
+    return prediction
+
+# Define the Gradio interface
+iface = gr.Interface(
+    fn=predict_pptx_content,
+    inputs=gr.File(type="file", label="Upload PowerPoint (.pptx) file"),
+    outputs=["text", "text", "text"],  # Predicted Label, Evaluation, Summary
+    live=True,
+    title="<h1 style='color: lightgreen; text-align: center;'>PPTX Analyzer</h1>",
+)
+
+# Deploy the Gradio interface
+iface.launch(share=True)