Update app.py

app.py

@@ -1,36 +1,114 @@
+import os
 import gradio as gr
 from openai import OpenAI
+from dotenv import load_dotenv
 
-# Konfigurasi OpenRouter
-client = OpenAI(
-    base_url="https://openrouter.ai/api/v1",
-    api_key="your-openrouter-api-key",  # Ganti dengan API key OpenRouter Anda
-)
+# Memuat environment variables
+load_dotenv()
+
+class Chatbot:
+    def __init__(self):
+        # Dapatkan API key dari environment variables
+        self.api_key = os.getenv("OPENROUTER_API_KEY")
+        
+        if not self.api_key:
+            raise ValueError(
+                "OpenRouter API key not found. "
+                "Please set it in Hugging Face Secrets or .env file."
+            )
+        
+        self.client = OpenAI(
+            base_url="https://openrouter.ai/api/v1",
+            api_key=self.api_key,
+        )
+        
+        # Konfigurasi model
+        self.model = "deepseek/deepseek-r1-zero:free"
+        self.system_prompt = """
+        Anda adalah asisten AI yang membantu. Berikan jawaban yang:
+        - Singkat dan jelas
+        - Ramah dan informatif
+        - Relevan dengan pertanyaan
+        """
+
+    def generate_response(self, message, history):
+        """Generate AI response based on message and conversation history"""
+        try:
+            # Format conversation history
+            messages = [{"role": "system", "content": self.system_prompt}]
+            
+            for i, h in enumerate(history):
+                messages.append({
+                    "role": "user" if i % 2 == 0 else "assistant",
+                    "content": h
+                })
+            
+            messages.append({"role": "user", "content": message})
+            
+            # Kirim permintaan ke API
+            completion = self.client.chat.completions.create(
+                extra_headers={
+                    "HTTP-Referer": "https://huggingface.co/spaces",
+                    "X-Title": "DeepSeek Chatbot",
+                },
+                model=self.model,
+                messages=messages,
+                temperature=0.7,
+                max_tokens=500
+            )
+            
+            return completion.choices[0].message.content
+            
+        except Exception as e:
+            print(f"Error generating response: {str(e)}")
+            return f"Maaf, terjadi error: {str(e)}"
+
+# Inisialisasi chatbot
+try:
+    chatbot = Chatbot()
+except ValueError as e:
+    print(str(e))
+    chatbot = None
 
 def respond(message, history):
-    # Kirim pesan ke model DeepSeek R1 Zero
-    completion = client.chat.completions.create(
-        extra_headers={
-            "HTTP-Referer": "https://huggingface.co/spaces",  # Optional
-            "X-Title": "DeepSeek Chatbot",  # Optional
-        },
-        model="deepseek/deepseek-r1-zero:free",
-        messages=[
-            {"role": "system", "content": "Anda adalah asisten AI yang membantu. Berikan jawaban yang singkat dan jelas."},
-            *[{"role": "user" if i % 2 == 0 else "assistant", "content": h} for i, h in enumerate(history)],
-            {"role": "user", "content": message}
-        ]
-    )
+    if not chatbot:
+        return "Error: Chatbot tidak dapat diinisialisasi. Periksa konfigurasi API key."
     
-    return completion.choices[0].message.content
-
-# Buat antarmuka chatbot
-demo = gr.ChatInterface(
-    respond,
-    title="DeepSeek R1 Zero Chatbot",
-    description="Chatbot menggunakan model DeepSeek R1 Zero melalui OpenRouter",
-    theme="soft",
-    examples=["Apa itu AI?", "Buatkan puisi pendek", "Jelaskan teori relativitas dengan sederhana"]
-)
-
-demo.launch()
+    return chatbot.generate_response(message, history)
+
+# Contoh pertanyaan
+examples = [
+    "Apa itu kecerdasan buatan?",
+    "Buatkan puisi tentang teknologi",
+    "Jelaskan teori relativitas dengan sederhana",
+    "Apa kelebihan model DeepSeek?"
+]
+
+# Buat antarmuka Gradio
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""
+    # 🤖 DeepSeek R1 Zero Chatbot
+    Chatbot cerdas menggunakan model DeepSeek R1 Zero melalui OpenRouter API
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            gr.ChatInterface(
+                respond,
+                examples=examples,
+                retry_btn="Coba Lagi",
+                undo_btn="Undo",
+                clear_btn="Bersihkan"
+            )
+        with gr.Column():
+            gr.Markdown("""
+            ### Tentang Chatbot Ini
+            - Menggunakan model **DeepSeek R1 Zero**
+            - Diakses melalui **OpenRouter API**
+            - Dibangun dengan **Gradio** di Hugging Face Spaces
+            
+            **Catatan**: Chatbot ini hanya untuk demonstrasi.
+            """)
+
+if __name__ == "__main__":
+    demo.launch()