ASG Models commited on
Commit
9c4bf4f
·
verified ·
1 Parent(s): d489b8c

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +87 -0
app.py ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import os
3
+
4
+ import google.generativeai as genai
5
+ api_key =os.environ.get("id_gmkey")
6
+
7
+ genai.configure(api_key=api_key)
8
+
9
+ generation_config = {
10
+ "temperature": 1,
11
+ "top_p": 0.95,
12
+ "top_k": 64,
13
+ "max_output_tokens": 8192,
14
+ "response_mime_type": "text/plain",
15
+ }
16
+
17
+ model = genai.GenerativeModel(
18
+ model_name="gemini-1.5-pro",
19
+ generation_config=generation_config,
20
+ # safety_settings = Adjust safety settings
21
+ # See https://ai.google.dev/gemini-api/docs/safety-settings
22
+ )
23
+
24
+ def create_chat_session():
25
+ chat_session = model.start_chat(
26
+ history=[
27
+ {
28
+ "role": "user",
29
+ "parts": [
30
+ "السلام عليكم اريد منك ان ترد على اسئلتي دائما باللهجة السعودية النجدية \n\n",
31
+ ],
32
+ },
33
+ {
34
+ "role": "model",
35
+ "parts": [
36
+ "هلا والله، إسأل ما في خاطرك وأنا حاضر أساعدك، بس بشرط واحد، أسئلتك تكون واضحة عشان أفهم عليك عدل وأعطيك الجواب الزين. قل وش تبي وأنا حاضر! \n",
37
+ ],
38
+ },
39
+ {
40
+ "role": "user",
41
+ "parts": [
42
+ "كيف حالك اخبارك\n",
43
+ ],
44
+ },
45
+ {
46
+ "role": "model",
47
+ "parts": [
48
+ "هلا والله وغلا، أنا طيب وبخير الحمد لله، انت كيفك؟ عساك طيب؟ \n \n وش عندك أخبار؟ عسى كلها زينة. \n",
49
+ ],
50
+ },
51
+ {
52
+ "role": "user",
53
+ "parts": [
54
+ "اريد ايضا ان تكون اجابتك مختصره على سبيل المثال ااكثر اجابة سطرين\n",
55
+ ],
56
+ },
57
+ {
58
+ "role": "model",
59
+ "parts": [
60
+ "خلاص، فهمتك. من عيوني، أسئلتك من اليوم وطالع أجوبتها ما تتعدى سطرين. \n \n إسأل وشف! \n",
61
+ ],
62
+ },
63
+ ]
64
+ )
65
+ return chat_session
66
+
67
+ AI=create_chat_session()
68
+
69
+
70
+
71
+ def get_answer_ai(text):
72
+ global AI
73
+ try:
74
+ response = AI.send_message(text)
75
+ return response.text
76
+
77
+
78
+ except :
79
+ AI=create_chat_session()
80
+ response = AI.send_message(text)
81
+ return response.text
82
+ def dash(text):
83
+ textai=get_answer_ai(text)
84
+ return textai
85
+
86
+ demo = gr.Interface(fn=greet, inputs="text", outputs="text")
87
+ demo.launch()