Xyro123 commited on
Commit
56db801
·
verified ·
1 Parent(s): 92c2fa2

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -72
app.py DELETED
@@ -1,72 +0,0 @@
1
- import gradio as gr
2
- import google.generativeai as genai
3
- import os
4
-
5
- # ----------------- ТОХИРГОО -----------------
6
- try:
7
- genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
8
- except KeyError:
9
- print("АЛДАА: GOOGLE_API_KEY орчны хувьсагч тохируулагдаагүй байна.")
10
- api_key_set = False
11
- else:
12
- api_key_set = True
13
-
14
- # Аюулгүй байдлын тохиргоо
15
- safety_settings = [
16
- {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
17
- {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
18
- {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
19
- {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
20
- ]
21
-
22
- # Gemini Pro загвар сонгох
23
- model = genai.GenerativeModel('gemini-pro', safety_settings=safety_settings)
24
-
25
- # Чатны түүхийг хадгалах
26
- chat = model.start_chat(history=[])
27
-
28
- # ----------------- GRADIO ФУНКЦ -----------------
29
- def gemini_chat(message, history):
30
- """
31
- Хоёр output-тэй байх: textbox (хоосон болгох), chatbot (түүх)
32
- """
33
- if not api_key_set:
34
- return "", history + [{"role": "system", "content": "АЛДАА: Google API түлхүүр тохируулагдаагүй байна."}]
35
-
36
- try:
37
- response = chat.send_message(message, stream=True)
38
- full_response = ""
39
- for chunk in response:
40
- full_response += chunk.text
41
- # Stream хийх үед tuple хэлбэрээр хоёр output буцаана
42
- yield "", history + [{"role": "user", "content": message}, {"role": "assistant", "content": full_response}]
43
- except Exception as e:
44
- return "", history + [{"role": "system", "content": f"Алдаа гарлаа: {e}"}]
45
-
46
- # ----------------- GRADIO ИНТЕРФЕЙС -----------------
47
- with gr.Blocks(theme=gr.themes.Soft(), title="Gemini Chat") as demo:
48
- gr.Markdown(
49
- """
50
- # Gemini Chat 🤖
51
- Энэ бол Google-ийн Gemini Pro загварыг ашигласан чатбот юм.
52
- """
53
- )
54
-
55
- chatbot = gr.Chatbot(height=600, show_label=False,
56
- avatar_images=(None, "https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png"))
57
-
58
- msg = gr.Textbox(
59
- placeholder="Энд бичнэ үү...",
60
- container=False,
61
- scale=7,
62
- )
63
-
64
- clear_btn = gr.Button("Цэвэрлэх")
65
-
66
- # Функцуудийг холбох
67
- msg.submit(gemini_chat, [msg, chatbot], [msg, chatbot])
68
- clear_btn.click(lambda: None, None, chatbot, queue=False)
69
-
70
- # ----------------- АЖИЛЛУУЛАХ -----------------
71
- if __name__ == "__main__":
72
- demo.launch(server_name="0.0.0.0", server_port=7860, debug=True)