Update app.py

app.py

@@ -17,7 +17,7 @@ def encode_image(image):
     with open(image_path, "rb") as image_file:
         return base64.b64encode(image_file.read()).decode('utf-8')
 
-def bot_streaming(message, history, api_key, model, temperature, max_tokens, top_p, top_k, frequency_penalty, presence_penalty, repetition_penalty, stop, min_p, top_a, seed, logit_bias, logprobs, top_logprobs, response_format, tools, tool_choice):
+def bot_streaming(message, history, api_key, model, system_prompt, temperature, max_tokens, top_p, top_k, frequency_penalty, presence_penalty, repetition_penalty, stop, min_p, top_a, seed, logit_bias, logprobs, top_logprobs, response_format, tools, tool_choice):
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
@@ -26,6 +26,9 @@ def bot_streaming(message, history, api_key, model, temperature, max_tokens, top
     messages = []
     images = []
 
+    if system_prompt:
+        messages.append({"role": "system", "content": system_prompt})
+
     for i, msg in enumerate(history):
         if isinstance(msg[0], tuple):
             image, text = msg[0]
@@ -116,12 +119,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     2. Choose a model or enter a custom model name/endpoint
     3. Start chatting!
 
-    ## 🔧 Settings:
-    - Adjust basic parameters in the "Common Settings" section
-    - Fine-tune options in the "Advanced Settings" section
-    - Access expert-level controls in the "Expert Settings" section
-    - Upload images for multimodal interactions
-
     Enjoy your AI-powered conversation!
     """)
 
@@ -150,6 +147,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 value="google/gemini-flash-1.5",
                 allow_custom_value=True
             )
+            system_prompt = gr.Textbox(label="System Prompt", placeholder="Enter a system prompt (optional)")
             
             with gr.Accordion("Common Settings", open=False):
                 temperature = gr.Slider(minimum=0, maximum=2, value=1, step=0.1, label="Temperature")
@@ -178,7 +176,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             chatbot = gr.ChatInterface(
                 fn=bot_streaming,
                 additional_inputs=[
-                    api_key, model, temperature, max_tokens, top_p, top_k,
+                    api_key, model, system_prompt, temperature, max_tokens, top_p, top_k,
                     frequency_penalty, presence_penalty, repetition_penalty, stop,
                     min_p, top_a, seed, logit_bias, logprobs, top_logprobs,
                     response_format, tools, tool_choice
@@ -193,4 +191,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 fill_height=True,
             )
 
+    gr.Markdown("""
+    ## 🔧 Settings:
+    - Adjust basic parameters in the "Common Settings" section
+    - Fine-tune options in the "Advanced Settings" section
+    - Access expert-level controls in the "Expert Settings" section
+    - Upload images for multimodal interactions
+    """)
+
 demo.launch(debug=True, share=True)