Update app.py

app.py

@@ -76,7 +76,34 @@ def aitext(word):
 
     return response.choices[0].text
 
+import json
+headers = {"Authorization": f"Bearer {'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'}"}
+def sumy(payload):
+    API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
+    data = json.dumps(payload)
+    response = requests.request("POST", API_URL, headers=headers, data=data)
+    return json.loads(response.content.decode("utf-8"))
+
+
+def query(payload):
+    API_URL = "https://api-inference.huggingface.co/models/gpt2"
+    data = json.dumps(payload)
+    response = requests.request("POST", API_URL, headers=headers, data=data)
+    return json.loads(response.content.decode("utf-8"))
+
 def google(name): 
+            if "give" in name:
+              r_c = name.split("give")[1]
+
+              result_count = int(r_c.split()[0])
+              #print(result_count)
+              name = name.split("give")[0]
+              print(name)
+
+            else:
+              result_count = 1
+
+            f_result = ""
             result = {"",""}
             text =""
             if "how to learn" in name or "steps for learning" in name or "step for learning" in name or "steps for" in name or "step for" in name:
@@ -91,13 +118,13 @@ def google(name):
             soup = BeautifulSoup(r.text,"html.parser")
             
             heading_object=soup.find_all('div')
-           
+
             for info in heading_object:
-    
                 
                 if '<div class="BNeawe s3v9rd AP7Wnd"><div><div><div class="BNeawe s3v9rd AP7Wnd">' in str(info):
                     if '›' not in str(info.text) :
                       result.add(info.text)
+            
             n=0
             for i in result:
                 if n!=0:
@@ -111,7 +138,27 @@ def google(name):
                    i = i[0]
                    text = text +str(n)+"\t"+i+"\n\n"
                 n=n+1
-            return text
+            
+            if result_count == 1:
+                 temp = ""
+                 for r in text.split("\n\n"): 
+                     temp = temp+r.split("...")[0]
+                 f_result = sumy({"inputs":temp,"parameters": {"do_sample": False,"max_length":300}})
+                 return f_result[0]['summary_text']
+            else:
+              for r in text.split("\n\n"): 
+                  if "..." in r:
+                    r = r.split("...")
+                    print(r[0].strip(),1)
+                    w = query(r[0].strip())[0]['generated_text'].strip()
+                    if "\n\n" in w:
+                      print("=")
+                    f_result = f_result +w+"\n\n"
+                  else:
+                    print("=")
+                    f_result = f_result + r.strip()+"\n\n"
+
+              return f_result
 
 def greet(name1):
         name = name1.lower()