Update app.py

app.py

@@ -36,7 +36,8 @@ def process_file(uploaded_file):
     if file_type in ["jpg", "jpeg", "png"]:
         return {"type": "image", "content": Image.open(uploaded_file).convert('RGB')}
     
-    if file_type in ["txt", *code_extensions := ["html", "css", "php", "js", "py", "java", "c", "cpp"]]:
+    code_extensions = ["html", "css", "php", "js", "py", "java", "c", "cpp"]
+    if file_type in ["txt"] + code_extensions:
         return {"type": "text", "content": uploaded_file.read().decode("utf-8")}
     
     if file_type in ["csv", "xlsx"]:
@@ -57,18 +58,17 @@ def process_file(uploaded_file):
 with st.sidebar:
     api_key = st.text_input("Google AI API Key", type="password")
     model = st.selectbox("Model", [
-            "gemini-1.5-flash",
-            "gemini-1.5-pro",
-            "gemini-1.5-flash-8B",
-            "gemini-1.5-pro-vision-latest",
-            "gemini-1.0-pro",
-            "gemini-1.0-pro-vision-latest",
-
-            "gemini-2.0-pro-exp-02-05",
-            "gemini-2.0-flash-lite",
-            "gemini-2.0-flash-exp-image-generation",
-            "gemini-2.0-flash",
-            "gemini-2.0-flash-thinking-exp-01-21"
+        "gemini-1.5-flash",
+        "gemini-1.5-pro",
+        "gemini-1.5-flash-8B",
+        "gemini-1.5-pro-vision-latest",
+        "gemini-1.0-pro",
+        "gemini-1.0-pro-vision-latest",
+        "gemini-2.0-pro-exp-02-05",
+        "gemini-2.0-flash-lite",
+        "gemini-2.0-flash-exp-image-generation",
+        "gemini-2.0-flash",
+        "gemini-2.0-flash-thinking-exp-01-21"
     ])
     temperature = st.slider("Temperature", 0.0, 1.0, 0.7)
     max_tokens = st.slider("Max Tokens", 1, 2048, 1000)