Dependencies fix

app.py

@@ -5,12 +5,13 @@ from typing import Dict, Any, List
 
 import gradio as gr
 import spacy
+import streamlit as st
 from pony.orm import db_session
 from spacy import displacy
 
-from app.classificator import Classificator, Classification
-from app.models.request import Request
-from settings import AUTH_ENDPOINT_TOKEN, ENDPOINT_URL, db
+from core.classificator import Classificator, Classification
+from core.models.request import Request
+from settings import db
 
 
 class Application:
@@ -80,12 +81,20 @@ class Application:
 
 
 if __name__ == '__main__':
+    db.bind(
+        provider='postgres',
+        user=st.secrets['pg_user'],
+        password=st.secrets['pg_password'],
+        host=st.secrets['pg_host'],
+        port=st.secrets['pg_port'],
+        database=st.secrets['pg_database']
+    )
     db.generate_mapping(create_tables=True)
     application = Application(
         classificator=Classificator(
             config={
-                'auth_endpoint_token': AUTH_ENDPOINT_TOKEN,
-                'endpoint_url': ENDPOINT_URL
+                'auth_endpoint_token': st.secrets["auth_endpoint_token"],
+                'endpoint_url': st.secrets["endpoint_url"]
             }
         ),
         options={"ents": ["Observation", "Evaluation"], "colors": {"Observation": "#9bddff", "Evaluation": "#f08080"}}

{app → core}/classificator.py

@@ -30,7 +30,7 @@ class Classificator:
         """
         self._config = config
 
-    def classify(self, text: str) -> List[Any]:
+    def classify(self, text: str) -> List[Classification]:
         raw_data = self.send_request(text)
         return self.post_process(raw_data)
 

settings.py

@@ -1,14 +1,3 @@
-import streamlit as st
 from pony.orm import Database
 
 db = Database()
-db.bind(
-    provider='postgres',
-    user=st.secrets['pg_user'],
-    password=st.secrets['pg_password'],
-    host=st.secrets['pg_host'],
-    port=st.secrets['pg_port'],
-    database=st.secrets['pg_database']
-)
-AUTH_ENDPOINT_TOKEN = st.secrets["auth_endpoint_token"]
-ENDPOINT_URL = st.secrets["endpoint_url"]