Update app.py

app.py

@@ -7,14 +7,6 @@ import os
 model = AutoModelForSequenceClassification.from_pretrained("Reem333/Citaion-Classifier")
 tokenizer = AutoTokenizer.from_pretrained("allenai/longformer-base-4096")
 
-def extract_text_from_pdf(file_path):
-    text = ''
-    with fitz.open(file_path) as pdf_document:
-        for page_number in range(pdf_document.page_count):
-            page = pdf_document.load_page(page_number)
-            text += page.get_text()
-    return text
-
 def predict_class(text):
     try:
         max_length = 4096
@@ -97,37 +89,4 @@ if option == "Text":
                                 unsafe_allow_html=True
                             )
                         else:
-                            st.text(label)
-
-elif option == "PDF":
-    uploaded_file = st.file_uploader("Upload a PDF file", type=["pdf"])
-
-    if uploaded_file is not None:
-        with st.spinner("Processing PDF..."):
-            file_path = os.path.join(uploaded_files_dir, uploaded_file.name)
-            with open(file_path, "wb") as f:
-                f.write(uploaded_file.getbuffer())
-            st.success("File uploaded successfully.")
-            st.text(f"File Path: {file_path}")
-            
-            file_text = extract_text_from_pdf(file_path)
-            st.text("Extracted Text:")
-            st.text(file_text)
-
-            if st.button("Predict from PDF Text"):
-                if not file_text.strip():
-                    st.warning("Please upload a PDF with text content.")
-                else:
-                    with st.spinner("Predicting..."):
-                        predicted_class = predict_class(file_text)
-                        if predicted_class is not None:
-                            class_labels = ["Level 1", "Level 2", "Level 3", "Level 4"]
-                            st.text("**Predicted Class:**")
-                            for i, label in enumerate(class_labels):
-                                if i == predicted_class:
-                                    st.markdown(
-                                        f'<div style="background-color: {class_colors[predicted_class]}; padding: 10px; border-radius: 5px; color: white; font-weight: bold;">{label}</div>',
-                                        unsafe_allow_html=True
-                                    )
-                                else:
-                                    st.text(label)
+                            st.text(label)