add api

api.py

@@ -0,0 +1,22 @@
+import uvicorn
+import argparse
+import random
+from fastapi import FastAPI, Form
+from fastapi.responses import Response
+
+parser = argparse.ArgumentParser(description="Chat API")
+parser.add_argument("--port", type=int, default=10024, help="Port")
+args = parser.parse_args()
+
+app = FastAPI()
+
+
+@app.post("/chat")
+async def chat_endpoint(text: str = Form(...)):
+    print("Input text:", text)
+    response = random.choice(["hello?", "hello world!"])
+    return Response(content=response)
+
+
+if __name__ == '__main__':
+    uvicorn.run(app, host="0.0.0.0", port=args.port)

app.py

@@ -1,16 +1,25 @@
 import os
 import gradio as gr
 import json
-# api_key = os.environ.get('OPENAI_KEY', None)
-from apidemo.TranslateDemo import createRequest
+import requests
+
+url = os.environ.get("URL", None)
+# url = "http://localhost:10024/chat"
+# from apidemo.TranslateDemo import createRequest
+
 
 def get_responce(msg):
-    
-    result_str = createRequest(msg)
-    result = json.loads(result_str)
-    return  result["translation"][0]
+    # result_str = createRequest(msg)
+    # result = json.loads(result_str)
+    # return result["translation"][0]
     # pass
 
+    data = {"text": msg}
+    response = requests.post(url, data=data)
+    print(response.content.decode('utf-8'))
+    return response.content.decode('utf-8')
+
+
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
@@ -26,10 +35,3 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     demo.launch()
-
-
-
-
-
-
-    

requirements.txt

@@ -1 +1 @@
-websocket
+websocket