Create app.py

app.py

@@ -0,0 +1,61 @@
+import gradio as gr
+import torch
+from transformers import pipeline
+
+# 1. Initialize the pipeline
+print("Loading Dawn-V1-mini...")
+pipe = pipeline(
+    "text-generation", 
+    model="Dawn-AI/Dawn-V1-mini", 
+    device_map="auto",
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+)
+print("Model loaded successfully!")
+
+# 2. Updated chat function with 'system_prompt'
+def chat_with_dawn(message, history, system_prompt):
+    messages = []
+    
+    # Add the system prompt first (if the user provided one)
+    if system_prompt.strip():
+        messages.append({"role": "system", "content": system_prompt})
+    
+    # Add the conversation history
+    for user_msg, assistant_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": assistant_msg})
+        
+    # Add the current user message
+    messages.append({"role": "user", "content": message})
+    
+    # Generate the response
+    response = pipe(messages, max_new_tokens=512, truncation=True)
+    generated_text = response[0]['generated_text']
+    
+    # Extract just the new reply
+    if isinstance(generated_text, list):
+        return generated_text[-1]['content']
+    else:
+        return generated_text
+
+# 3. Create the System Prompt text box
+system_textbox = gr.Textbox(
+    value="You are Dawn V1, a brilliant, logical, and highly capable reasoning AI optimized for coding and complex problem-solving. Always think step-by-step.",
+    label="System Prompt",
+    lines=3,
+    interactive=True
+)
+
+# 4. Create the Chat Interface and pass the system textbox to 'additional_inputs'
+demo = gr.ChatInterface(
+    fn=chat_with_dawn,
+    additional_inputs=[system_textbox], # This links the textbox to the third argument of your function
+    title="🌅 Dawn V1 Mini Chat",
+    description="A lightweight conversational interface powered by Dawn-V1-mini. **Click 'Additional Inputs' below to change Dawn's personality!**",
+    examples=["Who are you?", "Write a Python script to reverse a string.", "Explain quantum computing simply."],
+    theme="soft",
+)
+
+# 5. Launch the app
+if __name__ == "__main__":
+    demo.launch()