Update app.py

app.py

@@ -1,14 +1,14 @@
 from flask import Flask, request, jsonify
-import streamlit as st
+#import streamlit as st
 from langchain import PromptTemplate, LLMChain
 from langchain.memory import StreamlitChatMessageHistory
-from streamlit_chat import message
+#from streamlit_chat import message
 import numpy as np
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
-from streamlit.components.v1 import html
+#from streamlit.components.v1 import html
 from langchain import HuggingFaceHub
 import os
 from dotenv import load_dotenv
@@ -20,15 +20,15 @@ import requests
 #from time import sleep
 import uuid
 import sys
-from streamlit_extras.colored_header import colored_header
-from streamlit_extras.add_vertical_space import add_vertical_space
+#from streamlit_extras.colored_header import colored_header
+#from streamlit_extras.add_vertical_space import add_vertical_space
 
-st.set_page_config(page_title="AI Chatbot 100% Free", layout="wide")
-st.write('完全开源免费的AI智能聊天助手 | Absolute Free & Opensouce AI Chatbot')
+#st.set_page_config(page_title="AI Chatbot 100% Free", layout="wide")
+#st.write('完全开源免费的AI智能聊天助手 | Absolute Free & Opensouce AI Chatbot')
 
-css_file = "main.css"
-with open(css_file) as f:
-    st.markdown("<style>{}</style>".format(f.read()), unsafe_allow_html=True)
+#css_file = "main.css"
+#with open(css_file) as f:
+#    st.markdown("<style>{}</style>".format(f.read()), unsafe_allow_html=True)
 
 # 初始化Chatbot
 HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN')
@@ -57,7 +57,7 @@ assistant:
 
 llm_chain = LLMChain(llm=llm, prompt=PromptTemplate.from_template(prompt_template))
 
-temp_user_query = st.chat_input("Enter your question here.")
+#temp_user_query = st.chat_input("Enter your question here.")
 
 # 定义API端点
 app = Flask(__name__)
@@ -67,13 +67,14 @@ def chat():
     #user_query = data['query']
 #此处的['query']中的query可以自定义名称，例如修改为user_question，那么调用API的代码中，需要相应的使用data = {'user_question': user_query}，user_question需一致
     user_query = data['user_question']      
-    temp_user_query=user_query
+    #temp_user_query=user_query
     # 调用Chatbot
     initial_response = llm_chain.run(user_query)
-    st.write("AI Response")
-    st.write(initial_response)
+    #st.write("AI Response")
+    #st.write(initial_response)
     return jsonify({'response': initial_response})
 
 if __name__ == '__main__':    
     #app.run(host='0.0.0.0', port=port)
-    app.run(host='0.0.0.0')
+    #app.run(host='0.0.0.0')
+    uvicorn.run(app, host='0.0.0.0')