Update app.py

app.py

@@ -1,11 +1,20 @@
-import streamlit as st
 import requests
+import streamlit as st
 import json
 
 def main():
 
     st.title("FastAPI - Streamlit Integration")
 
+    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("http://127.0.0.1:8000/classify", files=files)
+            st.write(json.loads(res.text)['prediction'])
+
     # Эндпоинт для классификации текста
     text_input = st.text_input("Enter text for classification:")
     if st.button("Classify Text"):
@@ -13,14 +22,5 @@ def main():
         result = response.json()
         st.success(f"Classification result: {result['prediction']}")
 
-    # Эндпоинт для классификации изображений
-    image_path = st.file_uploader("Upload an image for classification:", type=["jpg", "png"])
-    if image_path is not None:
-        if st.button("Classify Image"):
-            files = {"file": image_path.read()}
-            response = requests.post("http://127.0.0.1:8000/classify", files=files)
-            result = response.json()
-            st.success(f"Classification result: {result['prediction']}")
-
 if __name__ == '__main__':
     main()