Spaces:

Tim-Saijun
/

Sagent

Paused

App Files Files Community

Tim-Saijun commited on Jun 25, 2024

Commit

8cd99ea

1 Parent(s): 387d8b7

feat: 文件上传脚本

Browse files

Files changed (3) hide show

playground/upload.html +14 -0
playground/upload_server.py +48 -0
playground/writer.py +72 -45

playground/upload.html ADDED Viewed

	@@ -0,0 +1,14 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>File Upload</title>
+</head>
+<body>
+    <h1>File Upload</h1>
+    <form action="/upload" method="post" enctype="multipart/form-data">
+        <input type="file" name="file">
+        <input type="submit" value="Upload">
+    </form>
+</body>
+</html>

playground/upload_server.py ADDED Viewed

	@@ -0,0 +1,48 @@

+import http.server
+import socketserver
+import os
+from urllib.parse import parse_qs, urlparse
+from http import cookies
+PORT = 8000
+DIRECTORY = "."
+class MyHttpRequestHandler(http.server.SimpleHTTPRequestHandler):
+    def do_GET(self):
+        if self.path == '/':
+            self.path = 'upload.html'
+        return super().do_GET()
+    def do_POST(self):
+        content_length = int(self.headers['Content-Length'])
+        post_data = self.rfile.read(content_length)
+        boundary = self.headers['Content-Type'].split('boundary=')[-1].encode()
+        parts = post_data.split(boundary)
+        for part in parts:
+            if b'Content-Disposition: form-data; name="file";' in part:
+                file_data = part.split(b'\r\n\r\n')[1].strip()
+                file_name = part.split(b'filename="')[1].split(b'"')[0].decode()
+                file_path = os.path.join(DIRECTORY, file_name)
+                with open(file_path, 'wb') as file:
+                    file.write(file_data)
+                self.send_response(200)
+                self.send_header('Content-Type', 'text/html')
+                self.end_headers()
+                self.wfile.write(b'File uploaded and overwritten successfully!')
+                return
+        self.send_response(400)
+        self.send_header('Content-Type', 'text/html')
+        self.end_headers()
+        self.wfile.write(b'File upload failed.')
+Handler = MyHttpRequestHandler
+with socketserver.TCPServer(("", PORT), Handler) as httpd:
+    print(f"Serving at port {PORT}")
+    httpd.serve_forever()

playground/writer.py CHANGED Viewed

@@ -1,6 +1,5 @@
 import os
 import getpass
-import quart
 from quart import Quart, request
