Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - Jekyll Master AI Demo
+# app.py - Jekyll Master AI Demo (FIXED)
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -29,14 +29,17 @@ def load_model():
         # Load model dengan quantization untuk hemat memory
         model = AutoModelForCausalLM.from_pretrained(
             MODEL_ID,
-            torch_dtype=torch.float16,
-            device_map="auto",
+            torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+            device_map="auto" if torch.cuda.is_available() else None,
             trust_remote_code=True,
             low_cpu_mem_usage=True
         )
         
         print("✅ Model loaded successfully!")
-        print(f"📱 Device: {model.device}")
+        if torch.cuda.is_available():
+            print(f"📱 Device: GPU ({torch.cuda.get_device_name(0)})")
+        else:
+            print(f"📱 Device: CPU")
         
     except Exception as e:
         print(f"❌ Error loading model: {e}")
@@ -59,7 +62,7 @@ def generate_jekyll_code(instruction, max_tokens=500, temperature=0.7):
         
         # Jika model tidak loaded, beri contoh
         if model is None or tokenizer is None:
-            return f"""# Jekyll Master AI - Example Output
+            example_output = """# Jekyll Master AI - Example Output
 
 # Model is loading or in fallback mode
 # Here's an example _config.yml for a tech blog:
@@ -81,4 +84,203 @@ plugins:
   - jekyll-feed
   - jekyll-seo-tag
 
-# Try the live demo when model is fully loaded
+# Try the live demo when model is fully loaded!"""
+            return example_output
+        
+        # Format prompt sederhana
+        prompt = f"Generate Jekyll code for: {instruction}\n\nCode:"
+        
+        # Tokenize
+        inputs = tokenizer(
+            prompt,
+            return_tensors="pt",
+            truncation=True,
+            max_length=512
+        )
+        
+        if torch.cuda.is_available():
+            inputs = inputs.to("cuda")
+        
+        # Generate
+        with torch.no_grad():
+            outputs = model.generate(
+                **inputs,
+                max_new_tokens=max_tokens,
+                temperature=temperature,
+                do_sample=True,
+                top_p=0.9,
+                repetition_penalty=1.1,
+                pad_token_id=tokenizer.pad_token_id if tokenizer.pad_token_id else tokenizer.eos_token_id,
+                eos_token_id=tokenizer.eos_token_id
+            )
+        
+        # Decode
+        generated = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        
+        # Extract code after prompt
+        if prompt in generated:
+            generated = generated.split(prompt)[-1].strip()
+        
+        # Clean up
+        generated = generated.replace(prompt, "").strip()
+        
+        # Jika output kosong
+        if not generated or len(generated) < 20:
+            generated = """# Generated Jekyll Code
+
+# Example _config.yml structure:
+
+title: "Your Site Title"
+description: "Your site description"
+baseurl: ""
+url: "https://yoursite.com"
+theme: minima
+
+# For better results, be more specific in your request!"""
+        
+        print(f"📤 Generated {len(generated)} characters")
+        return generated
+        
+    except Exception as e:
+        error_msg = f"Error: {str(e)}"
+        print(f"❌ {error_msg}")
+        return f"# Error\n{error_msg}\n\nPlease try again or simplify your request."
+
+# ================= GRADIO INTERFACE =================
+def create_interface():
+    """Create simple Gradio interface"""
+    
+    # Contoh instruksi
+    examples = [
+        ["Buat file _config.yml untuk blog teknologi"],
+        ["Buat layout post dengan featured image"],
+        ["Buat include untuk navigation bar"],
+        ["Buat plugin untuk reading time"],
+        ["Buat file Sass untuk buttons"],
+    ]
+    
+    # CSS sederhana
+    css = """
+    .gradio-container { max-width: 1200px; margin: 0 auto; }
+    .header { text-align: center; padding: 20px; }
+    .example-btn { margin: 5px; }
+    """
+    
+    with gr.Blocks(css=css, title="Jekyll Master AI") as demo:
+        
+        # Header
+        gr.Markdown("""
+        # 🎯 Jekyll Master AI
+        **Fine-tuned AI untuk Jekyll Static Site Generator**
+        
+        Generate clean, production-ready code untuk website Jekyll Anda.
+        """)
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                # Input
+                instruction = gr.Textbox(
+                    label="Instruksi",
+                    placeholder="Contoh: Buat file _config.yml untuk blog teknologi...",
+                    lines=4
+                )
+                
+                # Parameters
+                with gr.Accordion("⚙️ Parameters", open=False):
+                    max_tokens = gr.Slider(
+                        minimum=100,
+                        maximum=1000,
+                        value=500,
+                        step=50,
+                        label="Max Tokens"
+                    )
+                    temperature = gr.Slider(
+                        minimum=0.1,
+                        maximum=1.5,
+                        value=0.7,
+                        step=0.1,
+                        label="Temperature"
+                    )
+                
+                # Buttons
+                with gr.Row():
+                    generate_btn = gr.Button("🚀 Generate Code", variant="primary")
+                    clear_btn = gr.Button("🗑️ Clear", variant="secondary")
+            
+            with gr.Column(scale=2):
+                # Output
+                output = gr.Code(
+                    label="Generated Jekyll Code",
+                    language="yaml",
+                    lines=20
+                )
+        
+        # Quick examples
+        gr.Markdown("### 🚀 Contoh Cepat")
+        with gr.Row():
+            for text, example in [
+                ("📝 Config", "Buat _config.yml untuk blog"),
+                ("🎨 Layout", "Buat layout post"),
+                ("🔧 Plugin", "Buat plugin Jekyll"),
+            ]:
+                gr.Button(
+                    text,
+                    size="sm",
+                    variant="secondary"
+                ).click(
+                    lambda ex=example: ex,
+                    outputs=instruction
+                )
+        
+        # Full examples
+        gr.Examples(
+            examples=examples,
+            inputs=instruction,
+            outputs=output,
+            fn=generate_jekyll_code,
+            cache_examples=False,
+            label="Klik contoh untuk mencoba:"
+        )
+        
+        # Footer
+        gr.Markdown("---")
+        gr.Markdown(f"""
+        **Model:** [{MODEL_ID}](https://huggingface.co/{MODEL_ID}) | 
+        **Dataset:** [Jekyll Master Dataset](https://huggingface.co/datasets/daffaaditya/jekyll-master-dataset) |
+        **Built with:** Transformers + Gradio
+        """)
+        
+        # ============ EVENT HANDLERS ============
+        
+        # Generate button
+        generate_btn.click(
+            fn=generate_jekyll_code,
+            inputs=[instruction, max_tokens, temperature],
+            outputs=output
+        )
+        
+        # Clear button
+        clear_btn.click(
+            fn=lambda: ("", ""),
+            inputs=[],
+            outputs=[instruction, output]
+        )
+        
+        # Enter key submit
+        instruction.submit(
+            fn=generate_jekyll_code,
+            inputs=[instruction, max_tokens, temperature],
+            outputs=output
+        )
+    
+    return demo
+
+# ================= LAUNCH =================
+if __name__ == "__main__":
+    print("🚀 Launching Jekyll Master AI Demo...")
+    demo = create_interface()
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )