Update app.py

app.py

@@ -61,114 +61,6 @@ def hf_chat(api_key, model, text):
 
     return stream.choices[0].message.content
 
-
-
-import openai
-
-openai.api_key = os.getenv("HF_TOKEN")
-openai.api_base = "YOUR_API_ENDPOINT"  # 例: "https://your-api-provider.com/v1"
-model_name = "YOUR_MODEL_NAME" # 例: "my-custom-model"
-
-def generate_text(prompt):
-  try:
-    response = openai.Completion.create(
-        engine=model_name,
-        prompt=prompt,
-        n=1,             
-        stop=None,       
-        temperature=0.9  # ランダム性 (0: deterministic, 1: creative)
-    )
-    return response.choices[0].text.strip()
-  except Exception as e:
-    print(f"Error happend: {e}")
-    return None
-
-class GoogleSearchTool(Tool):
-    name = "web_search"
-    description = """Performs a google web search for your query then returns a string of the top search results."""
-    inputs = {
-        "query": {"type": "string", "description": "The search query to perform."},
-        "filter_year": {
-            "type": "integer",
-            "description": "Optionally restrict results to a certain year",
-            "nullable": True,
-        },
-    }
-    output_type = "string"
-
-    def __init__(self):
-        super().__init__(self)
-        import os
-
-        self.serpapi_key = os.getenv("SERPER_API_KEY")
-
-    def forward(self, query: str, filter_year: Optional[int] = None) -> str:
-        import requests
-
-        if self.serpapi_key is None:
-            raise ValueError("Missing SerpAPI key. Make sure you have 'SERPER_API_KEY' in your env variables.")
-
-        params = {
-            "engine": "google",
-            "q": query,
-            "api_key": self.serpapi_key,
-            "google_domain": "google.com",
-        }
-
-        headers = {
-        'X-API-KEY': self.serpapi_key,
-        'Content-Type': 'application/json'
-        }
-
-        if filter_year is not None:
-            params["tbs"] = f"cdr:1,cd_min:01/01/{filter_year},cd_max:12/31/{filter_year}"
-
-        response = requests.request("POST", "https://google.serper.dev/search", headers=headers, data=json.dumps(params))
-
-
-        if response.status_code == 200:
-            results = response.json()
-        else:
-            raise ValueError(response.json())
-
-        if "organic" not in results.keys():
-            print("REZZZ", results.keys())
-            if filter_year is not None:
-                raise Exception(
-                    f"No results found for query: '{query}' with filtering on year={filter_year}. Use a less restrictive query or do not filter on year."
-                )
-            else:
-                raise Exception(f"No results found for query: '{query}'. Use a less restrictive query.")
-        if len(results["organic"]) == 0:
-            year_filter_message = f" with filter year={filter_year}" if filter_year is not None else ""
-            return f"No results found for '{query}'{year_filter_message}. Try with a more general query, or remove the year filter."
-
-        web_snippets = []
-        if "organic" in results:
-            for idx, page in enumerate(results["organic"]):
-                date_published = ""
-                if "date" in page:
-                    date_published = "\nDate published: " + page["date"]
-
-                source = ""
-                if "source" in page:
-                    source = "\nSource: " + page["source"]
-
-                snippet = ""
-                if "snippet" in page:
-                    snippet = "\n" + page["snippet"]
-
-                redacted_version = f"{idx}. [{page['title']}]({page['link']}){date_published}{source}\n{snippet}"
-
-                redacted_version = redacted_version.replace("Your browser can't play this video.", "")
-                web_snippets.append(redacted_version)
-
-        return "## Search Results\n" + "\n\n".join(web_snippets)
-
-# web_search = GoogleSearchTool()
-
-# print(web_search(query="Donald Trump news"))
-# quit()
 AUTHORIZED_IMPORTS = [
     "requests",
     "zipfile",