[Python] Flask & Streamlit API 연동

Flask

from flask import Flask, render_template

app = Flask(__name__)

@app.route('/test', methods=['GET'])
def test():
    return 'hello'

def main():
    app.run(host='0.0.0.0', debug=False, port=5001)

if __name__ == '__main__':
    main() 

• /test 로 GET 요청이 오면 hello 를 반환하는 예시 코드이다.

Streamlit

import streamlit as st
import requests

# GET 요청 버튼 
if st.button('Send GET Request'):
    # Flask 서버로 GET 요청 보내기
    response = requests.get('http://localhost:5001/test')
    
    if response.status_code == 200:
        st.success('Response received: ' + response.text)
    else:
        st.error('Failed to receive response')

• response.status_code 가 200 즉, 성공일 때 받은 text를 출력한다.