-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathchatbot_app.py
65 lines (51 loc) · 2.17 KB
/
chatbot_app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from flask import Flask, request, jsonify, render_template
import pandas as pd
from transformers import pipeline
from pyngrok import ngrok
app = Flask(__name__)
# CSV dosyasını yükleme
dilekce = pd.read_csv("file.csv")
# Türkçe soru cevaplama modeli
model_name = "dbmdz/bert-base-turkish-cased"
qa_pipeline = pipeline("question-answering", model=model_name)
# Rastgele 10 bağlam seçip birleştiriyoruz ve sabitliyoruz
context = " ".join(dilekce['target_text'].dropna().astype(str).sample(n=10).tolist())
def complete_sentence(text):
# Tamamlanmamış cümleleri bul ve tamamla
if text and text[-1] not in '.!?':
last_sentence_end = max(text.rfind('.'), text.rfind('!'), text.rfind('?'))
if last_sentence_end != -1:
text = text[:last_sentence_end + 1]
return text
def extend_answer_length(answer, context, min_length=300):
words = answer.split()
if len(words) >= min_length:
return answer
remaining_context = context.replace(answer, "")
additional_text = remaining_context[:min_length * 5] # Geniş bir tampon alan alıyoruz
extended_answer = answer + additional_text
extended_answer = complete_sentence(extended_answer)
return extended_answer
def answer_question(question, context, min_length=300):
result = qa_pipeline(question=question, context=context, max_answer_len=500, min_answer_len=250)
answer = result['answer']
extended_answer = extend_answer_length(answer, context, min_length=min_length)
return extended_answer
@app.route("/", methods=["GET", "POST"])
def index():
return render_template("index.html")
@app.route("/get_response", methods=["POST"])
def get_response():
data = request.get_json()
if 'user_input' in data:
user_input = data['user_input']
response = answer_question(user_input, context)
return jsonify({"response": response})
else:
return jsonify({"error": "User input not found"}), 400
if __name__ == '__main__':
# Ngrok tünelini başlatma ve URL'yi alma
public_url = ngrok.connect(5000)
print(f"Flask uygulamasına erişmek için bu URL'yi kullanın: {public_url}")
# Flask uygulamasını başlatma
app.run(port=5000)