Spaces:
Sleeping
Sleeping
File size: 2,835 Bytes
661d930 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | import os
import gradio as gr
from google import genai
# Load Gemini API key (set this in your Hugging Face Space Secrets)
API_KEY = os.environ.get("GEMINI_API_KEY")
if not API_KEY:
raise RuntimeError("Missing GEMINI_API_KEY. Set it in Space Secrets.")
client = genai.Client(api_key=API_KEY)
MODEL = os.environ.get("GEMINI_MODEL", "gemini-2.5-flash")
def call_gemini(prompt: str) -> str:
"""Call Gemini and return model response text."""
try:
response = client.models.generate_content(model=MODEL, contents=prompt)
return getattr(response, "text", str(response))
except Exception as e:
return f"[Error calling Gemini API: {e}]"
def chat_turn(user_message: str, history: list):
"""Handle a single user chat turn (Gradio messages format)."""
if history is None:
history = []
user_message = user_message.strip()
if not user_message:
return history, history
history.append({"role": "user", "content": user_message})
reply = call_gemini(user_message)
history.append({"role": "assistant", "content": reply})
return history, history
def clear_chat():
return [], []
with gr.Blocks(
title="Muhammad Haris Chatbot",
css="""
body {background: linear-gradient(135deg, #e0f7fa, #e1bee7);}
#chatbot .user {
background-color: #0288d1 !important;
color: white !important;
border-radius: 16px !important;
padding: 10px !important;
}
#chatbot .assistant {
background-color: #ffffff !important;
border: 1px solid #ccc !important;
border-radius: 16px !important;
padding: 10px !important;
}
#title {
text-align: center;
font-size: 2em;
font-weight: bold;
color: #4a148c;
margin-bottom: 20px;
text-shadow: 1px 1px 2px #aaa;
}
#footer {
text-align: center;
font-size: 0.9em;
color: #555;
margin-top: 15px;
}
""",
) as demo:
gr.HTML('<div id="title">🤖 Muhammad Haris Chatbot</div>')
chatbot = gr.Chatbot(label="", type="messages", elem_id="chatbot")
state = gr.State([])
with gr.Row():
txt = gr.Textbox(
show_label=False,
placeholder="Type your message and press Enter...",
lines=1,
scale=9,
)
send = gr.Button("Send", scale=1)
txt.submit(fn=chat_turn, inputs=[txt, state], outputs=[chatbot, state])
send.click(fn=chat_turn, inputs=[txt, state], outputs=[chatbot, state])
gr.Button("Clear").click(fn=clear_chat, inputs=None, outputs=[chatbot, state])
gr.HTML('<div id="footer">Powered by Google Gemini · Built by Muhammad Haris</div>')
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
|