thunder-lord commited on
Commit
b7f3228
·
verified ·
1 Parent(s): f451e93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -4
app.py CHANGED
@@ -297,11 +297,21 @@ def gemini():
297
  if not prompt:
298
  return jsonify({'error': 'Prompt parameter is required'}), 400
299
 
300
- convo = model.start_chat(history=[prompt])
301
- convo.send_message(prompt) # Simulating user input with the prompt itself
302
- response = convo.last.text
 
 
 
 
 
 
 
303
 
304
- return jsonify({'response': response})
 
 
 
305
 
306
  if __name__ == "__main__":
307
  app.run(host="0.0.0.0", port=7860, debug=True)
 
297
  if not prompt:
298
  return jsonify({'error': 'Prompt parameter is required'}), 400
299
 
300
+ try:
301
+ app.logger.info("Received prompt: %s", prompt)
302
+ convo = model.start_chat(history=[prompt])
303
+
304
+ # Simulate multi-turn conversation by sending additional messages
305
+ for _ in range(3): # Send 3 additional messages (adjust as needed)
306
+ convo.send_message("") # Send an empty message placeholder
307
+
308
+ response = convo.last.text
309
+ return jsonify({'response': response})
310
 
311
+ except Exception as e:
312
+ error_message = str(e)
313
+ app.logger.error("Failed to generate content: %s", error_message)
314
+ return jsonify({'error': 'Failed to generate content. Error: {}'.format(error_message)}), 500
315
 
316
  if __name__ == "__main__":
317
  app.run(host="0.0.0.0", port=7860, debug=True)