Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from gradio_client import Client as GrClient
 from gradio import routes
 from typing import List, Type
 
@@ -29,32 +28,70 @@ def get_types(cls_set: List[Type], component: str):
     return docset, types
 routes.get_types = get_types
 
-history = dict()
+user_data = dict()
+chat_history = []
 
 
-def predict(user_id, gender, age):
-    # 추천 해서 반환
-    movie = "추천 영화"
-    return f"{movie}"
+def register(id, pw):
+    if not id in user_data:
+        user_data[id] = pw
+        return "ok"
+    else:
+        return "fail"
+
+def login(id, pw):
+    if not id in user_data:
+        return "fail"
+    else:
+        if user_data[id] != pw:
+            return "fail"
+        else:
+            return "ok"
+
+def chat(name, text):
+    if not name in user_data:
+        return "no id"
+    else:
+        chat_history.append({"name": name, "text":text})
+    return "ok"
 
-def add(user_id, movie):
-    if not user_id in history:
-        history[user_id] = []
-    history[user_id].append(movie)
+def get_data():
+    return chat_history
+
+def clear_data():
+    chat_history = []
     return "ok"
 
 with gr.Blocks() as demo:
     count = 0
     aa = gr.Interface(
-      fn=predict,
-      inputs=["text", "text", "text"],
+      fn=chat,
+      inputs=["text", "text"],
       outputs="text",
-      description="추천",
+      description="chat",
     )
     bb = gr.Interface(
-      fn=add,
+      fn=login,
       inputs=["text", "text"],
       outputs="text",
-      description="시청기록 추가",
+      description="login",
+    )
+    cc = gr.Interface(
+      fn=register,
+      inputs=["text", "text"],
+      outputs="text",
+      description="register",
+    )
+    dd = gr.Interface(
+      fn=get_data,
+      inputs=[],
+      outputs="text",
+      description="get_data",
+    )
+    gg = gr.Interface(
+      fn=clear_data,
+      inputs=[],
+      outputs="text",
+      description="clear_data",
     )
-    demo.queue(max_size=32).launch(enable_queue=True)
+    demo.queue(max_size=32).launch()