ko commited on
Commit
1c3b6c3
·
verified ·
1 Parent(s): 335b9a8

Upload 3 files

Browse files
Files changed (3) hide show
  1. app.py +56 -0
  2. kosen.txt +40 -0
  3. requirements.txt +2 -0
app.py ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from google import genai
2
+ import gradio as gr
3
+ import os
4
+
5
+ API_KEY = os.getenv("GEMINI_API_KEY")
6
+
7
+ client = genai.Client(
8
+ api_key=API_KEY
9
+ )
10
+
11
+ # RAG用知識ベース
12
+ with open("kosen.txt", encoding="utf-8") as f:
13
+ knowledge = f.read()
14
+
15
+
16
+ def rag_chat(message, history):
17
+ # 過去の会話を文字列化
18
+ conversation = ""
19
+ for msg in history:
20
+ role = msg["role"]
21
+ content = msg["content"]
22
+ conversation += f"{role}: {content}\n"
23
+ # Geminiへ渡すプロンプト
24
+ prompt = f"""
25
+ あなたは富山高等専門学校の案内AIです。
26
+ 以下の資料を参考に回答してください。
27
+ ただし、資料にない内容に関しては、「参考資料に情報がないため、geminiによる回答を行います。
28
+ ただし、不正確である場合がありますので予めご了承ください。」と伝えた上で、geminiによる回答を行ってください。
29
+ =====資料=====
30
+ {knowledge}
31
+ ============
32
+ =====過去の会話=====
33
+ {conversation}
34
+ ===================
35
+ ユーザー:
36
+ {message}
37
+ """
38
+ response = client.models.generate_content(
39
+ model="gemini-2.5-flash",
40
+ contents=prompt
41
+ )
42
+ return response.text
43
+ app = gr.Interface(
44
+ fn=rag_chat,
45
+ inputs=gr.Textbox(
46
+ label="質問",
47
+ lines=3
48
+ ),
49
+ outputs=gr.Textbox(
50
+ label="回答",
51
+ lines=10
52
+ ),
53
+ title="富山高専RAGアプリ"
54
+ )
55
+
56
+ app.launch()
kosen.txt ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 富山高等専門学校
2
+
3
+ 【概要】
4
+ 富山高等専門学校は富山県にある国立高等専門学校である。
5
+ 本郷キャンパスと射水キャンパスの2つのキャンパスを有する。
6
+
7
+ 【教育理念】
8
+ 実践的技術者の育成を目指し、
9
+ 創造力と国際性を備えたエンジニアを育成する。
10
+
11
+ 【アクセス】
12
+ 本郷キャンパス:
13
+ 富山県富山市本郷町13番地
14
+
15
+ 射水キャンパス:
16
+ 富山県射水市海老江練合1番2
17
+
18
+ 【学科】
19
+ 機械システム工学科
20
+ 電気制御システム工学科
21
+ 物質化学工学科
22
+ 電子情報工学科
23
+ 商船学科
24
+
25
+ 【入試情報】
26
+ 推薦選抜と学力選抜が実施される。
27
+ 学力選抜では国語、数学、理科、社会、英語の試験を行う。
28
+
29
+ 【進路】
30
+ 本科卒業後は就職または専攻科進学が可能である。
31
+ また大学3年次への編入学実績も多数ある。
32
+
33
+ 【特徴】
34
+ 実験・実習を重視した教育を行っている。
35
+ 企業との共同研究やインターンシップも盛んである。
36
+
37
+ 【5年の担任】
38
+ 電気制御システム工学科:北村拓也
39
+ 機械システム工学科:山本 久嗣
40
+ 物質化学工学科:福田 知博
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ gradio
2
+ google-genai