update

app.py

@@ -45,46 +45,23 @@ def generate_text(messages):
     else:
         return "Unexpected output format."
 
-def call_generate_text(prompt, system_message="You are a helpful assistant."):
-    if prompt =="":
-        print("empty prompt return")
-        return ""
-    
-    global histories
-        
-    messages = [
-        #{"role": "system", "content": system_message},
-    ]
-    
-    messages += histories 
-
-    user_message = {"role": "user", "content": prompt}
 
-    messages += [user_message]
+def call_generate_text(message, history):
+    print(message)
+    print(history)
+   
+    messages = history + message
     
     try:
         text = generate_text(messages)
-        histories += [user_message,{"role": "assistant", "content": text}]
-       
+        
         return text
     except RuntimeError  as e:
         print(f"An unexpected error occurred: {e}")
        
     return ""
 
-iface = gr.Interface(
-    fn=call_generate_text,
-    inputs=[
-        gr.Textbox(lines=3, label="Input Prompt"),
-        #gr.Textbox(lines=2, label="System Message", value="あなたは親切なアシスタントで常に日本語で返答します。"),
-    ],
-    outputs=gr.Textbox(label="Generated Text"),
-    title=f"{model_id}",
-    description=f"{model_id} jinja2.exceptions.TemplateError: System role not supported",
-)
-print("Initialized")
-
+demo = gr.ChatInterface(call_generate_text)
 
 if __name__ == "__main__":
-    print("Main")
-    iface.launch()
+    demo.launch()