Update app.py

app.py

@@ -75,6 +75,7 @@ const feedbackLink = `https://promptspellsmith.com/feedback`
 // Think logically and methodically, asking probing questions to encourage the user to articulate their thought process and reasoning. This approach not only helps 
 """
 
+
 def format_prompt(message, history):
     prompt = "<s>[SYSTEM] {} [/SYSTEM]".format(system_instruction)
     for user_prompt, bot_response in history:
@@ -83,11 +84,14 @@ def format_prompt(message, history):
     return prompt
 
 def generate(prompt, history, temperature=0.1, max_new_tokens=25000, top_p=0.95, repetition_penalty=1.0):
-    # 토큰 수 계산
     input_tokens = len(tokenizer.encode(prompt))
-    available_tokens = 32768 - input_tokens  # 최대 허용 토큰 수에서 입력 토큰 수를 뺀 값
-    max_new_tokens = min(max_new_tokens, available_tokens)  # 사용 가능한 최대 토큰 수 조정
+    available_tokens = 32768 - input_tokens
+    max_new_tokens = min(max_new_tokens, available_tokens)
     
+    if available_tokens <= 0:
+        yield "Error: 입력이 최대 허용 토큰 수를 초과합니다."
+        return
+
     formatted_prompt = format_prompt(prompt, history)
     
     try:
@@ -102,7 +106,7 @@ def generate(prompt, history, temperature=0.1, max_new_tokens=25000, top_p=0.95,
 
 mychatbot = gr.Chatbot(
     avatar_images=["./user.png", "./botm.png"],
-    bubble_full_width=False,
+    bubble_full_width=True,
     show_label=False,
     show_copy_button=True,
     likeable=True,
@@ -118,10 +122,11 @@ examples = [
 
 demo = gr.Interface(
     fn=generate,
-    inputs=gr.Textbox(placeholder="여기에 질문을 입력하세요...", lines=2),
+    inputs=[gr.Textbox(label="질문을 입력하세요", placeholder="여기에 질문을 입력하세요...", lines=2), gr.JSON(label="History", value=[])],
     outputs=gr.Markdown(),
     examples=examples,
-    title="AIQ 코드파일럿: OpenLLM v1.12"
+    title="AIQ 코드파일럿: OpenLLM v1.12",
+    description="AIQ Codepilot과 상호작용해 보세요."
 )
 
 demo.launch(show_api=False)