Update app.py

app.py

@@ -52,6 +52,14 @@ async def generate_response(request_body: RequestBody):
             pattern = r'https?://\S+\.(?:jpg|jpeg|png|gif)'
             match2 = re.search(pattern, decoded_string)
             all_chunk = match2.group()
+
+        if model == 'gemini2':
+            key_gemini2 = os.environ['key_gemini2']
+            headers = {'Content-Type': 'application/json',}
+            params = {'key': key_gemini2}
+            json_data = {'contents': [{'parts': [{'text': input_text}]}]}
+            response = requests.post('https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent', params=params, headers=headers, json=json_data)
+            all_chunk = response.json()['candidates'][0]['content']['parts'][0]['text']
             
     if key_body != key_true:
         all_chunk = "How's the hack going?"