add bs4

app.py

@@ -1,7 +1,9 @@
-import os
 import json
-import requests
+import os
 import re
+import requests
+
+from bs4 import BeautifulSoup
 import gradio as gr
 from spiralfilm import FilmCore, FilmConfig
 
@@ -22,13 +24,15 @@ async def summarize(input_text: str, input_url: str):
         {input_text}
         """
     elif input_url:
-        url_content = "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}
         """
     else:
-        _prompt = "文章が入力するように促してください。"
+        return "テキストかURLを入力してください。"
 
     config = FilmConfig(
         "gpt-3.5-turbo-16k",
@@ -57,11 +61,8 @@ async def chat(input_text, input_url):
     headers = {'API-Key': os.environ.get("TWINROOM_API_KEY")}
     json_payload = json.dumps(payload)
     response = requests.post(endpoint, headers=headers, data=json_payload)
-    print(response.text)
     response_msgs = extract_texts(response.text)
-    print(response_msgs)
     result = ''.join(response_msgs)
-    print(result)
     return result
 
 with gr.Blocks() as iface:

requirements.txt

@@ -1,2 +1,2 @@
 SpiralFilm==0.2.5
-
+BeautifulSoup4=4.12.3