srbmihaicode commited on
Commit
3626254
·
verified ·
1 Parent(s): 5175e14

Removed default suggestion function

Browse files
Files changed (1) hide show
  1. app.py +1 -22
app.py CHANGED
@@ -8,23 +8,6 @@ DEFAULT_MAX_TOKENS = 512
8
  DEFAULT_TEMPERATURE = 0.7
9
  DEFAULT_TOP_P = 0.95
10
 
11
- def generate_response(message, system_message):
12
- messages = [{"role": "system", "content": system_message}]
13
- messages.append({"role": "user", "content": message})
14
-
15
- response = ""
16
- for message in client.chat_completion(
17
- messages,
18
- max_tokens=DEFAULT_MAX_TOKENS,
19
- stream=True,
20
- temperature=DEFAULT_TEMPERATURE,
21
- top_p=DEFAULT_TOP_P,
22
- ):
23
- token = message.choices[0].delta.content
24
- response += token
25
-
26
- return response
27
-
28
  def generate_journal_suggestion(current_page):
29
  suggestion_prompt = (
30
  f"Pe baza înregistrării din jurnal: '{current_page}', sugerează ce ar putea scrie utilizatorul în continuare în jurnalul său. "
@@ -56,16 +39,12 @@ def chat():
56
  system_message = data.get("system_message", "You are a friendly chatbot.")
57
  journal_page = data.get("journal_page", "")
58
 
59
- if not message:
60
- return jsonify({"error": "Message is required."}), 400
61
-
62
- response = generate_response(message, system_message)
63
 
64
  suggestion = ""
65
  if journal_page:
66
  suggestion = generate_journal_suggestion(journal_page)
67
 
68
- return jsonify({"response": response, "journal_suggestion": suggestion})
69
  except Exception as e:
70
  return jsonify({"error": str(e)}), 500
71
 
 
8
  DEFAULT_TEMPERATURE = 0.7
9
  DEFAULT_TOP_P = 0.95
10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  def generate_journal_suggestion(current_page):
12
  suggestion_prompt = (
13
  f"Pe baza înregistrării din jurnal: '{current_page}', sugerează ce ar putea scrie utilizatorul în continuare în jurnalul său. "
 
39
  system_message = data.get("system_message", "You are a friendly chatbot.")
40
  journal_page = data.get("journal_page", "")
41
 
 
 
 
 
42
 
43
  suggestion = ""
44
  if journal_page:
45
  suggestion = generate_journal_suggestion(journal_page)
46
 
47
+ return jsonify({"journal_suggestion": suggestion})
48
  except Exception as e:
49
  return jsonify({"error": str(e)}), 500
50