add dfs simpler example

app.py

@@ -4,7 +4,6 @@ import requests as req
 
 code_nl = "function for db connection"
 
-# CT5_URL = "https://api-inference.huggingface.co/models/nielsr/codet5-small-code-summarization-ruby"
 CT5_URL = "https://api-inference.huggingface.co/models/stmnk/codet5-small-code-summarization-python"
 CT5_METHOD = 'POST'
 API_URL = CT5_URL
@@ -15,6 +14,15 @@ def query(payload):
 	return response.json()
 
 
+dfs_code = r"""
+def dfs(visited, graph, node):  #function for dfs 
+    if node not in visited:
+        print (node)
+        visited.add(node)
+        for neighbour in graph[node]:
+            dfs(visited, graph, neighbour)
+"""
+
 function_code = r"""
 def write_documents(self, documents: Union[List[dict], List[Document]], index: Optional[str] = None,
                         batch_size: int = 10_000, duplicate_documents: Optional[str] = None):
@@ -71,7 +79,8 @@ def write_documents(self, documents: Union[List[dict], List[Document]], index: O
 
 """
 
-task_code = f' Summarize Python: {function_code}'
+# task_code = f' Summarize Python: {function_code}'
+task_code = f' Summarize Python: {dfs_code}'
 
 real_docstring = r"""
         Indexes documents for later queries in Elasticsearch.
@@ -108,13 +117,14 @@ def docgen_func(function_code):
     return str(output)
 
 def pygen_func(nl_code_intent):
-    pass # TODO: generate code PL from intent NL
+    pass # TODO: generate code PL from intent NL + search in corpus
     # inputs = {'code_nl': code_nl}    
     # payload = json.dumps(inputs)
     # prediction = req.request(CT5_METHOD, CT5_URL, data=payload)
     # prediction = req.request(CT5_METHOD, CT5_URL, json=req_data)
     # answer = json.loads(prediction.content.decode("utf-8"))
     # return str(answer)
+    # CT5_URL = "https://api-inference.huggingface.co/models/nielsr/codet5-small-code-summarization-ruby"
 
 iface = gr.Interface(
     # pygen_func,