@@ -39,15 +38,16 @@ absract_plan_prompt = ChatPromptTemplate.from_messages(
     [
         (
             "system",
-            """Please generate an English SEO article outline based on the following title, core keywords, related keywords and user information. The writing style should be in the first person, and oral description is preferred.
-Requirements for generating an article outline:
-1. Secondary title: Generate multiple secondary titles based on the title, core keywords, and related keywords, no more than 6, and the content of the title should have a logical order and relevance.
-2. Summary: Based on each secondary title, generate a content summary around the secondary title.
-3. SEO word distribution strategy: Generate an SEO word distribution strategy for the current secondary title content.
-4. Heading: The heading uses the heading given by the user.
-5. Your outline should be coherent, so that the paragraphs are coupled with each other and complement each other, like a complete article.
-Note: Directly return to the secondary title, topic, and SEO word distribution strategy. Do not have a summary or conclusion.
-""",
         ),
         ("placeholder", "{messages}"),
     ]
@@ -69,8 +69,11 @@ pre_plan_prompt = ChatPromptTemplate.from_messages(
     [
         (
             "system",
-            """Please extract the sub-titles of each paragraph in the unstructured data of the received article outline, such as '1.xxx' and the corresponding 'Abstract' and 'SEO word distribution strategy', and add them to the list at once.
-For example, \"title:xxxx,topic:xxxx,seo:xxxx\"
 """
         ),
         ("placeholder", "{messages}"),
@@ -84,9 +87,35 @@ designer_prompt = ChatPromptTemplate.from_messages(
     [
         (
             "system",
-            """You are a professional SEO article writer. You need to generate the paragraph content of the current secondary title based on the secondary title [title], topic [topic], and SEO word strategy [seo] and retain the secondary title. The content is output in pure English.
-Note:1. Please ensure the relevance between the secondary title and the topic content.2. Please refer to the materials and user materials for generation.
-            """
         ),
         ("placeholder", "{messages}"),
     ]
@@ -148,22 +177,22 @@ async def DESIGNER(state: GraphState):
                 break
     response = await designer_model.ainvoke({"messages": [("user", f"{messages[::-1]}")]})
-    # payload = {"language": "en",
-    #            "text": response.content,
-    #            "strength": 4,
-    #            }
-    # async with aiohttp.ClientSession() as session:
-    #     async with session.post(url, headers=headers, json=payload) as res:
-    #         if res.status == 201:
-    #             rewritten_text = await res.json()
-    #             rewritten_text = await no_conclusion_model.ainvoke(
-    #                 {"messages": [("user", rewritten_text["rewrites"][0]["rewrite"])]})
-    #         else:
-    #             print("Error:", res.status, await res.text())
-    #             return {"messages": []}
     i = state["time"]
-    # return {"messages": [messages.content], "time": i + 1}
-    return {"messages": [response.content], "time": i + 1}
 from typing import Literal
@@ -173,7 +202,6 @@ def route(state: GraphState) -> Literal["__end__", "action"]:
     # If there is no function call, then we finish
     if state["time"] < len(state["plan"]):
         return "action"
-    # Otherwise if there is, we continue
     else:
         return "__end__"
@@ -182,7 +210,6 @@ def should_end(state: GraphState) -> Literal["__end__", "dliver"]:
     # If there is no function call, then we finish
     if state["time"] < len(state["plan"]):
         return "dliver"
-    # Otherwise if there is, we continue
     else:
         return "__end__"
@@ -192,33 +219,28 @@ from langgraph.graph import StateGraph
 # Define a new graph
 workflow = StateGraph(GraphState)
-# Define the two nodes we will cycle between
 workflow.add_node("abstract_planner", abstrct_plan_step)
 workflow.add_node("planner", plan_step)
 workflow.add_node("action", DESIGNER)
 workflow.add_node("dliver", DLIVER)
-# Set the entrypoint as `agent`
-# This means that this node is the first one called
 workflow.set_entry_point("abstract_planner")
-# We now add a conditional edge
 workflow.add_conditional_edges(
-    # First, we define the start node. We use `agent`.
-    # This means these are the edges taken after the `agent` node is called.
     "action",
-    # Next, we pass in the function that will determine which node is called next.
     should_end,
 )
-# We now add a normal edge from `tools` to `agent`.
-# This means that after `tools` is called, `agent` node is called next.
 workflow.add_edge("abstract_planner", "planner")
 workflow.add_edge("planner", "dliver")
 workflow.add_edge("dliver", "action")
-# Finally, we compile it!
-# This compiles it into a LangChain Runnable,
-# meaning you can use it as you would any other runnable
 app = workflow.compile()
@@ -242,7 +264,6 @@ config = {"recursion_limit": 500}
 q_app = Quart(__name__)
-@q_app.route('/write', methods=['GET'])
 async def main():
     title = request.args.get('title')
     core_keywords = request.args.get('core_keywords')
@@ -262,6 +283,12 @@ async def main():
     return messages
 import asyncio
 if __name__ == "__main__":

 import os
 import getpass
 from quart import Quart, request
     [
         (
             "system",
+            """
+            Please generate an English SEO article outline based on the following title, core keywords, related keywords and user information. The writing style should be first-person, and oral description is preferred.
+            Requirements for generating an article outline:
+            1. Based on the professional title, you must create at least 4 subheadings. You must expand each subheading with relevant third-level titles, and the title content must have a logical order and relevance.
+            2. Summary: Based on each subheading, generate a content summary around the subheading.
+            3. SEO wrd segmentation strategy: Generate an SEO word segmentation strategy for the current subheading content.
+            4. Title: The title uses the title given by the user.
+            5. Your outline should be coherent, so that the paragraphs are coupled and complement each other, like a complete article.
+            Note: Go directly back to the subheading, topic and SEO word segmentation strategy. No summary or conclusion.
+            """,
         ),
         ("placeholder", "{messages}"),
     ]
     [
         (
             "system",
+            """
+            Please extract the subtitles and third-level titles in the unstructured data of the
+            received article outline, such as '1.xxx','1.1.xxx' and the corresponding 'Abstract' and 'SEO word
+            distribution strategy', and add them to the list at one time.
+            For example, \"title:xxxx,third-level titles:xxx,topic:xxxx,seo:xxxx\"
 """
         ),
         ("placeholder", "{messages}"),
     [
         (
             "system",
+            """You are a professional SEO article writer. You need to generate the paragraph
+            content of the current secondary title based on the secondary title [title], topic [topic],
+             and SEO word strategy [seo] and retain the secondary title. The content is output in pure
+             English.
+             You must output in HTML format:
+             -- Subtitles: `<h2>` tag
+             -- Main text: `<p>` tag
+             Note:1. Please ensure the relevance between the secondary title and the topic content.
+             2. Please refer to the materials and user materials for generation.The number of words in each paragraph is 100-200.
+             Instructions:
+             -- You must return only the article content, no outline or explanations.
+             -- Core and related keywords must appear according to specified frequencies and at least once.
+             -- Content must be coherent, free of spelling errors, and logically consistent.
+             -- Core keyword frequency must be 3%.
+             -- Related keyword frequency must be 1%.
+             -- Do not display the generated title or the `<h1>` tag. Directly start with the content.
+             -- Do not use markdown syntax.
+             -- Core keywords must be used exactly as provided.
+             -- Content must be detailed and comprehensive.
+             -- Article must include an introduction and a conclusion.
+             -- The article length must be greater than 750 words.
+             -- Do not start with the <html> tag.
+             -- Do not include the </html> tag at the end.
+             -- You must follow the above steps and instructions meticulously.
+             -- You must ensure the article aligns with SEO bes practices.
+             -- Maintain an engaging and anthropomorphic writing style.
+             -- Strive for a perfect balance between keyword density and natural readability.
+             -- If related keywords are not provided, proceed with generating the article using only the core keywords.
+          """
         ),
         ("placeholder", "{messages}"),
     ]
                 break
     response = await designer_model.ainvoke({"messages": [("user", f"{messages[::-1]}")]})
+    payload = {"language": "en",
+               "text": response.content,
+               "strength": 4,
+               }
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, headers=headers, json=payload) as res:
+            if res.status == 201:
+                rewritten_text = await res.json()
+                rewritten_text = await no_conclusion_model.ainvoke(
+                    {"messages": [("user", rewritten_text["rewrites"][0]["rewrite"])]})
+            else:
+                print("Error:", res.status, await res.text())
+                return {"messages": []}
     i = state["time"]
+    return {"messages": [rewritten_text.content], "time": i + 1}
+    # return {"messages": [response.content], "time": i + 1}
 from typing import Literal
     # If there is no function call, then we finish
     if state["time"] < len(state["plan"]):
         return "action"
     else:
         return "__end__"
     # If there is no function call, then we finish
     if state["time"] < len(state["plan"]):
         return "dliver"
     else:
         return "__end__"
 # Define a new graph
 workflow = StateGraph(GraphState)
 workflow.add_node("abstract_planner", abstrct_plan_step)
 workflow.add_node("planner", plan_step)
 workflow.add_node("action", DESIGNER)
 workflow.add_node("dliver", DLIVER)
 workflow.set_entry_point("abstract_planner")
 workflow.add_conditional_edges(
     "action",
     should_end,
 )
 workflow.add_edge("abstract_planner", "planner")
 workflow.add_edge("planner", "dliver")
 workflow.add_edge("dliver", "action")
 app = workflow.compile()
 q_app = Quart(__name__)
 async def main():
     title = request.args.get('title')
     core_keywords = request.args.get('core_keywords')
     return messages
+@q_app.route('/write', methods=['GET'])
+async def request_result():
+    result = await main()
+    return result
 import asyncio
 if __name__ == "__main__":