Moonfanz commited on
Commit
bc665a5
·
verified ·
1 Parent(s): dea081e

Upload 4 files

Browse files
Files changed (2) hide show
  1. func.py +8 -44
  2. requirements.txt +2 -4
func.py CHANGED
@@ -1,17 +1,11 @@
1
  from flask import jsonify
2
  import logging
3
  import os
4
-
5
- formatter = logging.Formatter('%(message)s')
6
  logger = logging.getLogger(__name__)
7
- logger.setLevel(logging.INFO)
8
- handler = logging.StreamHandler()
9
- handler.setFormatter(formatter)
10
- logger.addHandler(handler)
11
 
12
  request_counts = {}
13
 
14
- password = os.environ['password']
15
 
16
  def authenticate_request(request):
17
  auth_header = request.headers.get('Authorization')
@@ -37,25 +31,18 @@ def process_messages_for_gemini(messages, use_system_prompt=False):
37
  errors = []
38
  system_instruction_text = ""
39
  is_system_phase = use_system_prompt
40
-
41
-
42
  for i, message in enumerate(messages):
43
-
44
  role = message.get('role')
45
  content = message.get('content')
46
 
47
  if isinstance(content, str):
48
-
49
  if is_system_phase and role == 'system':
50
-
51
  if system_instruction_text:
52
  system_instruction_text += "\n" + content
53
  else:
54
  system_instruction_text = content
55
-
56
  else:
57
  is_system_phase = False
58
-
59
 
60
  if role in ['user', 'system']:
61
  role_to_use = 'user'
@@ -63,27 +50,20 @@ def process_messages_for_gemini(messages, use_system_prompt=False):
63
  role_to_use = 'model'
64
  else:
65
  errors.append(f"Invalid role: {role}")
66
- logger.error(f"Invalid role: {role}")
67
  continue
68
 
69
  if gemini_history and gemini_history[-1]['role'] == role_to_use:
70
  gemini_history[-1]['parts'].append({"text": content})
71
-
72
  else:
73
  gemini_history.append({"role": role_to_use, "parts": [{"text": content}]})
74
-
75
 
76
  elif isinstance(content, list):
77
-
78
  parts = []
79
  for item in content:
80
-
81
  if item.get('type') == 'text':
82
  parts.append({"text": item.get('text')})
83
-
84
  elif item.get('type') == 'image_url':
85
  image_data = item.get('image_url', {}).get('url', '')
86
-
87
  if image_data.startswith('data:image/'):
88
  try:
89
  mime_type, base64_data = image_data.split(';')[0].split(':')[1], image_data.split(',')[1]
@@ -93,18 +73,12 @@ def process_messages_for_gemini(messages, use_system_prompt=False):
93
  "data": base64_data
94
  }
95
  })
96
-
97
- except (IndexError, ValueError) as e:
98
- error_message = f"Invalid data URI for image: {image_data}. Error: {e}"
99
- errors.append(error_message)
100
- logger.error(error_message)
101
  else:
102
- error_message = f"Invalid image URL format for item: {item}. URL should start with 'data:image/' for inline images."
103
- errors.append(error_message)
104
- logger.error(error_message)
105
  elif item.get('type') == 'file_url':
106
  file_data = item.get('file_url', {}).get('url', '')
107
-
108
  if file_data.startswith('data:'):
109
  try:
110
  mime_type, base64_data = file_data.split(';')[0].split(':')[1], file_data.split(',')[1]
@@ -114,15 +88,10 @@ def process_messages_for_gemini(messages, use_system_prompt=False):
114
  "data": base64_data
115
  }
116
  })
117
-
118
- except (IndexError, ValueError) as e:
119
- error_message = f"Invalid data URI for file: {file_data}. Error: {e}"
120
- errors.append(error_message)
121
- logger.error(error_message)
122
  else:
123
- error_message = f"Invalid file URL format for item: {item}. URL should start with 'data:' for inline files."
124
- errors.append(error_message)
125
- logger.error(error_message)
126
 
127
  if parts:
128
  if role in ['user', 'system']:
@@ -131,18 +100,13 @@ def process_messages_for_gemini(messages, use_system_prompt=False):
131
  role_to_use = 'model'
132
  else:
133
  errors.append(f"Invalid role: {role}")
134
- logger.error(f"Invalid role: {role}")
135
  continue
136
  if gemini_history and gemini_history[-1]['role'] == role_to_use:
