add form validation

app.py

@@ -3,14 +3,23 @@ import os
 import re
 import requests
 
-from bs4 import BeautifulSoup
 import gradio as gr
+from bs4 import BeautifulSoup
 from spiralfilm import FilmCore, FilmConfig
 
+from logging import getLogger, DEBUG, StreamHandler
+
+logger = getLogger()
+logger.setLevel(DEBUG)
+handler = StreamHandler()
+handler.setLevel(DEBUG)
+logger.addHandler(handler)
+
 
 def greet(name):
     return "こんにちは " + name + "さん!! \n僕はパスカルくんだよ。よろしくね"
 
+
 def extract_texts(input_str):
     pattern = r"msg='([^']*)'"
     matches = re.findall(pattern, input_str)
@@ -38,29 +47,39 @@ async def summarize(input_text: str, input_url: str):
             config=config
         ).run_async()
     if input_url:
-        res = requests.get(input_url)
-        soup = BeautifulSoup(res.text)
-        url_content = soup.find('title').text + '\n' + soup.find('body').text
-        _prompt = f"""
-        以下の文章を要約してください。
-        {url_content}
-        """
+        try:
+            res = requests.get(input_url)
+            soup = BeautifulSoup(res.text)
+            url_content = soup.find('title').text + '\n' + soup.find('body').text
+            _prompt = f"""
+            以下の文章を要約してください。
+            {url_content}
+            """
+        except Exception:
+            raise gr.Error("WEBページの取得に失敗しました。")
 
-        res = await FilmCore(
+        return await FilmCore(
             prompt=_prompt,
             system_prompt="あなたは優秀なライターです。",
             config=config
         ).run_async()
+    else:
+        raise gr.Error("LLMの実行に失敗しました。")
 
-        return res
 
+def validate_input_form(input_text, input_url):
+    input_value = input_text + input_url
+    # Check if the text is not blank
+    if len(input_value) < 1:
+        raise gr.Error("テキストかURLを入力してください。")
     else:
-        return "テキストかURLを入力してください。"
-
+        return
 
 
 async def chat(input_text, input_url):
+    validate_input_form(input_text, input_url)
     summary = await summarize(input_text, input_url)
+    logger.info(f"summary: {summary}")
     endpoint = os.environ.get("TWINROOM_API_BASE")
     payload = {
         "content": summary
@@ -74,13 +93,15 @@ async def chat(input_text, input_url):
         result += f'\n{input_url}'
     return result
 
+
 with gr.Blocks() as iface:
     # UI
     with gr.Row():
         with gr.Column():
             input_text = gr.Textbox(label="テキスト")
             input_url = gr.Textbox(label="URL")
-            chat_btn = gr.Button("Chat")
+
+            chat_btn = gr.Button("Chats")
         with gr.Column():
             output_text = gr.Textbox(label="回答")
 
@@ -89,4 +110,4 @@ with gr.Blocks() as iface:
 
 
 if __name__ == "__main__":
-    iface.launch(auth=("spiralai", "spiralai"), share=True)
+    iface.launch(auth=("spiralai", "spiralai"), share=True, server_name="0.0.0.0")

requirements.txt

@@ -1,2 +1,3 @@
 SpiralFilm==0.2.5
-BeautifulSoup4==4.12.3
+BeautifulSoup4==4.12.3
+gradio==4.25.0