Update app.py

app.py

@@ -4,12 +4,12 @@ from sentence_transformers import SentenceTransformer
 from google import genai
 import gradio as gr
 
-# 환경 설정
+# === 환경 설정 ===
 DB_DIR = os.getenv("CHROMA_DB_DIR", os.path.join(os.getcwd(), "chromadb_KH_media"))
 os.environ["CHROMA_DB_DIR"] = DB_DIR
-API_KEY = os.getenv("GOOGLE_API_KEY", "AIzaSyBuBsC5k9yw2JwUfVFn1Zu1qM_ifwGx6cM")
+API_KEY = os.getenv("GOOGLE_API_KEY", "YOUR_API_KEY_HERE")
 
-# RAG 시스템
+# === Simple RAG 시스템 ===
 class SimpleRAGSystem:
     def __init__(self, db_path=None, collection_name="KH_media_docs"):
         path = db_path or DB_DIR
@@ -22,38 +22,41 @@ class SimpleRAGSystem:
         if not self.available:
             return []
         emb = self.encoder.encode(query).tolist()
-        docs = self.collection.query(
-            query_embeddings=[emb], n_results=top_k,
+        result = self.collection.query(
+            query_embeddings=[emb],
+            n_results=top_k,
             include=["documents"]
         )
-        return docs["documents"][0] if docs.get("documents") else []
+        return result.get("documents", [[]])[0]
 
 rag = SimpleRAGSystem()
 
-# Google GenAI 클라이언트
+# === Google GenAI 클라이언트 ===
 client = genai.Client(api_key=API_KEY)
 
-# Gradio 응답 함수
-SYSTEM_MSG = f"""
+# === 시스템 메시지 ===
+SYSTEM_MSG = """
 당신은 경희대학교 미디어학과 전문 상담 AI입니다.
 """
 
+# === 응답 함수 ===
 def respond(message, history, system_message, max_tokens, temperature, top_p, model_name):
-    # RAG 컨텍스트
     docs = rag.search(message) if rag.available else []
     ctx = "\n".join(f"참고문서{i+1}: {d}" for i, d in enumerate(docs))
-    sys = system_message + ("\n# 참고문서:\n" + ctx if ctx else "")
-    # 대화 컨텍스트
+    sys_msg = system_message + ("\n# 참고문서:\n" + ctx if ctx else "")
     convo = "".join(f"사용자: {u}\nAI: {a}\n" for u, a in history)
-    prompt = f"{sys}\n{convo}사용자: {message}\nAI:"
-    # API 호출
+    prompt = f"{sys_msg}\n{convo}사용자: {message}\nAI:"
     try:
-        res = client.models.generate_content(
+        response = client.models.generate_content(
             model=model_name,
             contents=prompt,
-            config={"max_output_tokens": max_tokens, "temperature": temperature, "top_p": top_p}
+            config={
+                "max_output_tokens": max_tokens,
+                "temperature": temperature,
+                "top_p": top_p
+            }
         )
-        return res.text or "응답이 없습니다."
+        return response.text or "응답이 없습니다."
     except Exception as e:
         err = str(e).lower()
         if "quota" in err:
@@ -62,27 +65,29 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, mo
             return "인증 오류: API 키를 확인하세요."
         return f"오류 발생: {e}"
 
-# Gradio 인터페이스
+# === Gradio 인터페이스 ===
 demo = gr.ChatInterface(
     fn=respond,
     title="🎬 경희대학교 미디어학과 AI 상담사",
     description="경희대학교 미디어학과에 대해 물어보세요!",
     additional_inputs=[
-        gr.Slider(128, 2048, 1024, step=64, label="최대 토큰"),
-        gr.Slider(0.1, 1.0, 0.7, step=0.1, label="Temperature"),
-        gr.Slider(0.1, 1.0, 0.9, step=0.05, label="Top-p"),
-        gr.Dropdown([
-            "gemini-2.0-flash", "gemini-2.0-flash-lite",
-            "gemini-1.5-flash", "gemini-1.5-pro",
-            "gemma-3-27b-it", "gemma-3-12b-it", "gemma-3-4b-it"
-        ], value="gemini-2.0-flash", label="모델 선택")
+        gr.Textbox(value=SYSTEM_MSG, label="시스템 메시지", lines=2),
+        gr.Slider(128, 2048, value=1024, step=64, label="최대 토큰"),
+        gr.Slider(0.1, 1.0, value=0.7, step=0.1, label="Temperature"),
+        gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p"),
+        gr.Dropdown(
+            choices=[
+                "gemini-2.0-flash", "gemini-2.0-flash-lite",
+                "gemini-1.5-flash", "gemini-1.5-pro",
+                "gemma-3-27b-it", "gemma-3-12b-it", "gemma-3-4b-it"
+            ],
+            value="gemini-2.0-flash",
+            label="모델 선택"
+        )
     ],
     theme="soft",
     analytics_enabled=False,
 )
 
-def main():
-    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)), share=False)
-
 if __name__ == "__main__":
-    main()
+    demo.launch(share=False)