Update app.py

app.py

@@ -2,6 +2,7 @@ import os, io, sys, subprocess, shutil
 from dotenv import load_dotenv
 from openai import OpenAI
 from google import genai
+from google.genai import types
 import gradio as gr
 from datetime import datetime
 from placeholder_python_code import pi_1, pi_2
@@ -67,7 +68,7 @@ def write_output(cpp: str):
     with open(f"/tmp/optimized-{current_time}.cpp", "w") as f:
         f.write(code)
 
-def convert_and_optimize_code_with_openai(python):
+def convert_and_optimize_code_with_openai(python: str):
     stream = openai_client.chat.completions.create(
         model=OPENAI_MODEL,
         messages=messages_for_python(python),
@@ -81,12 +82,15 @@ def convert_and_optimize_code_with_openai(python):
 
         yield fragment
 
-def convert_and_optimize_code_with_gemini(python):
+def convert_and_optimize_code_with_gemini(python: str):
     user_prompt = user_prompt_for_python(python)
 
     stream = gemini_client.models.generate_content_stream(
         model=GEMINI_MODEL,
-        contents=user_prompt
+        contents=user_prompt,
+        config=types.GenerateContentConfig(
+            system_instruction=system_message_for_python()
+        ),
     )
 
     for chunk in stream: