Update app.py

app.py

@@ -33,18 +33,46 @@ def get_types(cls_set: List[Type], component: str):
     return docset, types
 routes.get_types = get_types
 
-hf_hub_download(repo_id='StarFox7/Llama-2-ko-7B-chat-gguf', filename='Llama-2-ko-7B-chat-gguf-q4_0.bin', local_dir='./')
-
-llm = Llama(model_path = 'Llama-2-ko-7B-chat-gguf-q4_0.bin',
-            n_ctx=512,
-      )
 # App code
+
+account_list = dict()
+
+account_list['id'] = "pass"
+
+name_list = dict()
+name_list['id'] = 'name'
+
 def chat(x):    
-    #prom = f"다음은 A와 B의 역할극이야. 너는 B야. A와 대화하고 있어. 친구에게 친근하고 간결하게 잘 대답해줘.\n\n### A:\n{x}\n\n### B:\n"
-    #output = llm(prom, max_tokens=20, stop=["###"], echo=True)
-    #return output['choices'][0]['text'][len(prom):-1]
-    return "AI 응답입니다."
     
+    return "AI 응답입니다."
+
+
+def register(id, pass):
+    if id in account_list:
+        return "exist"
+    else:
+        account_list[id] = pass
+        return "ok"
+
+def login(id, pass):
+    if id in account_list:
+        if account_list[id] == pass:
+            return "login"
+        else:
+            return "password error"
+    else:
+        return "no id"
+
+def add_name(id, name):
+    name_list[id] = name
+    return "ok"
+
+def get_name(id):
+    if id in name_list:
+        return name_list['id']
+    else:
+        return "no id"
+
 with gr.Blocks() as demo:
     count = 0
     aa = gr.Interface(
@@ -54,5 +82,32 @@ with gr.Blocks() as demo:
       description="call",
     )
 
+    rr = gr.Interface(
+      fn=register,
+      inputs=["text", "text"],
+      outputs="text",
+      description="call",
+    )
+
+    ll = gr.Interface(
+      fn=login,
+      inputs=["text", "text"],
+      outputs="text",
+      description="call",
+    )
 
+    ad = gr.Interface(
+      fn=add_name,
+      inputs=["text", "text"],
+      outputs="text",
+      description="call",
+    )
+    
+    nn = gr.Interface(
+      fn=get_name,
+      inputs=["text"],
+      outputs="text",
+      description="call",
+    )
+    
     demo.queue(max_size=32).launch(enable_queue=True)