137
- gemini_history[-1]['parts'].extend(parts)
138
-
139
  else:
140
  gemini_history.append({"role": role_to_use, "parts": parts})
141
-
142
 
143
  if errors:
144
- logger.warning(f"Errors encountered during message processing: {errors}")
145
  return gemini_history, {"parts": [{"text": system_instruction_text}]}, (jsonify({'error': errors}), 400)
146
  else:
147
-
148
  return gemini_history, {"parts": [{"text": system_instruction_text}]}, None
 
1
  from flask import jsonify
2
  import logging
3
  import os
 
 
4
  logger = logging.getLogger(__name__)
 
 
 
 
5
 
6
  request_counts = {}
7
 
8
+ password = os.environ['ProxyPassword']
9
 
10
  def authenticate_request(request):
11
  auth_header = request.headers.get('Authorization')
 
31
  errors = []
32
  system_instruction_text = ""
33
  is_system_phase = use_system_prompt
 
 
34
  for i, message in enumerate(messages):
 
35
  role = message.get('role')
36
  content = message.get('content')
37
 
38
  if isinstance(content, str):
 
39
  if is_system_phase and role == 'system':
 
40
  if system_instruction_text:
41
  system_instruction_text += "\n" + content
42
  else:
43
  system_instruction_text = content
 
44
  else:
45
  is_system_phase = False
 
46
 
47
  if role in ['user', 'system']:
48
  role_to_use = 'user'
 
50
  role_to_use = 'model'
51
  else:
52
  errors.append(f"Invalid role: {role}")
 
53
  continue
54
 
55
  if gemini_history and gemini_history[-1]['role'] == role_to_use:
56
  gemini_history[-1]['parts'].append({"text": content})
 
57
  else:
58
  gemini_history.append({"role": role_to_use, "parts": [{"text": content}]})
 
59
 
60
  elif isinstance(content, list):
 
61
  parts = []
62
  for item in content:
 
63
  if item.get('type') == 'text':
64
  parts.append({"text": item.get('text')})
 
65
  elif item.get('type') == 'image_url':
66
  image_data = item.get('image_url', {}).get('url', '')
 
67
  if image_data.startswith('data:image/'):
68
  try:
69
  mime_type, base64_data = image_data.split(';')[0].split(':')[1], image_data.split(',')[1]
 
73
  "data": base64_data
74
  }
75
  })
76
+ except (IndexError, ValueError):
77
+ errors.append(f"Invalid data URI for image: {image_data}")
 
 
 
78
  else:
79
+ errors.append(f"Invalid image URL format for item: {item}")
 
 
80
  elif item.get('type') == 'file_url':
81
  file_data = item.get('file_url', {}).get('url', '')
 
82
  if file_data.startswith('data:'):
83
  try:
84
  mime_type, base64_data = file_data.split(';')[0].split(':')[1], file_data.split(',')[1]
 
88
  "data": base64_data
89
  }
90
  })
91
+ except (IndexError, ValueError):
92
+ errors.append(f"Invalid data URI for file: {file_data}")
 
 
 
93
  else:
94
+ errors.append(f"Invalid file URL format for item: {item}")
 
 
95
 
96
  if parts:
97
  if role in ['user', 'system']:
 
100
  role_to_use = 'model'
101
  else:
102
  errors.append(f"Invalid role: {role}")
 
103
  continue
104
  if gemini_history and gemini_history[-1]['role'] == role_to_use:
105
+ gemini_history[-1]['parts'].extend(parts)
 
106
  else:
107
  gemini_history.append({"role": role_to_use, "parts": parts})
 
108
 
109
  if errors:
 
110
  return gemini_history, {"parts": [{"text": system_instruction_text}]}, (jsonify({'error': errors}), 400)
111
  else:
 
112
  return gemini_history, {"parts": [{"text": system_instruction_text}]}, None
requirements.txt CHANGED
@@ -1,8 +1,6 @@
1
- Flask==3.1.0
2
- Flask-CORS==5.0.0
3
  requests==2.32.3
4
  Werkzeug==3.1.3
5
- google==3.0.0
6
- google-generativeai==0.8.3
7
  pillow==10.3.0
8
  APScheduler==3.11.0
 
1
+ Flask==2.0.3
2
+ Flask-CORS==3.0.10
3
  requests==2.32.3
4
  Werkzeug==3.1.3
 
 
5
  pillow==10.3.0
6
  APScheduler==3.11.0