Update app.py

app.py

@@ -7,16 +7,17 @@ import requests
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=os.getenv("HF_TOKEN"))
 #hf_client = InferenceClient("CohereForAI/aya-23-35B", token=os.getenv("HF_TOKEN"))
 
-def load_fashion_code():
+def load_code(filename):
     try:
-        with open('fashion.cod', 'r', encoding='utf-8') as file:
+        with open(filename, 'r', encoding='utf-8') as file:
             return file.read()
     except FileNotFoundError:
-        return "fashion.cod 파일을 찾을 수 없습니다."
+        return f"{filename} 파일을 찾을 수 없습니다."
     except Exception as e:
         return f"파일을 읽는 중 오류가 발생했습니다: {str(e)}"
 
-fashion_code = load_fashion_code()
+fashion_code = load_code('fashion.cod')
+uhdimage_code = load_code('uhdimage.cod')
 
 def respond(
     message,
@@ -26,13 +27,18 @@ def respond(
     temperature=0.7,  # 기본값 추가
     top_p=0.9,  # 기본값 추가
 ):
-    global fashion_code
+    global fashion_code, uhdimage_code
     system_prefix = """반드시 한글로 답변할것. 너는 주어진 소스코드를 기반으로 "서비스 사용 설명 및 안내, qna를 하는 역할이다". 아주 친절하고 자세하게 4000토큰 이상 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해 할 만 한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라. """
 
     if message.lower() == "패션 코드 실행":
         system_message = system_message or ""  # None인 경우 빈 문자열로 초기화
         system_message += f"\n\n패션 코드 내용:\n{fashion_code}"
-        message = "패션 코드에 대해 설명해주세요."
+        message = "패션 가상피팅에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL("https://aiqcamp-fash.hf.space")을 통해 테스트 해보라고 출력하라."
+    elif message.lower() == "uhd 이미지 코드 실행":
+        system_message = system_message or ""  # None인 경우 빈 문자열로 초기화
+        system_message += f"\n\nUHD 이미지 코드 내용:\n{uhdimage_code}"
+        message = "UHD 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL("https://openfree-ultpixgen.hf.space")을 통해 테스트 해보라고 출력하라."
+
 
     messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]  # prefix 추가
     for val in history:
@@ -66,6 +72,7 @@ demo = gr.ChatInterface(
     ],
     examples=[
         ["패션 코드 실행"],
+        ["UHD 이미지 코드 실행"],
         ["사용 방법을 4000 토큰 이상 자세히 설명하라"],
         ["사용 방법을 4000 토큰 이상 유튜브 영상 스크립트 형태로 작성하라"],
         ["사용 방법을 SEO 최적화하여 블로그 포스트로 4000 토큰 이상 작성하라"],