Update app.py

app.py

@@ -1,4 +1,6 @@
-import streamlit as st
+import flask
+from flask import Flask, request, jsonify
+#import streamlit as st
 from langchain.chat_models import ChatOpenAI
 from langchain.document_loaders import WebBaseLoader
 from langchain.chains.summarize import load_summarize_chain
@@ -30,35 +32,58 @@ llm = HuggingFaceHub(repo_id=repo_id,  #for Llama2
 #llm = ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo-16k")
 chain = load_summarize_chain(llm, chain_type="stuff")
 
-#**************************************************************#
-#url=st.text_input("Enter webiste URL to summarize (format: https://www.usinoip.com):")
-#loader = WebBaseLoader("https://www.usinoip.com/")
-#loader = WebBaseLoader(url)
-#with st.spinner("AI Thinking...Please wait a while to Cheers!"):
-#    docs = loader.load()
-#    result=chain.run(docs)
-#    print(result)
-#    st.write("AI Summarization:")
-#    st.write(result)
-#**************************************************************#
+app = Flask(__name__)
+@app.route('/', methods=['POST'])
+def home_api():
+    data = request.get_json()
+    user_query = data['user_question']
+    print(user_query)
+    return {"Message":"Flask Home API Deploy Success on HF"}  
 
-#**************************************************************#
-url=st.text_input("Enter webiste URL to summarize (format: https://www.usinoip.com):")
-
-#loader = WebBaseLoader("https://www.usinoip.com/")
-
-if url !="" and not url.strip().isspace() and not url == "" and not url.strip() == "" and not url.isspace():
+@app.route('/api/chat', methods=['POST'])
+def chat():
+#async def chat():  #Failed - Flask API 不支持async???
+    data = request.get_json()
+    user_query = data['user_question']
+    if user_query !="" and not user_query.strip().isspace() and not user_query == "" and not user_query.strip() == "" and not user_query.isspace():
     try:
-        loader = WebBaseLoader(url)
+        loader = WebBaseLoader(user_query)
         with st.spinner("AI Thinking...Please wait a while to Cheers!"):
             docs = loader.load()
             result=chain.run(docs)
-            print(url)
+            print(user_query)
             print("AI Summarization: "+result)
-            st.write("AI Summarization:")
-            st.write(result)
+            #st.write("AI Summarization:")
+            #st.write(result)
+            return jsonify({'response': result})
     except Exception as e:
-        st.write("Wrong URL or URL not parsable.")    
+        #st.write("Wrong URL or URL not parsable.") 
+        err_msg="Wrong URL or URL not parsable."
+        return jsonify({'response': err_msg})
+        
+    #initial_response = llm_chain.run(user_query)
+    #return jsonify({'response': initial_response})
+    #找到问题了：jsonify在Huggingface不支持；在Github然后部署到Render是可以的！---NO No No, it's supported
+    #return {'response': initial_response}
+    #return jsonify({'response': initial_response})  #tried and OKed!
+    
+
+#url=st.text_input("Enter webiste URL to summarize (format: https://www.usinoip.com):")
+
+#loader = WebBaseLoader("https://www.usinoip.com/")
+
+#if url !="" and not url.strip().isspace() and not url == "" and not url.strip() == "" and not url.isspace():
+#    try:
+#        loader = WebBaseLoader(url)
+#        with st.spinner("AI Thinking...Please wait a while to Cheers!"):
+#            docs = loader.load()
+#            result=chain.run(docs)
+#            print(url)
+#            print("AI Summarization: "+result)
+#            st.write("AI Summarization:")
+#            st.write(result)
+#    except Exception as e:
+#        st.write("Wrong URL or URL not parsable.")    
 #**************************************************************#
 
 #try: