changed

app.py

@@ -4,9 +4,32 @@ import gradio as gr
 from typing import List, Tuple
 import numpy as np
 
+import google.generativeai as genai
+
 def reset() -> List:
     return []
 
+def interact_summarization(prompt: str, article: str, temp = 1.0) -> List[Tuple[str, str]]:
+    '''
+      * Arguments
+        - prompt: the prompt that we use in this section
+        - article: the article to be summarized
+        - temp: the temperature parameter of this model. Temperature is used to control the output of the chatbot. The higher the temperature is, the more creative response you will get.
+    '''
+    input = f"{prompt}\n{article}"
+    response = model.generate_content(
+      input,
+      generation_config=genai.types.GenerationConfig(temperature=temp),
+      safety_settings=[
+          {"category": "HARM_CATEGORY_HARASSMENT","threshold": "BLOCK_NONE",},
+          {"category": "HARM_CATEGORY_HATE_SPEECH","threshold": "BLOCK_NONE",},
+          {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","threshold": "BLOCK_NONE",},
+          {"category": "HARM_CATEGORY_DANGEROUS_CONTENT","threshold": "BLOCK_NONE",},
+          ]
+    )
+
+    return [(input, response.text)]
+
 def interact(chatbot: List[Tuple[str, str]], user_input: str) -> List[Tuple[str, str]]:
     responses = ["You are right", "HaHa", "I don't know"]
     response = np.random.choice(responses, 1)[0]
@@ -14,14 +37,24 @@ def interact(chatbot: List[Tuple[str, str]], user_input: str) -> List[Tuple[str,
 
     return chatbot
 
-with gr.Blocks() as demo:
-    gr.Markdown(f"# Gradio Tutorial")
-    chatbot = gr.Chatbot()
-    input_textbox = gr.Textbox(label="Input", value = "")
-    with gr.Row():
-        sent_button = gr.Button(value="Send")
-        reset_button = gr.Button(value="Reset")
-    sent_button.click(interact, inputs=[chatbot, input_textbox], outputs=[chatbot])
-    reset_button.click(reset, outputs=[chatbot])
-
-demo.launch(debug = True)
+
+if __name__ == '__main__':
+    GOOGLE_API_KEY="AIzaSyB3pzEEQcMhqmJeIhfkOVebH7C4dm_4yTc"
+    genai.configure(api_key=GOOGLE_API_KEY)
+    model = genai.GenerativeModel('gemini-pro')
+
+    with gr.Blocks() as demo:
+        gr.Markdown(f"# Gradio Tutorial")
+        chatbot = gr.Chatbot()
+        prompt_textbox = gr.Textbox(label="Prompt", value='Please summarize the following article', visible=True)
+        input_textbox = gr.Textbox(label="Input", value = "")
+        with gr.Column():
+            gr.Markdown("#  Temperature\n Temperature is used to control the output of the chatbot. The higher the temperature is, the more creative response you will get.")
+            temperature_slider = gr.Slider(0.0, 1.0, 0.7, step = 0.1, label="Temperature")
+        with gr.Row():
+            sent_button = gr.Button(value="Send")
+            reset_button = gr.Button(value="Reset")
+        sent_button.click(interact_summarization, inputs=[prompt_textbox, input_textbox, temperature_slider], outputs=[chatbot])
+        reset_button.click(reset, outputs=[chatbot])
+
+    demo.launch(debug = True)