Update app.py

app.py

@@ -1,66 +1,90 @@
 import gradio as gr
-
------------------ SIMPLE AUTH SYSTEM -----------------
-
-users = {"admin": "1234"}
-
-def login(username, password): if username in users and users[username] == password: return gr.update(visible=False), gr.update(visible=True), f"Welcome, {username}!" return gr.update(), gr.update(), "❌ Invalid credentials!"
-
-def signup(username, password): if username in users: return "❌ User already exists!" users[username] = password return "✅ Account created! You can now log in."
-
------------------ CHATBOT FUNCTION -----------------
-
-def chat_bot(msg, history): history = history + [(msg, "✨ Your bot reply will appear here...")] return history, history
-
------------------ CUSTOM CSS (Clean Modern UI) -----------------
-
-css = """ body { background: #0d0d0d; font-family: Inter, sans-serif; } #main_title { color: white; text-align: center; margin-top: 40px; font-size: 36px; font-weight: 900; }
-
-.card-btn { background: linear-gradient(145deg, #1a1a1a, #111); border-radius: 20px; padding: 18px; border: 1px solid #2a2a2a; color: white; font-size: 18px; width: 100%; transition: 0.25s; } .card-btn:hover { background: #222; transform: translateY(-3px); } .chatbox-holder { border-radius: 18px; overflow: hidden; } """
-
------------------ APP UI -----------------
-
-with gr.Blocks(css=css, theme="soft") as app:
-
-# -------- AUTH SCREEN --------
-with gr.Group(visible=True) as auth_screen:
-    gr.Markdown("<h1 id='main_title'>Welcome</h1>")
-
-    with gr.Tabs():
-        with gr.Tab("Login"):
-            login_user = gr.Textbox(label="Username")
-            login_pass = gr.Textbox(label="Password", type="password")
-            login_btn = gr.Button("Log In")
-            login_status = gr.Markdown("")
-
-        with gr.Tab("Sign Up"):
-            sign_user = gr.Textbox(label="New Username")
-            sign_pass = gr.Textbox(label="New Password", type="password")
-            sign_btn = gr.Button("Create Account")
-            signup_status = gr.Markdown("")
-
-# -------- MAIN HOME SCREEN --------
-with gr.Group(visible=False) as home_screen:
-
-    gr.Markdown("<h1 id='main_title'>What can I help you with?</h1>")
-
-    with gr.Row():
-        gr.Button("🎨 Create Image", elem_classes="card-btn")
-        gr.Button("💻 Write Code", elem_classes="card-btn")
-
-    with gr.Row():
-        gr.Button("📝 Summarize Text", elem_classes="card-btn")
-        gr.Button("📅 Make a Plan", elem_classes="card-btn")
-
-    chatbot = gr.Chatbot(height=450, elem_classes="chatbox-holder")
-    msg = gr.Textbox(placeholder="Ask something...", scale=8)
-    send = gr.Button("Send", scale=2)
-
-    send.click(chat_bot, [msg, chatbot], [chatbot, chatbot])
-    msg.submit(chat_bot, [msg, chatbot], [chatbot, chatbot])
-
-# -------- BUTTON EVENTS --------
-login_btn.click(login, [login_user, login_pass], [auth_screen, home_screen, login_status])
-sign_btn.click(signup, [sign_user, sign_pass], signup_status)
-
-app.launch()
+import google.generativeai as genai
+import os
+
+# ----------------- ТОХИРГОО -----------------
+# ЧУХАЛ: API түлхүүрээ орчны хувьсагч (environment variable) болгож тохируулна уу.
+# Google AI Studio-с API түлхүүрээ авна уу: https://aistudio.google.com/app/apikey
+#
+# Терминал дээрээ дараах байдлаар тохируулж болно:
+# export GOOGLE_API_KEY="ЭНД_API_ТҮЛХҮҮРЭЭ_ОРУУЛНА_УУ"
+
+try:
+    genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+except KeyError:
+    print("АЛДАА: GOOGLE_API_KEY орчны хувьсагч тохируулагдаагүй байна.")
+    # Хэрэв түлхүүр байхгүй бол Gradio интерфэйс дээр алдаа харуулна
+    api_key_set = False
+else:
+    api_key_set = True
+
+# Аюулгүй байдлын тохиргоог зөөлрүүлэх (зарим тохиолдолд хариулт блоклогдохоос сэргийлнэ)
+safety_settings = [
+    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
+    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
+    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
+    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
+]
+
+# Gemini Pro загварыг сонгох
+model = genai.GenerativeModel('gemini-pro', safety_settings=safety_settings)
+
+# Чатны түүхийг хадгалах (Gradio-ийн session state)
+chat = model.start_chat(history=[])
+
+# ----------------- GRADIO ФУНКЦ -----------------
+def gemini_chat(message, history):
+    """
+    Gradio чатны интерфэйст зориулсан функц.
+    Хэрэглэгчийн зурвасыг Gemini API руу илгээж, хариултыг нь буцаана.
+    """
+    if not api_key_set:
+        yield "АЛДАА: Google API түлхүүр тохируулагдаагүй байна. Програмыг унтрааж, API түлхүүрээ тохируулаад дахин ажиллуулна уу."
+        return
+
+    try:
+        # Mesasge-ийг загвар луу илгээнэ
+        response = chat.send_message(message, stream=True)
+
+        # Хариултыг хэсэгчлэн (stream) авах
+        full_response = ""
+        for chunk in response:
+            full_response += chunk.text
+            yield full_response
+
+    except Exception as e:
+        yield f"Алдаа гарлаа: {e}"
+
+
+# ----------------- GRADIO ИНТЕРФЕЙС -----------------
+with gr.Blocks(theme=gr.themes.Soft(), title="Gemini Chat") as demo:
+    gr.Markdown(
+        """
+    # Gemini Chat 🤖
+    Энэ бол Google-ийн Gemini Pro загварыг ашигласан чатбот юм.
+    """
+    )
+    # ChatGPT-тэй ижил дизайнтай чат интерфэйс
+    chatbot = gr.ChatInterface(
+        fn=gemini_chat,
+        chatbot=gr.Chatbot(
+            height=600,
+            show_label=False,
+            avatar_images=(None, "https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png"),
+        ),
+        textbox=gr.Textbox(
+            placeholder="Энд бичнэ үү...",
+            container=False,
+            scale=7,
+        ),
+        title=None,
+        submit_btn="Илгээх",
+        retry_btn=None,
+        undo_btn="Буцаах",
+        clear_btn="Цэвэрлэх",
+    )
+
+# ----------------- АЖИЛЛУУЛАХ -----------------
+if __name__ == "__main__":
+    # Gradio аппликейшнийг ажиллуулах
+    demo.launch(debug=True)