Using langchain

app.py

@@ -1,9 +1,17 @@
+from langchain.chains import SimpleChain
+from langchain.llms import OpenAI
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!"
+# 創建一個簡單的 LangChain
+llm = OpenAI(api_key="YOUR_OPENAI_API_KEY")
+chain = SimpleChain(llm=llm, input_key="input", output_key="response")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+def langchain_chat(input_text):
+    response = chain.run({"input": input_text})
+    return response['response']
+
+# 使用 Gradio 創建界面
+demo = gr.Interface(fn=langchain_chat, inputs="text", outputs="text")
 
 if __name__ == "__main__":
     demo.launch()

requirements.txt

@@ -5,3 +5,7 @@ torch
 transformers
 xformers
 gradio
+
+openai
+langchain
+gradio