Update app.py

app.py

@@ -2,8 +2,7 @@ import streamlit as st
 from streamlit.components.v1 import html
 import os
 import PyPDF2
-
-st.write(os.environ("TEST"))
+import requests
 
 def get_pdf_text(pdf_path):
     # creating a pdf file object
@@ -24,6 +23,22 @@ def get_pdf_text(pdf_path):
 
     return pdf_text
 
+headers = {"Authorization": "Bearer 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"}
+
+def create_tags(payload):
+    API_URL_TAGS = "https://api-inference.huggingface.co/models/fabiochiu/t5-base-tag-generation"
+    
+    response = requests.post(API_URL_TAGS, headers=headers, json=payload)
+    return response.json()
+
+def summarize_text(payload):
+    API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
+
+# Start of the app code
+
 tab_general_topics, tab_your_paper = st.tabs(["Research topics", "Summarize your paper(s)"])
 
 with tab_general_topics:
@@ -46,7 +61,7 @@ with tab_your_paper:
         for pdf in pdf_files:
             # Saving the files
             pdf_data = pdf.getvalue()
-            pdf_path = os.path.join(pdf.name)
+            pdf_path = os.path.join("pdfs", pdf.name)
             with open(pdf_path, "wb") as f:
                 f.write(pdf_data)
                 recently_added.append(pdf_path)
@@ -55,6 +70,7 @@ with tab_your_paper:
         for recent_pdf in recently_added:
             # Reading the pdf files
             pdf_content = get_pdf_text(recent_pdf)
+            print("**", pdf_content)
             pdfs_content_list.append(pdf_content)
 
             # Delete the files
@@ -62,4 +78,33 @@ with tab_your_paper:
 
         all_text_together = " ".join(pdfs_content_list)
 
-        st.write(all_text_together)
+        try:
+            tags = create_tags({
+                "inputs": all_text_together,
+            })[0]["generated_text"]
+            tags_available = True
+        except:
+            tags_available = False
+        	
+        summary = summarize_text({
+            "inputs": all_text_together
+        })[0]["summary_text"]
+
+        col1, col2 = st.columns(2)
+        with col1:
+            if len(recently_added) > 1:
+                st.markdown("#### Summary of your paper(s):")
+            else:
+                st.markdown("#### Summary of your paper:")
+            st.write(summary)
+
+        if tags_available == True:
+            with col2:
+                if len(recently_added) > 1:
+                    st.markdown("#### Identified topics of your paper(s):")
+                else:
+                    st.markdown("#### Identified topics of your paper:")
+                st.write(tags)
+
+        with st.expander("See your total text"):
+            st.write(all_text_together)