added files to hf

azure_ai.py

@@ -0,0 +1,41 @@
+import os
+from dotenv import load_dotenv
+from openai import AzureOpenAI
+
+load_dotenv()
+
+client = AzureOpenAI(
+    api_version="2024-12-01-preview",
+    azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
+    api_key=os.getenv("AZURE_OPENAI_KEY")
+)
+
+def enhance_with_ai(repo_info):
+    file_summaries = "\n".join([
+        f"### {name}\n```{content[:300]}```"
+        for name, content in list(repo_info['files'].items())[:10]
+    ])
+
+    prompt = f"""
+    Create a professional README.md for a GitHub project named "{repo_info['name']}".
+
+    Description: {repo_info['description']}
+    Topics: {', '.join(repo_info['topics'])}
+
+    The project includes the following files:
+    {file_summaries}
+
+    README must include: Introduction, Features, Installation, Usage, Contributing, and License.
+    """
+
+    response = client.chat.completions.create(
+        model=os.getenv("AZURE_DEPLOYMENT_NAME"),
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant that creates GitHub READMEs. You must follow the provided structure and include all necessary sections. Use necessary emojies and markdown formatting."},
+            {"role": "user", "content": prompt}
+        ],
+        temperature=0.7,
+        max_tokens=1000
+    )
+
+    return response.choices[0].message.content

github_analyzer.py

@@ -0,0 +1,55 @@
+from github import Github
+from urllib.parse import urlparse
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+def extract_repo_name(url):
+    path = urlparse(url).path.strip("/")
+    return path.split("/")[-2], path.split("/")[-1]
+
+def is_important_file(path):
+    ignored_dirs = ["node_modules", "test", "__tests__", ".git", "build", ".next", ".vscode"]
+    important_dirs = ["src", "components", "pages", "hooks", "controller", "service"]
+    important_exts = [".py", ".js", ".ts", ".jsx", ".tsx", ".java"]
+
+    if any(ignored in path for ignored in ignored_dirs):
+        return False
+
+    if any(dir_ in path for dir_ in important_dirs) and any(path.endswith(ext) for ext in important_exts):
+        return True
+
+    if path.count("/") <= 1 and any(path.endswith(ext) for ext in important_exts):
+        return True
+
+    return False
+
+def get_filtered_file_contents(repo, path=""):
+    contents = repo.get_contents(path)
+    all_files = {}
+    for content in contents:
+        if content.type == "dir":
+            all_files.update(get_filtered_file_contents(repo, content.path))
+        else:
+            if is_important_file(content.path):
+                try:
+                    file_data = content.decoded_content.decode("utf-8")
+                    all_files[content.path] = file_data
+                except Exception as e:
+                    all_files[content.path] = f"Unable to read: {e}"
+    return all_files
+
+def analyze_repo(repo_url):
+    g = Github(os.getenv("GITHUB_TOKEN"))
+    owner, name = extract_repo_name(repo_url)
+    repo = g.get_repo(f"{owner}/{name}")
+
+    info = {
+        "name": repo.name,
+        "description": repo.description,
+        "topics": repo.get_topics(),
+        "files": get_filtered_file_contents(repo),
+        "url": repo.html_url
+    }
+    return info

license_generator.py

@@ -0,0 +1,19 @@
+import datetime
+import os
+
+def generate_license(license_type, username):
+    year = datetime.datetime.now().year
+    template_path = f"templates/licenses/{license_type}.txt"
+
+    if not os.path.exists(template_path):
+        return None
+
+    with open(template_path, "r", encoding="utf-8") as file:
+        content = file.read()
+
+    content = content.replace("[year]", str(year)).replace("[fullname]", username)
+
+    with open("LICENSE", "w", encoding="utf-8") as file:
+        file.write(content)
+
+    return license_type

readme_generator.py

@@ -0,0 +1,7 @@
+def generate_readme(repo_info, content, license_type=None):
+    with open("README.md", "w", encoding="utf-8") as f:
+        f.write(f"# {repo_info['name']}\n\n")
+        f.write(content)
+        if license_type:
+            f.write(f"\n\n## License\nThis project is licensed under the **{license_type}** License.")
+        f.write(f"\n\n---\n🔗 GitHub Repo: {repo_info['url']}")

requirements.txt

@@ -0,0 +1,6 @@
+Flask
+Flask-Cors
+openai
+PyGithub
+python-dotenv
+gunicorn