Update app.py

app.py

@@ -15,25 +15,25 @@ TITLE = """<h1 align="center">ReffidGPT Playground 💬</h1>"""
 
 AVATAR_IMAGES = (
     None,
-    "https://i.postimg.cc/T1hC17DL/20240603-211345.png"
+    "https://i.postimg.cc/wT1WbBdL/20240603-204837.png"
 )
 
 IMAGE_CACHE_DIRECTORY = "/tmp"
 IMAGE_WIDTH = 512
 CHAT_HISTORY = List[Tuple[Optional[Union[Tuple[str], str]], Optional[str]]]
 
+# Default system prompt (integrated into the bot function directly)
+SYSTEM_PROMPT = "You are ReffidGPT, a helpful assistant. Respond in a friendly and informative manner. Your Name ReffidGPT & Your Creator Is Groqcin Technologies Inc."
 
 def preprocess_stop_sequences(stop_sequences: str) -> Optional[List[str]]:
     if not stop_sequences:
         return None
     return [sequence.strip() for sequence in stop_sequences.split(",")]
 
-
 def preprocess_image(image: Image.Image) -> Optional[Image.Image]:
     image_height = int(image.height * IMAGE_WIDTH / image.width)
     return image.resize((IMAGE_WIDTH, image_height))
 
-
 def cache_pil_image(image: Image.Image) -> str:
     image_filename = f"{uuid.uuid4()}.jpeg"
     os.makedirs(IMAGE_CACHE_DIRECTORY, exist_ok=True)
@@ -41,7 +41,6 @@ def cache_pil_image(image: Image.Image) -> str:
     image.save(image_path, "JPEG")
     return image_path
 
-
 def preprocess_chat_history(
     history: CHAT_HISTORY
 ) -> List[Dict[str, Union[str, List[str]]]]:
@@ -52,10 +51,9 @@ def preprocess_chat_history(
         elif user_message is not None:
             messages.append({'role': 'user', 'parts': [user_message]})
         if model_message is not None:
-            messages.append({'role': 'model', 'parts': [model_message]})
+            messages.append({'role': 'user', 'parts': [model_message]})
     return messages
 
-
 def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
     for file in files:
         image = Image.open(file).convert('RGB')
@@ -64,13 +62,11 @@ def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
         chatbot.append(((image_path,), None))
     return chatbot
 
-
 def user(text_prompt: str, chatbot: CHAT_HISTORY):
     if text_prompt:
         chatbot.append((text_prompt, None))
     return "", chatbot
 
-
 def bot(
     google_key: str,
     files: Optional[List[str]],
@@ -98,6 +94,9 @@ def bot(
         top_k=top_k,
         top_p=top_p)
 
+    # Integrate system prompt directly in the input to the model
+    system_prompt_message = [{'role': 'user', 'parts': [SYSTEM_PROMPT]}]
+
     if files:
         text_prompt = [chatbot[-1][0]] \
             if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
@@ -110,6 +109,7 @@ def bot(
             generation_config=generation_config)
     else:
         messages = preprocess_chat_history(chatbot)
+        messages = system_prompt_message + messages  # Prepend system prompt
         model = genai.GenerativeModel('gemini-pro')
         response = model.generate_content(
             messages,
@@ -125,7 +125,6 @@ def bot(
             time.sleep(0.01)
             yield chatbot
 
-
 google_key_component = gr.Textbox(
     label="GOOGLE API KEY",
     value="",
@@ -147,7 +146,7 @@ text_prompt_component = gr.Textbox(
 upload_button_component = gr.UploadButton(
     label="Upload Images", file_count="multiple", file_types=["image"], scale=1
 )
-run_button_component = gr.Button(value="Send", variant="primary", scale=1)
+run_button_component = gr.Button(value="Run", variant="primary", scale=1)
 temperature_component = gr.Slider(
     minimum=0,
     maximum=1.0,