wassemgtk commited on
Commit
5629412
·
verified ·
1 Parent(s): daa6ac3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -14
app.py CHANGED
@@ -5,7 +5,7 @@ import json
5
 
6
  # These will be set as Hugging Face Spaces secrets
7
  API_KEY = os.environ.get("FIREWORKS_API_KEY", "")
8
- SYSTEM_PROMPT = os.environ.get("SYSTEM_PROMPT", "You are a helpful AI assistant.")
9
 
10
  # API endpoint
11
  API_URL = "https://api.fireworks.ai/inference/v1/chat/completions"
@@ -109,7 +109,7 @@ footer {
109
 
110
  # Create Gradio interface
111
  with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="purple", secondary_hue="pink")) as demo:
112
- gr.HTML("<h1 id='title'>✨ AI Playground</h1>")
113
  gr.HTML("<p id='subtitle'>Intelligent conversations powered by advanced language models</p>")
114
 
115
  with gr.Row():
@@ -189,15 +189,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="purple", second
189
  balanced_btn = gr.Button("⚖️ Balanced", size="sm")
190
  precise_btn = gr.Button("🎯 Precise", size="sm")
191
 
192
- gr.Markdown("---")
193
- gr.Markdown("### 🔧 Setup Guide")
194
- gr.Markdown("""
195
- **Space Settings → Secrets:**
196
- - `FIREWORKS_API_KEY`
197
- - `SYSTEM_PROMPT`
198
-
199
- Then restart your Space ♻️
200
- """)
201
 
202
  # Handle message submission with streaming
203
  def respond(message, chat_history, temp, max_tok, top_p_val, top_k_val):
@@ -260,9 +252,9 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="purple", second
260
  # Add examples
261
  gr.Examples(
262
  examples=[
263
- ["Write a creative short story about time travel"],
264
- ["Explain quantum computing in simple terms"],
265
- ["Help me brainstorm ideas for a mobile app"],
266
  ["What are the best practices for Python code?"],
267
  ],
268
  inputs=msg,
 
5
 
6
  # These will be set as Hugging Face Spaces secrets
7
  API_KEY = os.environ.get("FIREWORKS_API_KEY", "")
8
+ SYSTEM_PROMPT = os.environ.get("SYSTEM_PROMPT")
9
 
10
  # API endpoint
11
  API_URL = "https://api.fireworks.ai/inference/v1/chat/completions"
 
109
 
110
  # Create Gradio interface
111
  with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="purple", secondary_hue="pink")) as demo:
112
+ gr.HTML("<h1 id='title'>Palmyra-Sec Playground</h1>")
113
  gr.HTML("<p id='subtitle'>Intelligent conversations powered by advanced language models</p>")
114
 
115
  with gr.Row():
 
189
  balanced_btn = gr.Button("⚖️ Balanced", size="sm")
190
  precise_btn = gr.Button("🎯 Precise", size="sm")
191
 
192
+
 
 
 
 
 
 
 
 
193
 
194
  # Handle message submission with streaming
195
  def respond(message, chat_history, temp, max_tok, top_p_val, top_k_val):
 
252
  # Add examples
253
  gr.Examples(
254
  examples=[
255
+ ["Analyze the risks of CVE-2025-1234. Explain the root cause, attack vector, possible business impact, and give me defensive actions. Also map the mitigations to NIST and ISO 27001 controls."],
256
+ ["A financial services company detected unusual outbound traffic from an internal SQL server at 2am. Provide an executive summary of what might be happening, a technical analysis, likely MITRE ATT&CK techniques, and immediate incident response steps the SOC should take."],
257
+ ["We are preparing for a SOC-2 audit. Draft a short security policy on how our engineering team should handle secrets (API keys, SSH keys, environment variables) in source code and CI/CD pipelines. Include recommended tools and preventive controls."],
258
  ["What are the best practices for Python code?"],
259
  ],
260
  inputs=msg,