aiqcamp commited on
Commit
24b5de4
·
verified ·
1 Parent(s): 826ffd7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -36,12 +36,11 @@ def stream_gemini_response(message_input: str|gr.File, messages: list) -> Iterat
36
  user_message = message_input
37
  print(f"\n=== New Request (Text) ===")
38
  print(f"User message: {user_message}")
39
- if not user_message.strip(): # Check if text message is empty or whitespace
40
- messages.append(ChatMessage(role="assistant", content="Please input a text message or upload a file. Empty input is not allowed."))
41
  yield messages
42
  return
43
 
44
-
45
  elif isinstance(message_input, gr.File): #gr.File directly should be used with newer gradio versions (v4+)
46
  input_file = message_input.name # Access the temporary file path
47
  file_type = message_input.original_name.split('.')[-1].lower() #Get original filename's extension
@@ -63,7 +62,7 @@ def stream_gemini_response(message_input: str|gr.File, messages: list) -> Iterat
63
  yield messages
64
  return
65
  else:
66
- messages.append(ChatMessage(role="assistant", content="Sorry, I cannot understand this input format."))
67
  yield messages
68
  return
69
 
@@ -81,7 +80,6 @@ def stream_gemini_response(message_input: str|gr.File, messages: list) -> Iterat
81
  response_buffer = ""
82
  thinking_complete = False
83
 
84
-
85
  # Add initial thinking message - same as before
86
  messages.append(
87
  ChatMessage(
@@ -144,7 +142,6 @@ def stream_gemini_response(message_input: str|gr.File, messages: list) -> Iterat
144
 
145
  print(f"\n=== Final Response ===\n{response_buffer}")
146
 
147
-
148
  except Exception as e:
149
  print(f"\n=== Error ===\n{str(e)}")
150
  messages.append(
 
36
  user_message = message_input
37
  print(f"\n=== New Request (Text) ===")
38
  print(f"User message: {user_message}")
39
+ if not user_message.strip(): # Robust check: if text message is empty or whitespace
40
+ messages.append(ChatMessage(role="assistant", content="Please provide a non-empty text message or upload a file.")) #More specific message
41
  yield messages
42
  return
43
 
 
44
  elif isinstance(message_input, gr.File): #gr.File directly should be used with newer gradio versions (v4+)
45
  input_file = message_input.name # Access the temporary file path
46
  file_type = message_input.original_name.split('.')[-1].lower() #Get original filename's extension
 
62
  yield messages
63
  return
64
  else:
65
+ messages.append(ChatMessage(role="assistant", content="Sorry, I cannot understand this input format. Please use text or upload a valid file.")) # More informative error
66
  yield messages
67
  return
68
 
 
80
  response_buffer = ""
81
  thinking_complete = False
82
 
 
83
  # Add initial thinking message - same as before
84
  messages.append(
85
  ChatMessage(
 
142
 
143
  print(f"\n=== Final Response ===\n{response_buffer}")
144
 
 
145
  except Exception as e:
146
  print(f"\n=== Error ===\n{str(e)}")
147
  messages.append(