Create app.py

app.py

@@ -0,0 +1,107 @@
+import streamlit as st
+from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
+import pandas as pd
+import numpy as np
+
+# Load the fine-tuned model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained("slachitoff/CS-GY_6613_Milestone_3")
+model = AutoModelForSequenceClassification.from_pretrained("slachitoff/CS-GY_6613_Milestone_3")
+
+# Define the app layout
+st.set_page_config(page_title="Patentability Classifier App")
+st.title("Patentability Classifier")
+st.markdown("Enter the Patent ID, abstract and claims:")
+
+# Define SessionState to store patent scores and ideas within the session
+class SessionState:
+    def __init__(self):
+        self.patentData = {
+            "1234567890": {
+                "abstract": "A system and method for monitoring and predicting patient health is disclosed. The system utilizes a wearable device that collects data from various sensors, including heart rate, blood pressure, and oxygen saturation. The collected data is analyzed by a machine learning algorithm to predict health issues, such as heart attacks or strokes, before they occur.",
+                "claims": "A method for predicting patient health comprising: collecting data from a wearable device comprising a plurality of sensors that measure heart rate, blood pressure, and oxygen saturation; and analyzing said collected data using a machine learning algorithm to predict potential health issues."
+            },
+            "2345678901": {
+                "abstract": "A software application for real-time translation and transcription of audio recordings is disclosed. The application comprises a speech recognition module that transcribes audio recordings in one language and a machine translation module that translates the transcribed text into another language.",
+                "claims": "A software application for real-time translation and transcription of audio recordings comprising: a speech recognition module configured to transcribe audio recordings in a first language; and a machine translation module configured to translate said transcribed text into a second language."
+            },
+            "3456789012": {
+                "abstract": "A system and method for energy management in commercial buildings is disclosed. The system utilizes a combination of sensors, energy storage devices, and control algorithms to monitor and optimize energy consumption. The system is designed to reduce energy waste and improve the efficiency of HVAC and lighting systems in commercial buildings.",
+                "claims": "A system for energy management in commercial buildings comprising: a plurality of sensors configured to measure temperature, humidity, and occupancy levels; an energy storage device configured to store excess energy generated by HVAC and lighting systems; and a control algorithm configured to optimize energy consumption based on data from said sensors."
+            },
+            "4567890123": {
+                "abstract": "An electric vehicle with improved range and performance is disclosed. The vehicle comprises a regenerative braking system that converts kinetic energy into electrical energy during braking and a power management system that utilizes the stored energy during acceleration. The regenerative braking system improves the vehicle's overall efficiency and extends its range.",
+                "claims": "An electric vehicle with improved range and performance comprising: a regenerative braking system configured to convert kinetic energy into electrical energy during braking and store said electrical energy in a battery; and a power management system configured to provide said stored electrical energy to an electric motor during acceleration to improve the vehicle's performance and extend its range."
+            },
+            "5678901234": {
+                "abstract": "A device for preventing pets from accessing restricted areas is disclosed. The device comprises a motion sensor that detects when a pet enters a restricted area and emits a high-pitched sound to deter the pet from entering. The device is designed to be easily installed and can be used to keep pets away from certain rooms or pieces of furniture.",
+                "claims": "A pet deterrent device comprising: a motion sensor configured to detect when a pet enters a restricted area; and a speaker configured to emit a high-pitched sound to deter said pet from entering said restricted area."
+            },
+            "6789012345": {
+                "abstract": "A system for tracking the location of personal belongings is disclosed. The system utilizes a combination of RFID tags and a mobile application to track the location of personal items, such as keys or wallets. The mobile application can be used to view the location of the tracked items and receive notifications when they are moved or taken outside of a certain range.",
+                "claims": "A system for tracking personal belongings comprising: a plurality of RFID tags configured to be attached to personal items; a mobile application configured to communicate with said RFID tags and display the location of said personal items; and a notification system configured to notify a user when said personal items are moved or taken outside of a certain range."
+            },
+            "7890123456": {
+                "abstract": "A device for filtering and purifying air is disclosed. The device comprises a HEPA filter and an activated carbon filter that work together to remove pollutants and allergens from the air. The device is designed to be compact and portable, making it ideal for use in small spaces such as offices or bedrooms.",
+                "claims": "An air purifying device comprising: a HEPA filter configured to remove pollutants and allergens from the air; and an activated carbon filter configured to remove odors and other harmful substances from the air."
+            },
+            "8901234567": {
+                "abstract": "A device for making coffee is disclosed. The device comprises a container for holding water, a coffee filter for holding ground coffee, and a heating element for heating the water. The device is designed to be easy to use and can make a single cup of coffee at a time.",
+                "claims": "A coffee-making device comprising: a container for holding water; a coffee filter for holding ground coffee; and a heating element configured to heat said water and pass it through said coffee filter to make coffee."
+            },
+            "9012345678": {
+                "abstract": "A software application for tracking daily exercise is disclosed. The application allows users to input their daily exercise routines and track their progress over time. The application also includes social features that allow users to connect with friends and share their progress.",
+                "claims": "A software application for tracking daily exercise comprising: a user interface configured to allow users to input their daily exercise routines; a tracking system configured to record said exercise routines and display progress over time; and a social network feature configured to allow users to connect with friends and share their progress."
+            },
+            "0123456789": {
+                "abstract": "A device for holding books is disclosed. The device comprises a stand with adjustable arms that can hold books of various sizes. The device is designed to be portable and can be used in a variety of settings, such as in bed or on a table.",
+                "claims": "A book holder device comprising: a stand with adjustable arms configured to hold books of various sizes, and a base configured to provide stability to said stand."
+            }
+        }
+
+state = SessionState()
+
+# Define the function to calculate the patentability score
+@st.cache_data()
+def calculateScore(patent_id, abstract, claims):
+    inputs = tokenizer(abstract, claims, padding=True, truncation=True, max_length=512, return_tensors='pt')
+    outputs = model(**inputs)
+    logits = outputs.logits
+    scores = logits.softmax(dim=1).tolist()[0]
+    score, idx = np.max(scores), np.argmax(scores)
+    certainty = score*100
+    decision = 'Patentable' if idx==1 else 'Not Patentable'
+    return {'Abstract': abstract, 'Claims': claims, 'Decision': decision, 'Certainty': certainty}
+
+
+# Define the patent IDs
+patentIds = ["1234567890", "2345678901", "3456789012", "4567890123", "5678901234",
+              "6789012345", "7890123456", "8901234567", "9012345678", "0123456789"]
+
+# Define the dropdown menu for the patent ID
+selectedPatentId = st.selectbox("Select Patent ID", patentIds)
+
+# Autofill the Abstract and Claims text boxes based on the selected patent ID
+if selectedPatentId != "":
+    abstractText = state.patentData[selectedPatentId]["abstract"]
+    claimsText = state.patentData[selectedPatentId]["claims"]
+else:
+    abstractText = ""
+    claimsText = ""
+
+# Define the text area widgets for abstract and claims
+abstractText = st.text_area("Abstract", height=200, value=abstractText)
+claimsText = st.text_area("Claims", height=200, value=claimsText)
+
+# Define the submit button for patent ideas and scores
+if st.button("Submit"):
+    if not abstractText.strip() or not claimsText.strip():
+        st.write("Please enter the abstract and claims.")
+    else:
+        scoreData = calculateScore(selectedPatentId, abstractText, claimsText)
+        patentData = {'Abstract': abstractText, 'Claims': claimsText, 'Score': scoreData}
+        state.patentData[selectedPatentId] = patentData
+        st.write("Patent ID:", selectedPatentId)
+        st.write("Abstract:", abstractText)
+        st.write("Claims:", claimsText)
+        st.write("Patentability Score:", scoreData['Decision'])
+        st.write("Certainty (%):", scoreData['Certainty'])