Update app.py

app.py

@@ -3,22 +3,28 @@ import streamlit as st
 import json
 
 def main():
-
     st.title("FastAPI - Streamlit Integration")
 
+    # Получаем базовый URL из streamlit
+    base_url = st.experimental_get_query_params()['base_url'][0] if 'base_url' in st.experimental_get_query_params() else None
+
     image = st.file_uploader("Choose an image", type=['jpg', 'jpeg'])
 
     if st.button("Classify!"):
         if image is not None:
             st.image(image)
             files = {"file": image.getvalue()}
-            res = requests.post("/classify", files=files)
+            # Используем базовый URL для формирования полного URL
+            url = f"{base_url}/classify"
+            res = requests.post(url, files=files)
             st.write(json.loads(res.text)['prediction'])
 
     # Эндпоинт для классификации текста
     text_input = st.text_input("Enter text for classification:")
     if st.button("Classify Text"):
-        response = requests.post("/clf_text", json={"text": text_input})
+        # Используем базовый URL для формирования полного URL
+        url = f"{base_url}/clf_text"
+        response = requests.post(url, json={"text": text_input})
         result = response.json()
         st.success(f"Classification result: {result['prediction']}")